refactored-src

src/handlers/callback_unlink.py

@@ -0,0 +1,54 @@
+from api import send_message, delete_message, kick_member
+from handlers.command_my import handle_command_my
+from utils.mention import userdata_extract
+from storage import Profile
+
+# remove link of callback sender 
+# from member vouched before
+def handle_unlink(payload):
+    print('handle unlink button pressed or command, private chat only')
+    
+    from_id = str(payload['from']['id'])
+    linked_id = ''
+    if 'data' in payload:
+        linked_id = str(payload['data'].replace('unlink', ''))
+    elif 'text' in payload:
+        linked_id = str(payload['text'].replace('/unlink ', ''))
+
+    # удаляем связь с потомком
+    actor = Profile.get(from_id, payload)
+    actor['children'].remove(str(linked_id))
+    Profile.save(actor)
+
+    # удаляем связь с предком
+    linked = Profile.get(linked_id)
+    linked['parents'].remove(str(from_id))
+    Profile.save(linked)
+
+    
+    # удаляем старое сообщение с кнопками-unlink
+    reply_msg_id = payload['message']['message_id']
+    r = delete_message(from_id, reply_msg_id)
+    print(r)
+     
+    # если ещё есть связи - посылаем новое сообщение                                                    
+    if len(actor['children']) > 0:
+        handle_command_my(payload)
+
+    lang = payload['from'].get('language_code', 'ru')
+    for chat_id in linked['chats']:
+        
+        # если больше никто не поручился - kick out
+        if len(linked['parents']) == 0:
+            r = kick_member(chat_id, linked_id)
+            print(r)
+            if r['ok']:
+                _, identity, username = userdata_extract(linked['result']['user'])
+                body = ('Участник %s%s был удалён' if lang == 'ru' else 'Member %s%s was deleted') % (identity, username)
+                r = send_message(chat_id, body)
+                print(r)
+
+        # обновление счётчика
+        update_button(linked_id, chat_id)
+    
+            

src/handlers/callback_vouch.py

@@ -0,0 +1,66 @@
+from api import send_message, forward_message, delete_message, \
+    approve_chat_join_request, edit_replymarkup, get_chat
+from storage import Profile, storage
+
+
+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 = edit_replymarkup(chat_id, button_message_id, reply_markup=rm)
+        print(r)
+
+
+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(callback_query['message']['chat']['id'])
+
+                print('accept join request')
+                r = approve_chat_join_request(chat_id, newcomer_id)
+                print(r)
+
+                update_button(chat_id, newcomer_id)

src/handlers/command_ask.py

@@ -0,0 +1,17 @@
+from storage import Profile
+from handlers.send_button import show_request_msg
+from api import get_member
+
+def handle_command_ask(msg):
+    print(f'handling request resend')
+    cmd, chat_id, member_id = msg['text'].split(' ')
+    chat_id = chat_id.replace('-', '-100')
+    r = get_member(chat_id, member_id)
+    print(r)
+    m = {}
+    if 'result' in r:
+        m['from'] = r['result']['user']
+        m['chat'] = { 'id': str(chat_id) }
+        show_request_msg(m)
+    elif 'error_code' in r:
+        print(r)

src/handlers/command_graph.py

@@ -0,0 +1,12 @@
+from utils.graph import generate_chart
+from api import send_document
+from storage import storage, scan
+import json
+
+
+async def handle_command_graph(msg):
+    usr_ids, members = scan(match='usr-*', count=100)
+    data = generate_chart(members)
+    if data:
+        r = await send_document(msg['chat']['id'], data, 'chart.svg')
+        print(r)

src/handlers/command_my.py

