47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
from tgbot.api import send_message, delete_message, get_chat_administrators
|
||
from tgbot.storage import Profile
|
||
from tgbot.handlers.send_button import show_request_msg
|
||
|
||
|
||
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)
|
||
|
||
# удалить предыдушее сообщение с кнопкой в этом чате
|
||
if sender['request_msg_id'].startswith(chat_id):
|
||
chat_id, rmid = sender['request_msg_id'].split(':')
|
||
r = delete_message(chat_id, rmid)
|
||
print(r)
|
||
|
||
# показать новое сообщение с кнопкой
|
||
request_msg_id = show_request_msg(msg)
|
||
sender['request_msg_id'] = f'{chat_id}:{request_msg_id}'
|
||
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 = r['result'][0]['id'] # DEBUG!!
|
||
|
||
sender['parents'].append(owner_id)
|
||
|
||
# обновляем профиль владельца
|
||
owner = Profile.get(owner_id)
|
||
owner['children'].append(from_id)
|
||
Profile.save(owner)
|
||
|
||
# сохранить профиль отправителя
|
||
Profile.save(sender) |