welcomecenterbot/handlers/callback_vouch.py

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)