@@ -0,0 +1,56 @@
+from storage import Profile, scan
+from api import get_member, send_message, get_chat_administrators
+from utils.mention import userdata_extract
+
+
+def construct_unlink_buttons(actor):
+    buttons = []
+    for vouch in actor['children']:
+        for chat_id in actor['chats']:
+            r = get_member(chat_id, vouch)
+            member = r['result']['user']
+            uid, identity, username = userdata_extract(member)
+            buttons.append({
+                'text': f'{identity} {username}',
+                'callback_data': 'unlink' + vouch
+            })
+    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)
+
+    handle_command_owner_my(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)
+
+
+def handle_command_owner_my(msg):
+    chat_id = msg['chat']['id']
+    r = get_chat_administrators(chat_id)
+    print(r)
+    owner_id = ''
+    for admin in r['result']:
+        if admin['status'] == 'creator':
+            owner_id = str(admin['user']['id'])
+            break
+    if owner_id:
+        owner = Profile.get(owner_id, msg)
+        uids, members = scan()
+        for mdata in members:
+            m = json.loads(mdata.decode('utf-8'))
+            if owner_id in m['parents']:
+                if str(m['id']) not in owner['children']:
+                    owner['children'].append(str(m['id']))
+                    Profile.save(owner)

src/handlers/handle_default.py

@@ -0,0 +1,42 @@
+from api import send_message, delete_message, get_chat_administrators
+from storage import Profile
+from handlers.send_button import show_request_msg
+from storage import storage
+
+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)
+
+        # показать новое сообщение с кнопкой
+        show_request_msg(msg)
+    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 = ''
+            for admin in r['result']:
+                if admin['status'] == 'creator':
+                    owner_id = admin['user']['id']
+                    break
+            if owner_id:
+                sender['parents'].append(str(owner_id))
+                # обновляем профиль владельца
+                owner = Profile.get(owner_id)
+                owner['children'].append(str(from_id))
+                Profile.save(owner)
+
+    # сохранить профиль отправителя
+    Profile.save(sender)

src/handlers/handle_feedback.py

@@ -0,0 +1,44 @@
+import json
+
+from api import send_message, forward_message, delete_message, get_chat_administrators
+from handlers.send_button import show_request_msg
+from utils.mention import userdata_extract
+from storage import storage, Profile
+from config import FEEDBACK_CHAT_ID
+
+
+def handle_feedback(msg):
+    mid = msg['message_id']
+    cid = msg['chat']['id']
+    if msg['text'] == '/start':
+        r = send_message(cid, 'Напишите своё сообщение для администрации чата')
+        print(r)
+    else:
+        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):
+    answered_msg = msg['reply_to_message']
+    r = get_chat_administrators(msg['chat']['id'])
+    print(r)
+    admins = []
+    for a in r['result']:
+        admins.append(a['user']['id'])
+    if answered_msg['from']['is_bot'] and msg['from']['id'] in admins:
+        support_msg_id = str(answered_msg['message_id'])
+        # получение сохраненного информации о сообщении для ответа
+        stored_feedback = storage.get(f'fbk-{support_msg_id}')
+        if stored_feedback:
+            print(f'handle answer from support')
+            stored_feedback = json.loads(stored_feedback)
+            r = send_message(f'{stored_feedback["chat_id"]}', msg['text'], reply_to=stored_feedback["message_id"])
+            print(r)
+

src/handlers/handle_join_request.py

@@ -0,0 +1,20 @@
+from api import approve_chat_join_request, delete_message           
+from handlers.send_button import show_request_msg
+from storage import Profile, storage
+
+
+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:
+        # показываем сообщение с кнопкой "поручиться"
+        show_request_msg(msg)
+    else:
+        # за пользователя поручились ранее
+        r = approve_chat_join_request(chat_id, from_id)
+        print(r)
+    Profile.save(actor)
+

src/handlers/handle_members_change.py

