welcomecenterbot/bot/handlers/handle_startup.py

35 lines
1.4 KiB
Python
Raw Normal View History

2023-09-11 20:21:55 +00:00
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
2023-04-28 12:24:14 +00:00
2023-09-11 20:04:53 +00:00
2023-04-28 12:24:14 +00:00
# устанавливает соответствие данных
2023-09-11 20:04:53 +00:00
async def handle_startup():
btn_ids, _btns = scan(match="btn-*", count=100)
2023-04-28 12:24:14 +00:00
for btnid in btn_ids:
# для каждой ранее созданной кнопки
2023-04-29 20:37:28 +00:00
btnid_str = btnid.decode("utf-8").replace("btn-", "")
2023-09-11 20:04:53 +00:00
parts = btnid_str.split("-")
2023-04-29 20:37:28 +00:00
print(parts)
_, chat_id, member_id = parts
chat_id = "-" + chat_id
newcomer = Profile.get(member_id)
2023-09-11 20:04:53 +00:00
if len(newcomer.get("parents", [])) > 0:
2023-04-29 20:37:28 +00:00
# принять заявку если её нажимали
2023-09-11 20:04:53 +00:00
r = await approve_chat_join_request(chat_id, member_id)
2023-04-29 20:37:28 +00:00
print(r)
2023-05-04 08:42:52 +00:00
update_button(chat_id, member_id)
2023-09-11 20:04:53 +00:00
elif len(newcomer.get("parents", [])) == 0:
r = await kick_member(chat_id, member_id)
2023-04-29 20:37:28 +00:00
print(r)
2023-09-11 20:04:53 +00:00
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 = await send_message(chat_id, body)
print(r)