135 lines
4.5 KiB
Python
135 lines
4.5 KiB
Python
import asyncio
|
|
import logging
|
|
import signal
|
|
from aiohttp import ClientSession
|
|
from handlers.command_my import send_message
|
|
from handlers.routing import handle_routing
|
|
from handlers.callback_unlink import handle_unlink
|
|
from handlers.callback_vouch import handle_button
|
|
from handlers.handle_join_request import handle_join_request
|
|
from handlers.handle_startup import handle_startup
|
|
from handlers.handle_members_change import handle_join, handle_left
|
|
from bot.config import BOT_TOKEN, FEEDBACK_CHAT_ID
|
|
from bot.state import State
|
|
from storage import Profile
|
|
|
|
logger = logging.getLogger(__name__)
|
|
logging.basicConfig(level=logging.INFO)
|
|
state = State()
|
|
|
|
api_url = f'https://api.telegram.org/bot{BOT_TOKEN}/'
|
|
|
|
|
|
async def fetch(session, url):
|
|
async with session.get(url) as response:
|
|
return await response.json()
|
|
|
|
|
|
async def command_start_handler(message):
|
|
caption = "Напишите своё сообщение для нас" if message.from_user.llanguage_code == 'ru' else "Write your message for us"
|
|
await message.answer(caption)
|
|
|
|
|
|
async def process_callback(callback_query):
|
|
cbq = vars(callback_query)
|
|
try:
|
|
cbq["from"] = vars(callback_query.from_user)
|
|
cbq["message"] = vars(callback_query.message)
|
|
data = cbq["data"]
|
|
if data.startswith("vouch"):
|
|
await handle_button(cbq)
|
|
elif data.startswith("unlink"):
|
|
await handle_unlink(cbq, state)
|
|
except Exception as e:
|
|
logger.error(f"[main.process_callback] ERROR {e}")
|
|
logger.debug(cbq)
|
|
import traceback
|
|
text = traceback.format_exc()
|
|
await send_message(FEEDBACK_CHAT_ID, text)
|
|
|
|
|
|
async def join_request_handler(update):
|
|
print("chat join request")
|
|
join_request = vars(update)
|
|
try:
|
|
join_request["from"] = vars(update.from_user)
|
|
join_request["chat"] = vars(update.chat)
|
|
await handle_join_request(join_request)
|
|
except Exception as e:
|
|
logger.error(f"[main.join_request_handler] ERROR {e}")
|
|
logger.debug(join_request)
|
|
import traceback
|
|
text = traceback.format_exc()
|
|
await send_message(FEEDBACK_CHAT_ID, text)
|
|
|
|
|
|
async def all_handler(message):
|
|
msg = vars(message)
|
|
try:
|
|
msg["from"] = vars(message.from_user)
|
|
msg["chat"] = vars(message.chat)
|
|
if message.reply_to_message:
|
|
msg["reply_to_message"] = vars(message.reply_to_message)
|
|
await handle_routing(msg, state)
|
|
await asyncio.sleep(1.0)
|
|
except Exception as e:
|
|
logger.error(f"[main.all_handler] ERROR {e}")
|
|
logger.debug(msg)
|
|
import traceback
|
|
text = traceback.format_exc()
|
|
await send_message(FEEDBACK_CHAT_ID, text)
|
|
|
|
|
|
async def chat_members_change(update):
|
|
member_updated = vars(update)
|
|
try:
|
|
member_updated["chat"] = vars(update.chat)
|
|
member_updated["from"] = vars(update.from_user)
|
|
old_member = vars(member_updated["old_chat_member"])
|
|
new_member = vars(member_updated["new_chat_member"])
|
|
if old_member:
|
|
if old_member.status == 'KICKED':
|
|
Profile.erase(member_updated["from"]["id"])
|
|
await handle_left(member_updated)
|
|
elif new_member:
|
|
await handle_join(member_updated)
|
|
else:
|
|
logger.info("unhandled members update")
|
|
|
|
except Exception as e:
|
|
logger.error(f"[main.my_chat_member] ERROR {e}")
|
|
logger.debug(member_updated)
|
|
import traceback
|
|
text = traceback.format_exc()
|
|
await send_message(FEEDBACK_CHAT_ID, text)
|
|
|
|
|
|
async def main():
|
|
# storage revalidation
|
|
await handle_startup()
|
|
|
|
async with ClientSession() as session:
|
|
offset = 0 # начальное значение offset
|
|
while True:
|
|
updates = await fetch(session, f"{api_url}getUpdates?offset={offset}")
|
|
for update in updates["result"]:
|
|
if "message" in update:
|
|
await all_handler(update["message"])
|
|
elif "callback_query" in update:
|
|
await process_callback(update["callback_query"])
|
|
elif "join_chat_request" in update:
|
|
await join_request_handler(update["join_chat_request"])
|
|
elif "my_chat_member" in update:
|
|
await chat_members_change(update["my_chat_member"])
|
|
|
|
offset = update["update_id"] + 1 # обновление offset
|
|
|
|
await asyncio.sleep(1.0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logging.basicConfig(level=logging.INFO, stream=sys.stdout)
|
|
|
|
# Запуск асинхронного цикла
|
|
asyncio.run(main())
|