@@ -0,0 +1,44 @@
+from handlers.send_button import show_request_msg
+from api import delete_message
+from storage import Profile, storage
+from config import FEEDBACK_CHAT_ID
+
+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 and str(chat_id) != FEEDBACK_CHAT_ID:
+            # показываем сообщение с кнопкой "поручиться"
+            show_request_msg(msg)
+        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(str(from_id))
+            Profile.save(newcomer)
+            actor['children'].append(str(m['id']))
+        # обновляем профиль пригласившего
+        Profile.save(actor)
+
+
+def handle_left(msg):
+    print(f'handling member leaving')
+    member_id = msg["left_chat_member"]["id"]
+    chat_id = msg['chat']['id']
+
+    # удаление сообщения с кнопкой в этом чате
+    prev_msg = storage.get(f'btn-{chat_id}-{member_id}')
+    if prev_msg:
+        r = delete_message(chat_id, prev_msg['id'])
+        print(r)
+        storage.remove(f'btn-{chat_id}-{member_id}')
+
+

src/handlers/handle_startup.py

@@ -0,0 +1,29 @@
+from storage import scan, Profile
+from api import approve_chat_join_request, kick_member
+from handlers.callback_vouch import update_button
+from utils.mention import userdata_extract
+
+# устанавливает соответствие данных
+def handle_startup():
+    btn_ids, btns = scan(match='btn-*', count=100)
+    for btnid in btn_ids:
+        # для каждой ранее созданной кнопки
+        btnid_str = btnid.decode("utf-8").replace("btn-", "")
+        parts = btnid_str.split('-')
+        print(parts)
+        _, chat_id, member_id = parts
+        chat_id = "-" + chat_id
+        newcomer = Profile.get(member_id)
+        if len(newcomer.get('parents', [])) > 0:
+            # принять заявку если её нажимали
+            r = approve_chat_join_request(chat_id, member_id)
+            print(r)
+            update_button(chat_id, member_id)
+        elif len(newcomer.get('parents', [])) == 0:
+            r = kick_member(chat_id, member_id)
+            print(r)
+            if r['ok']:
+                _, identity, username = userdata_extract(newcomer['result']['user'])
+                body = ('Участник %s%s был удалён' if lang == 'ru' else 'Member %s%s was deleted') % (identity, username)
+                r = send_message(chat_id, body)
+                print(r)

src/handlers/send_button.py

@@ -0,0 +1,52 @@
+from api import send_message, send_photo, get_userphotos
+from utils.mention import mention, userdata_extract
+from storage import storage
+
+def show_request_msg(msg):
+    chat_id = str(msg['chat']['id'])
+    from_id = str(msg['from']['id'])
+    lang = msg['from'].get('language_code', 'ru')
+    reply_markup = {
+        "inline_keyboard": [
+            [
+                {
+                    "text": '❤️', 
+                    "callback_data": 'vouch' + from_id
+                }
+            ]
+        ]
+    }
+    newcomer_message = "Нажмите, чтобы одобрить заявку " if lang == 'ru' \
+        else "There is a newcomer, press the button if you are connected with "
+    r = get_userphotos(user_id=from_id)
+    print(r)
+    if r['ok'] and r['result']['total_count'] > 0:
+        print('show button with photo')
+        file_id = r['result']['photos'][0][0]['file_id']
+        _uid, identity, username = userdata_extract(msg['from'])
+        r = send_photo(
+            chat_id,
+            file_id,
+            caption=newcomer_message + f'{identity}{username}',
+            reply_to=msg.get('message_id', ''),
+            reply_markup=reply_markup
+        )
+    else:
+        print('show button without photo')
+        r = send_message(
+            chat_id,
+            newcomer_message + mention(msg['from']),   
+            reply_to=msg.get('message_id', ''),
+            reply_markup=reply_markup
+        )
+    print(r)
+    if 'message_id' in r:
+        # удаляем предыдущее сообщение с кнопкой в этом чате
+        prevbtn = storage.get(f'btn-{chat_id}-{from_id}')
+        if prevbtn:
+            r = delete_message(chat_id, prevbtn)
+            print(r)
+        # создаём новое
+        newbtn = r['message_id']
+        print(f'button message id: {newbtn}')
+        storage.set(f'btn-{chat_id}-{from_id}', newbtn)