Files
welcomecenterbot/tgbot/handlers.py

167 lines
6.3 KiB
Python
Raw Normal View History

2023-04-21 13:06:48 +03:00
from tgbot.api import send_message, forward_message, delete_message, \
2023-04-21 13:09:10 +03:00
ban_member, unban_member, set_chatpermissions
2023-04-18 00:42:25 +03:00
from tgbot.config import FEEDBACK_CHAT_ID, WELCOME_MSG, BUTTON_NO, \
2023-04-21 13:09:10 +03:00
BUTTON_OK, CHAT_ID, REDIS_URL
2023-04-18 00:42:25 +03:00
import json
import redis
2023-04-21 13:09:10 +03:00
from tgbot.profile import Profile as ProfileObj
2023-04-18 00:42:25 +03:00
# сохраняет сессии и пересылаемые сообщения между перезагрузками
storage = redis.from_url(REDIS_URL)
2023-04-21 13:09:10 +03:00
# хранение необходимой информации о пользователях
Profile = ProfileObj(storage)
2023-04-21 13:06:48 +03:00
2023-04-18 00:42:25 +03:00
def handle_feedback(msg):
mid = msg['message_id']
cid = msg['chat']['id']
r = forward_message(cid, mid, FEEDBACK_CHAT_ID).json()
support_msg_id = r['result']['message_id']
2023-04-21 13:09:10 +03:00
# сохранение айди сообщения в приватной переписке с ботом
2023-04-18 00:42:25 +03:00
storage.set(f'fbk-{support_msg_id}', json.dumps({
2023-04-21 13:06:48 +03:00
"author_id": msg["from"]["id"],
2023-04-18 00:42:25 +03:00
"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'])
2023-04-21 13:09:10 +03:00
# получение сохраненного айди сообщения из личной переписки с ботом
2023-04-18 00:42:25 +03:00
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"]) # notice 'u' before private chat ID
print(r.json())
2023-04-21 13:09:10 +03:00
def handle_join(msg):
2023-04-18 00:42:25 +03:00
chat_id = str(msg['chat']['id'])
from_id = str(msg['from']['id'])
member_id = str(msg['new_chat_member']['id'])
2023-04-21 13:09:10 +03:00
2023-04-18 00:42:25 +03:00
if from_id == member_id:
2023-04-21 13:09:10 +03:00
newcomer = Profile.get(member_id)
2023-04-18 00:42:25 +03:00
print(f'new self-joined member {member_id}')
reply_markup = {
"inline_keyboard": [
[
{"text": BUTTON_NO, "callback_data": BUTTON_NO},
2023-04-21 13:09:10 +03:00
{"text": BUTTON_OK, "callback_data": BUTTON_OK},
{"text": BUTTON_VOUCH, "callback_data": BUTTON_VOUCH}
2023-04-18 00:42:25 +03:00
]
]
}
r = send_message(
chat_id,
WELCOME_MSG,
reply_to=msg['message_id'],
reply_markup=reply_markup
)
welcome_msg_id = r.json()['result']['message_id']
2023-04-21 13:06:48 +03:00
print(r.json())
2023-04-18 00:42:25 +03:00
print(f'welcome message id: {welcome_msg_id}')
2023-04-21 13:09:10 +03:00
newcomer["newcomer"] = True
newcomer["welcome_id"] = welcome_msg_id
2023-04-21 13:06:48 +03:00
perms = {
"can_send_messages": False
}
r = set_chatpermissions(CHAT_ID, perms)
print(r.json())
2023-04-21 13:09:10 +03:00
# обновляем профиль новичка
Profile.save(newcomer)
2023-04-18 00:42:25 +03:00
2023-04-21 13:09:10 +03:00
elif 'new_chat_members' in msg:
# кто-то пригласил новых участников
print(f'{len(msg["new_chat_members"])} members were invited by {from_id}')
# получаем его профиль
inviter = Profile.get(from_id)
for m in msg['new_chat_members']:
newcomer = Profile.get(m['id'])
newcomer['vouched_by'].append(from_id)
Profile.save(newcomer)
inviter['vouched_for'].append(m['id'])
# обновляем профиль пригласившего
Profile.save(inviter)
2023-04-18 00:42:25 +03:00
def handle_left(msg):
2023-04-18 12:22:43 +03:00
print(f'handling member leaving')
2023-04-18 00:42:25 +03:00
member_id = msg["left_chat_member"]["id"]
2023-04-21 13:09:10 +03:00
# профиль покидающего чат
leaver = Profile.get(member_id)
2023-04-18 00:42:25 +03:00
2023-04-21 13:09:10 +03:00
r = delete_message(CHAT_ID, leaver['welcome_id'])
print(r.json())
2023-04-18 00:42:25 +03:00
2023-04-21 13:09:10 +03:00
Profile.leaving(leaver)
2023-04-18 00:42:25 +03:00
def handle_button(callback_query):
2023-04-18 12:22:43 +03:00
if 'reply_to_message' not in callback_query['message']:
2023-04-21 13:09:10 +03:00
# удаляет сообщение с кнопкой, если оно уже ни на что не отвечает
2023-04-18 12:22:43 +03:00
r = delete_message(CHAT_ID, callback_query['message'])
print(r.json())
else:
member_id = str(callback_query['from']['id'])
callback_data = callback_query['data']
2023-04-21 13:09:10 +03:00
welcomed_member_id = str(callback_query['message']['reply_to_message']['from']['id'])
2023-04-18 12:22:43 +03:00
welcome_msg_id = str(callback_query['message']['message_id'])
enter_msg_id = str(callback_query['message']['reply_to_message']['message_id'])
2023-04-21 13:09:10 +03:00
# получаем профиль нажавшего кнопку
actor = Profile.get(member_id)
if welcomed_member_id == member_id:
2023-04-18 12:22:43 +03:00
print(f'callback_query in {CHAT_ID}')
if callback_data == BUTTON_NO:
print('wrong answer, cleanup')
r = delete_message(CHAT_ID, enter_msg_id)
print(r.json())
r = delete_message(CHAT_ID, welcome_msg_id)
print(r.json())
print('ban member')
r = ban_member(CHAT_ID, member_id)
print(r.json())
2023-04-21 13:09:10 +03:00
# обработка профиля заблокированного пользователя
Profile.leaving(actor)
2023-04-18 12:22:43 +03:00
elif callback_data == BUTTON_OK:
print('proper answer, cleanup')
r = delete_message(CHAT_ID, welcome_msg_id)
print(r.json())
2023-04-21 13:09:10 +03:00
actor['newcomer'] = False
r = delete_message(CHAT_ID, author["welcome_id"])
print(r.json())
r = set_chatpermissions(CHAT_ID, { "can_send_messages": True })
print(r.json())
# обновление профиля нажавшего правильную кнопку
Profile.save(actor)
elif callback_data == BUTTON_VOUCH:
# это кнопка поручения
print(f'vouch button pressed by {member_id}')
newcomer = Profile.get(welcomed_member_id)
if welcomed_member_id not in inviter['vouched_for'] and \
member_id not in newcomer['vouched_by']:
newcomer['vouched_by'].append(welcomed_member_id)
actor['vouched_for'].append(member_id)
Profile.save(newcomer)
Profile.save(actor)
print('vouch success, unban member')
r = unban_member(CHAT_ID, member_id)
print(r.json())