dokku-comp

bot/handlers/handle_default.py

@@ -2,18 +2,19 @@ from bot.api import send_message, delete_message, get_chat_administrators
 from bot.storage import Profile, storage
 from bot.handlers.send_button import show_request_msg
 
-def handle_default(msg):
-    print('default handler for all messages')
-    chat_id = str(msg['chat']['id'])
-    from_id = str(msg['from']['id'])
+
+async def handle_default(msg):
+    print("default handler for all messages")
+    chat_id = str(msg["chat"]["id"])
+    from_id = str(msg["from"]["id"])
     sender = Profile.get(from_id, msg)
 
-    if msg['text'].startswith('/my'):
+    if msg["text"].startswith("/my"):
         # команда в групповом чате
-        print('remove some messages in group chat')
+        print("remove some messages in group chat")
 
         # удалить сообщение с командой /my
-        r = delete_message(chat_id, msg['message_id'])
+        r = await delete_message(chat_id, msg["message_id"])
         print(r)
 
         # показать новое сообщение с кнопкой
@@ -21,22 +22,22 @@ def handle_default(msg):
         show_request_msg(msg)
     else:
         # любое другое сообщение
-        if len(sender['parents']) == 0:
+        if len(sender["parents"]) == 0:
             # владелец чата автоматически ручается
-            print(f'setting owner as parent for {from_id}')
-            r = get_chat_administrators(chat_id)
+            print(f"setting owner as parent for {from_id}")
+            r = await get_chat_administrators(chat_id)
             print(r)
-            owner_id = ''
-            for admin in r['result']:
-                if admin['status'] == 'creator':
-                    owner_id = admin['user']['id']
+            owner_id = ""
+            for admin in r["result"]:
+                if admin["status"] == "creator":
+                    owner_id = admin["user"]["id"]
                     break
             if owner_id:
-                sender['parents'].append(str(owner_id))
+                sender["parents"].append(str(owner_id))
                 # обновляем профиль владельца
                 owner = Profile.get(owner_id)
-                owner['children'].append(str(from_id))
+                owner["children"].append(str(from_id))
                 Profile.save(owner)
 
     # сохранить профиль отправителя
-    Profile.save(sender)
+    Profile.save(sender)