version-0.2.0

handlers/callback_vouch.py

@@ -0,0 +1,62 @@
+from bot.api import approve_chat_join_request, edit_replymarkup
+from storage import Profile, storage
+import logging
+
+logger = logging.getLogger(__name__)
+logging.basicConfig(level=logging.INFO)
+
+
+async def update_button(chat_id, member_id, text="❤️"):
+    button_message_id = storage.get(f"btn-{chat_id}-{member_id}")
+    print(f"button_message_id: {button_message_id}")
+    if button_message_id:
+        button_message_id = button_message_id.decode("utf-8")
+        print(f"button_message_id: {button_message_id}")
+        print("update reply markup")
+        newcomer = Profile.get(member_id)
+        amount = len(newcomer["parents"]) + 1
+        text = f"❤️ {amount}"
+        rm = {
+            "inline_keyboard": [[{"text": text, "callback_data": "vouch" + member_id}]]
+        }
+        r = await edit_replymarkup(chat_id, button_message_id, reply_markup=rm)
+        print(r)
+
+
+async def handle_button(callback_query):
+    # получаем профиль нажавшего кнопку
+    actor_id = str(callback_query["from"]["id"])
+    actor = Profile.get(actor_id, callback_query)
+
+    callback_data = callback_query["data"]
+    if callback_data.startswith("vouch"):
+        print(f"button pressed by {actor_id}")
+
+        newcomer_id = callback_data[5:]
+        print(f"button pressed for {newcomer_id}")
+
+        newcomer = Profile.get(newcomer_id)
+        print(f"newcomer profile {newcomer}")
+        if newcomer_id == actor_id:
+            # нажал сам, не реагируем, прописываем данные
+            _newcomer = Profile.get(newcomer_id, callback_query)
+        else:
+            # нажал кто-то другой
+
+            if str(actor_id) not in newcomer["parents"]:
+                print(f"save parent for {newcomer_id}")
+                newcomer["parents"].append(str(actor_id))
+                Profile.save(newcomer)
+
+            if str(newcomer_id) not in actor["children"]:
+                print(f"save child for {actor_id}")
+                actor["children"].append(str(newcomer_id))
+                Profile.save(actor)
+
+            chat_id = str(vars(callback_query["message"]["chat"])["id"])
+
+            print("accept join request")
+            r = await approve_chat_join_request(chat_id, newcomer_id)
+            print(r)
+
+            await update_button(chat_id, newcomer_id)