0.0.8-newage

tgbot/handlers/callback_unlink.py

@@ -0,0 +1,23 @@
+from tgbot.api import send_message, delete_message
+from tgbot.storage import Profile
+
+# remove link of callback sender 
+# from member vouched before
+def handle_unlink(callback_query):
+    print('handle unlink button pressed, private chat only')
+    
+    from_id = str(callback_query['from']['id'])
+    reply_msg_id = callback_query['message']['message_id']
+
+    actor = Profile.get(from_id, callback_query)
+    actor['parents'].remove(from_id)
+    
+    # удаляем старое сообщение с кнопками
+    r = delete_message(from_id, reply_msg_id)
+    print(r)
+
+    # если ещё есть связи - посылаем новое сообщение
+    if len(actor['parents']) > 0:
+        body = construct_unlink_buttons(actor)
+        r = send_message(from_id, body)
+        print(r)

tgbot/handlers/callback_vouch.py

@@ -0,0 +1,38 @@
+from tgbot.api import send_message, forward_message, delete_message
+from tgbot.storage import Profile
+
+
+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(BUTTON_VOUCH):
+        print(f'vouch button pressed by {actor_id}')
+
+        newcomer_id = callback_data[len(BUTTON_VOUCH):]
+        newcomer = Profile.get(newcomer_id)
+        if newcomer_id == actor_id:
+            # нажал сам, не реагируем, прописываем данные
+            newcomer = Profile.get(newcomer_id, callback_query)
+        elif newcomer_id not in actor['children'] and \
+            actor_id not in newcomer['parents']:
+                newcomer['parents'].append(actor_id)
+                actor['children'].append(newcomer_id)
+                Profile.save(newcomer)
+                Profile.save(actor)
+                try:
+                    chat_id = str(callback_query['message']['chat']['id'])
+
+                    print('unmute newcomer')
+                    r = unmute_member(chat_id, newcomer_id)
+                    print(r)
+
+                    print('accept join request')
+                    r = approve_chat_join_request(chat_id, newcomer_id)
+                    print(r)
+
+                except:
+                    pass
+       

tgbot/handlers/command_graph.py

@@ -0,0 +1,29 @@
+from tgbot.utils.graph import generate_chart
+from tgbot.api import send_document
+from tgbot.storage import storage
+import json
+
+
+async def handle_command_graph(msg):
+    cursor = 0
+    keys = []
+    r = storage
+    while True:
+        # Scan for keys starting with 'usr-*' in batches of 100
+        cursor, batch_keys = r.scan(cursor=cursor, match='usr-*', count=100)
+        keys += batch_keys
+        # If the cursor is 0, then we've reached the end of the keys
+        if cursor == 0:
+            break
+    # Get the values of all the keys
+    values = r.mget(keys)
+    # Parse the JSON data from each value
+    members = []
+    for value in values:
+        value_str = value.decode('utf-8')
+        member = json.loads(value_str)
+        members.append(member)
+    print(f'found {len(members)} members')
+    data = generate_chart(members)
+    r = await send_document(msg['chat']['id'], data, 'chart.svg')
+    print(r)

tgbot/handlers/command_my.py

@@ -0,0 +1,40 @@
+from tgbot.storage import Profile
+from tgbot.api import get_member
+
+def construct_unlink_buttons(actor):
+    buttons = []
+    for vouch in actor['children']:
+        vouch_added = False
+        for chat_id in actor['chats']:
+            if not vouch_added:
+                r = get_member(chat_id, vouch)
+                member = r.get('result')
+                if member:
+                    try:
+                        buttons.append({
+                            'text': mention(r['result']),
+                            'callback_data': 'unlink' + vouch
+                        })
+                        vouch_added = True
+                    except:
+                        print('member result error')
+                        print(member)
+                else:
+                    print(r)
+    return { "inline_keyboard": [ buttons, ] }
+
+def handle_command_my(msg):
+    print(f'handle my command')
+    from_id = str(msg['from']['id'])
+    sender = Profile.get(from_id, msg)
+
+    # генерируем кнопки для всех, за кого поручились
+    reply_markup = construct_unlink_buttons(sender)
+
+    if msg['from'].get('language_code', 'ru') == 'ru':
+        body = 'Нажмите кнопки ниже, чтобы удалить ваши связи'
+    else:
+        body = 'Unlink your connections pressing the buttons below'
+
+    r = send_message(from_id, body, reply_markup=reply_markup)
+    print(r)

