dokku-comp

bot/handlers/handle_members_change.py

@@ -3,42 +3,42 @@ from bot.api import delete_message
 from bot.storage import Profile, storage
 from bot.config import FEEDBACK_CHAT_ID
 
-def handle_join(msg):
-    chat_id = str(msg['chat']['id'])
-    from_id = str(msg['from']['id'])
+
+async def handle_join(msg):
+    chat_id = str(msg["chat"]["id"])
+    from_id = str(msg["from"]["id"])
 
     actor = Profile.get(from_id, msg)
 
-    newcomer_id = str(msg['new_chat_member']['id'])
+    newcomer_id = str(msg["new_chat_member"]["id"])
     if from_id == newcomer_id:
-        if len(actor['parents']) == 0 and str(chat_id) != FEEDBACK_CHAT_ID:
+        if len(actor["parents"]) == 0 and str(chat_id) != FEEDBACK_CHAT_ID:
             # показываем сообщение с кнопкой "поручиться"
-            show_request_msg(msg)
+            r = await show_request_msg(msg)
+            print(r)
         else:
             # за пользователя поручились ранее
             pass
     else:
         # пользователи приглашены другим участником
-        print(f'{len(msg["new_chat_members"])} members were invited by {from_id}')    
-        for m in msg['new_chat_members']:
-            newcomer = Profile.get(m['id'])
-            newcomer['parents'].append(str(from_id))
+        print(f'{len(msg["new_chat_members"])} members were invited by {from_id}')
+        for m in msg["new_chat_members"]:
+            newcomer = Profile.get(m["id"])
+            newcomer["parents"].append(str(from_id))
             Profile.save(newcomer)
-            actor['children'].append(str(m['id']))
+            actor["children"].append(str(m["id"]))
         # обновляем профиль пригласившего
         Profile.save(actor)
 
 
-def handle_left(msg):
-    print(f'handling member leaving')
+async def handle_left(msg):
+    print("handling member leaving")
     member_id = msg["left_chat_member"]["id"]
-    chat_id = msg['chat']['id']
+    chat_id = msg["chat"]["id"]
 
     # удаление сообщения с кнопкой в этом чате
-    prev_msg = storage.get(f'btn-{chat_id}-{member_id}')
+    prev_msg = storage.get(f"btn-{chat_id}-{member_id}")
     if prev_msg:
-        r = delete_message(chat_id, prev_msg['id'])
+        r = await delete_message(chat_id, prev_msg["id"])
         print(r)
-        storage.remove(f'btn-{chat_id}-{member_id}')
-
-
+        storage.remove(f"btn-{chat_id}-{member_id}")