Fixed group add event

middlewares/user_data_middleware.py

@@ -1,19 +1,33 @@
 import logging
 
 from vkwave.bots import BaseMiddleware, BotEvent, MiddlewareResult, SimpleBotEvent
+from vkwave.types.bot_events import GroupJoinObject, MessageNewObject
+from vkwave.types.objects import UsersUser
 
 from db import DB, Candidate
 
 
+async def get_user(api_ctx, user_id):
+    raw_user = (
+        await api_ctx.api_request("users.get", {"user_ids": user_id})
+    )["response"][0]
+    return UsersUser(**raw_user)
+
+
 class UserMiddleware(BaseMiddleware):
     async def pre_process_event(self, event: BotEvent) -> MiddlewareResult:
         db = DB()
-        botevent = SimpleBotEvent(event)
-        user_id = event.object.object.message.from_id
+
+        if isinstance(event.object.object, GroupJoinObject):
+            user_id = event.object.object.user_id
+        elif isinstance(event.object.object, MessageNewObject):
+            user_id = event.object.object.message.from_id
+        else:
+            return MiddlewareResult(True)
 
         user = db.get_user(user_id)
         if not user:
-            user_info = await botevent.get_user()
+            user_info = await get_user(event.api_ctx, user_id)
             logging.debug(f"Got user info: {user_info}")
             user = Candidate(id=user_id, sex=user_info.sex, name=user_info.first_name, last_name=user_info.last_name)
             db.add_candidate(user)