94 lines
3.3 KiB
Python
94 lines
3.3 KiB
Python
from sanic import Sanic
|
||
from sanic.response import text
|
||
from bot.handlers.handle_feedback import handle_feedback, handle_answer
|
||
from bot.handlers.handle_members_change import handle_join, handle_left
|
||
from bot.handlers.handle_join_request import handle_join_request
|
||
from bot.handlers.handle_default import handle_default
|
||
from bot.handlers.command_my import handle_command_my
|
||
from bot.handlers.command_graph import handle_command_graph
|
||
from bot.handlers.command_ask import handle_command_ask
|
||
from bot.handlers.callback_vouch import handle_button
|
||
from bot.handlers.callback_unlink import handle_unlink
|
||
from bot.handlers.handle_startup import handle_startup
|
||
from bot.api import register_webhook, send_message
|
||
from bot.config import FEEDBACK_CHAT_ID
|
||
from bot.state import State
|
||
|
||
app = Sanic(name="welcomecenter")
|
||
app.config.REGISTERED = False
|
||
|
||
state = State()
|
||
|
||
|
||
@app.get('/')
|
||
async def register(req):
|
||
if not app.config.REGISTERED:
|
||
print(register_webhook())
|
||
app.config.REGISTERED = True
|
||
handle_startup()
|
||
return text('ok')
|
||
|
||
|
||
@app.post('/')
|
||
async def handle(req):
|
||
print(req)
|
||
try:
|
||
update = req.json
|
||
print(update)
|
||
|
||
# видимые сообщения
|
||
msg = update.get('message', update.get('edited_message'))
|
||
if msg:
|
||
msg['edit'] = 'edited_message' in update
|
||
|
||
if msg['chat']['id'] == msg['from']['id']:
|
||
# сообщения в личке с ботом
|
||
print('private chat message')
|
||
if msg['text'].startswith('/my'):
|
||
handle_command_my(msg)
|
||
elif msg['text'].startswith('/unlink'):
|
||
handle_unlink(msg)
|
||
else:
|
||
handle_feedback(msg, state)
|
||
|
||
elif str(msg['chat']['id']) == FEEDBACK_CHAT_ID:
|
||
# сообщения из группы обратной связи
|
||
print('feedback chat message')
|
||
if 'reply_to_message' in msg:
|
||
handle_answer(msg)
|
||
elif msg['text'] == '/graph':
|
||
await handle_command_graph(msg)
|
||
elif msg['text'].startswith('/ask'):
|
||
handle_command_ask(msg)
|
||
|
||
else:
|
||
# сообщения из всех остальных групп
|
||
cid = msg['chat']['id']
|
||
print(f'group {cid} chat message')
|
||
if 'text' in msg:
|
||
handle_default(msg)
|
||
elif 'new_chat_member' in msg:
|
||
handle_join(msg)
|
||
elif 'left_chat_member' in msg:
|
||
handle_left(msg)
|
||
|
||
# кнопки
|
||
elif 'callback_query' in update:
|
||
data = update['callback_query']['data']
|
||
if data.startswith('vouch'):
|
||
handle_button(update['callback_query'])
|
||
elif data.startswith('unlink'):
|
||
handle_unlink(update['callback_query'])
|
||
|
||
# заявки
|
||
elif 'chat_join_request' in update:
|
||
print('chat join request')
|
||
handle_join_request(update['chat_join_request'])
|
||
|
||
|
||
except Exception:
|
||
import traceback
|
||
r = send_message(FEEDBACK_CHAT_ID, f'<pre>\n{traceback.format_exc()}\n</pre>')
|
||
traceback.print_exc()
|
||
return text('ok')
|