tgbot/handlers/handle_default.py

@@ -0,0 +1,47 @@
+from tgbot.api import send_message, delete_message, get_chat_administrators
+from tgbot.storage import Profile
+from tgbot.handlers.send_button import show_request_msg
+
+
+def handle_default(msg):
+    print(f'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'):
+        # команда в групповом чате
+        print(f'remove some messages in group chat')
+
+        # удалить сообщение с командой /my
+        r = delete_message(chat_id, msg['message_id'])
+        print(r)
+
+        # удалить предыдушее сообщение с кнопкой в этом чате
+        if sender['request_msg_id'].startswith(chat_id):
+            chat_id, rmid = sender['request_msg_id'].split(':')
+            r = delete_message(chat_id, rmid)
+            print(r)
+
+        # показать новое сообщение с кнопкой
+        request_msg_id = show_request_msg(msg)
+        sender['request_msg_id'] = f'{chat_id}:{request_msg_id}'
+    else:
+        # любое другое сообщение
+        if len(sender['parents']) == 0:
+            # владелец чата автоматически ручается
+            print(f'setting owner as parent for {from_id}')
+            r = get_chat_administrators(chat_id)
+            print(r)
+            
+            owner_id = r['result'][0]['id']  # DEBUG!!
+
+            sender['parents'].append(owner_id)
+            
+            # обновляем профиль владельца
+            owner = Profile.get(owner_id)
+            owner['children'].append(from_id)
+            Profile.save(owner)
+
+    # сохранить профиль отправителя
+    Profile.save(sender)

tgbot/handlers/handle_feedback.py

@@ -0,0 +1,29 @@
+import json
+
+from tgbot.api import send_message, forward_message, delete_message
+from tgbot.storage import storage, Profile
+from tgbot.config import FEEDBACK_CHAT_ID
+
+
+def handle_feedback(msg):
+    mid = msg['message_id']
+    cid = msg['chat']['id']
+    r = forward_message(cid, mid, FEEDBACK_CHAT_ID)
+    support_msg_id = r['result']['message_id']
+    # сохранение айди сообщения в приватной переписке с ботом
+    storage.set(f'fbk-{support_msg_id}', json.dumps({
+        "author_id": msg["from"]["id"],
+        "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'])
+    # получение сохраненного айди сообщения из личной переписки с ботом
+    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"])
+    print(r)
+

tgbot/handlers/handle_join_request.py

@@ -0,0 +1,26 @@
+from tgbot.api import approve_chat_join_request, delete_message           
+from tgbot.handlers.send_button import show_request_msg
+from tgbot.storage import Profile
+
+
+def handle_join_request(msg):
+    print(f'handle join request {msg}')
+    chat_id = str(msg['chat']['id'])
+    from_id = str(msg['from']['id'])
+    actor = Profile.get(from_id, msg)
+
+    if len(actor['parents']) == 0:
+        # показываем сообщение с кнопкой "поручиться"
+        request_msg_id = show_request_msg(msg)
+        # удаляем предыдущее сообщение с кнопкой в этом чате
+        if actor['request_msg_id'].startswith(chat_id):
+            chat_id, rmid = actor['request_msg_id'].split(':')
+            r = delete_message(chat_id, rmid)
+            print(r)
+        actor['request_msg_id'] = f'{chat_id}:{request_msg_id}'
+    else:
+        # за пользователя поручились ранее
+        r = approve_chat_join_request(chat_id, from_id)
+        print(r)
+    Profile.save(actor)
+

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)
+

tgbot/handlers/send_button.py

@@ -0,0 +1,27 @@
+from tgbot.api import send_message
+from tgbot.config import BUTTON_VOUCH, NEWCOMER_MSG
+from tgbot.utils import mention
+
+def show_request_msg(msg):
+    reply_markup = {
+        "inline_keyboard": [
+            [
+                {
+                    "text": BUTTON_VOUCH, 
+                    "callback_data": BUTTON_VOUCH + str(msg['from']['id'])
+                }
+            ]
+        ]
+    }
+    
+    r = send_message(
+        msg['chat']['id'],
+        NEWCOMER_MSG + mention(msg['from']),   
+        reply_to=msg.get('message_id', ''),
+        reply_markup=reply_markup
+    )
+    request_msg_id = r['result']['message_id']
+    chat_id = msg['chat']['id']
+    print(r)
+    print(f'request message id: {chat_id}:{request_msg_id}')
+    return request_msg_id