0.0.3-debugged

2023-04-18 00:42:25 +03:00
parent a1d9b95a58
commit 023666ddf1
7 changed files with 217 additions and 181 deletions
--- a/tgbot/handlers.py
+++ b/tgbot/handlers.py
@@ -0,0 +1,150 @@
+from tgbot.api import send_message, forward_message, delete_message, ban_member
+from tgbot.config import FEEDBACK_CHAT_ID, WELCOME_MSG, BUTTON_NO, \
+    BUTTON_OK, CHAT_ID, BUTTON_OK2, REDIS_URL
+import json
+import redis
+
+# сохраняет сессии и пересылаемые сообщения между перезагрузками
+storage = redis.from_url(REDIS_URL)
+
+
+def handle_feedback(msg):
+    mid = msg['message_id']
+    cid = msg['chat']['id']
+    r = forward_message(cid, mid, FEEDBACK_CHAT_ID).json()
+    support_msg_id = r['result']['message_id']
+    # store private chat message id 
+    # fbk-<support-chat-message-id> -> <private-chat-id>:<private-message-id>
+    storage.set(f'fbk-{support_msg_id}', json.dumps({
+        "message_id": mid,
+        "chat_id": cid
+    }))
+
+
+def handle_answer(msg):
+    print(f'handle answer from support')
+    support_msg_id = str(msg['reply_to_message']['message_id'])
+    # get stored private chat id
+    stored_feedback = storage.get(f'fbk-{support_msg_id}')
+    stored_feedback = json.loads(stored_feedback)
+    r = send_message(f'{stored_feedback["chat_id"]}', msg['text'], reply_to=stored_feedback["message_id"])  # notice 'u' before private chat ID
+    print(r.json())
+
+
+def handle_welcome(msg):
+    chat_id = str(msg['chat']['id'])
+    from_id = str(msg['from']['id'])
+    member_id = str(msg['new_chat_member']['id'])
+    s = {}
+    if from_id == member_id:
+        print(f'new self-joined member {member_id}')
+        reply_markup = {
+            "inline_keyboard": [
+                [
+                    {"text": BUTTON_NO, "callback_data": BUTTON_NO},
+                    {"text": BUTTON_OK, "callback_data": BUTTON_OK}
+                ]
+            ]
+        }
+        r = send_message(
+            chat_id,
+            WELCOME_MSG,    
+            reply_to=msg['message_id'],
+            reply_markup=reply_markup
+        )
+        welcome_msg_id = r.json()['result']['message_id']
+        print(f'welcome message id: {welcome_msg_id}')
+        s["newcomer"] = True
+        s["welcome_id"] = welcome_msg_id
+    else:
+        s['newcomer'] = False
+
+    # create new member session
+    storage.set(f'usr-{member_id}', json.dumps(s))
+
+
+def handle_left(msg):
+    member_id = msg["left_chat_member"]["id"]
+
+    # read member session
+    s = storage.get(f'usr-{member_id}')
+    if s:
+        s = json.loads(s)
+        r = delete_message(CHAT_ID, s['welcome_id'])
+        print(r.json())
+
+        # remove left member session
+        storage.delete(f'usr-{member_id}')
+
+
+def handle_text(msg):
+    member_id = str(msg['from']['id'])
+
+    # check if author is self-joined newcomer
+    author = storage.get(f'usr-{member_id}')
+
+    if author:
+        author = json.loads(author)
+        if author.get("newcomer"):
+            print(f'new member speaks {msg["text"]}')
+            answer = msg['text']
+            if BUTTON_OK.lower() in answer.lower() or \
+                BUTTON_OK2.lower() in answer.lower():
+                print('found answer, cleanup')
+                r = delete_message(CHAT_ID, author["welcome_id"])
+                print(r.json())
+                author["newcomer"] = False
+
+                # set author as not a newcomer
+                storage.set(f'usr-{member_id}', json.dumps(author))
+
+            else:
+                print('remove some message')
+                r = delete_message(CHAT_ID, msg['message_id'])
+                print(r.json())
+        else:
+            print(f'old member speaks {msg["text"]}')
+
+
+def handle_button(callback_query):
+    member_id = str(callback_query['from']['id'])
+    callback_data = callback_query['data']
+    reply_owner = str(callback_query['message']['reply_to_message']['from']['id'])
+    welcome_msg_id = str(callback_query['message']['message_id'])
+    enter_msg_id = str(callback_query['message']['reply_to_message']['message_id'])
+    if reply_owner == member_id:
+        print(f'callback_query in {CHAT_ID}')
+        
+        # read session
+        s = storage.get(f'usr-{member_id}')
+        if s:
+            s = json.loads(s)
+        else:
+            print('no user session found, create')
+            s = {
+                'newcomer': True,
+                'welcome_id': welcome_msg_id
+            }
+            storage.set(f'usr-{member_id}', json.dumps(s))
+        
+        if callback_data == BUTTON_NO:
+            print('wrong answer, cleanup')
+            r = delete_message(CHAT_ID, enter_msg_id)
+            print(r.json())
+            r = delete_message(CHAT_ID, welcome_msg_id)
+            print(r.json())
+
+            # remove banned member session
+            storage.delete(f'usr-{member_id}')
+
+            print('ban member')
+            r = ban_member(CHAT_ID, member_id)
+            print(r.json())
+        elif callback_data == BUTTON_OK:
+            print('proper answer, cleanup')
+            r = delete_message(CHAT_ID, welcome_msg_id)
+            print(r.json())
+            s['newcomer'] = False
+
+            # store new member session
+            storage.set(f'usr-{member_id}', json.dumps(s))