0.0.8-newage

tgbot/handlers/handle_members_change.py

@@ -0,0 +1,57 @@
+from tgbot.handlers.send_button import show_request_msg
+from tgbot.api import unmute_member, mute_member, delete_message
+from tgbot.storage import Profile
+
+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'])
+    if from_id == newcomer_id:
+        if len(actor['parents']) == 0:
+            # показываем сообщение с кнопкой "поручиться"
+            request_msg_id = show_request_msg(msg)
+
+            # до одобрения - мьют
+            r = mute_member(chat_id, newcomer_id)
+            print(r)
+
+            # обновляем профиль присоединившегося
+            actor['request_msg_id'] = f'{chat_id}:{request_msg_id}'
+            Profile.save(actor)
+        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(from_id)
+            Profile.save(newcomer)
+            actor['children'].append(m['id'])
+            r = unmute_member(chat_id, newcomer['id'])
+            print(r)
+
+        # обновляем профиль пригласившего
+        Profile.save(actor)
+
+
+def handle_left(msg):
+    print(f'handling member leaving')
+    member_id = msg["left_chat_member"]["id"]
+    chat_id = msg['chat']['id']
+
+    # профиль покидающего чат
+    leaver = Profile.get(member_id)
+
+    # удаление сообщения с кнопкой в этом чате
+    if leaver['request_msg_id'].startswith(chat_id):
+        chat_id, rmid = leaver['request_msg_id'].split(':')
+        r = delete_message(chat_id, rmid)
+        print(r)
+
+    Profile.leaving(leaver)
+