63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
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)
|