VK

Author: dyakovri

migrations/versions/1cacaf803a1d_user_defined_groups.py

@@ -66,7 +66,7 @@ def upgrade():
     op.create_table(
         'vk_chat',
         sa.Column('id', sa.Integer(), nullable=False),
-        sa.Column('chat_id', sa.Integer(), nullable=False),
+        sa.Column('peer_id', sa.Integer(), nullable=False),
         sa.ForeignKeyConstraint(
             ['id'],
             ['group.id'],

social/models/group.py

@@ -35,7 +35,7 @@ class VkGroup(Group):
 
 class VkChat(Group):
     id: Mapped[int] = mapped_column(sa.ForeignKey("group.id"), primary_key=True)
-    chat_id: Mapped[int]
+    peer_id: Mapped[int]
 
     __mapper_args__ = {
         "polymorphic_identity": "vk_chat",

social/routes/vk.py

@@ -1,4 +1,5 @@
 import logging
+from datetime import UTC, datetime
 
 from auth_lib.fastapi import UnionAuth
 from fastapi import APIRouter, Depends, Request
@@ -7,7 +8,7 @@
 from pydantic import BaseModel, ConfigDict
 
 from social.handlers_telegram import get_application
-from social.models.group import VkGroup
+from social.models.group import VkChat, VkGroup
 from social.models.webhook_storage import WebhookStorage, WebhookSystems
 from social.settings import get_settings
 from social.utils.string import random_string
@@ -54,6 +55,27 @@ async def vk_webhook(request: Request) -> str:
     )
     db.session.commit()
 
+    if request_data.get("type") == "message_new":
+        # Получение сообщения в чате ВК
+        try:
+            peer_id = request_data["object"]["message"]["peer_id"]
+            obj = db.session.query(VkChat).where(VkChat.peer_id == peer_id).one_or_none()
+            if obj is None:
+                # Надо будет добавлять название группы
+                # conversation = requests.post("https://api.vk.com/method/messages.getConversationsById", json={
+                #     "peer_ids": peer_id,
+                #     "group_id": 222099060,
+                #     "access_token": settings.VK_BOT_TOKEN,
+                #     "v": 5.199,
+                # })
+                # chat_title = conversation["response"]["items"][0]["chat_settings"]["title"]
+                obj = VkChat(chat_id=peer_id)
+                db.session.add(obj)
+            obj.last_active_ts = datetime.now(UTC)
+            db.session.commit()
+        except Exception as exc:
+            logger.exception(exc)
+
     return PlainTextResponse('ok')
 
 

social/settings.py

@@ -20,6 +20,8 @@ class Settings(BaseSettings):
 
     TELEGRAM_BOT_TOKEN: str | None = None
 
+    VK_BOT_TOKEN: str | None = None
+
     GITHUB_APP_ID: int | None = None
     GITHUB_WEBHOOK_SECRET: str | None = None
     GITHUB_PRIVATE_KEY: str | None = None