118 lines
3.8 KiB
Python
118 lines
3.8 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(cbq):
|
|
try:
|
|
data = cbq.get("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(join_request):
|
|
try:
|
|
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):
|
|
try:
|
|
await handle_routing(message, state)
|
|
except Exception as e:
|
|
logger.error(f"[main.all_handler] ERROR {e}")
|
|
logger.debug(message)
|
|
import traceback
|
|
text = traceback.format_exc()
|
|
await send_message(FEEDBACK_CHAT_ID, text)
|
|
|
|
|
|
async def chat_members_change(update):
|
|
try:
|
|
old = update.get("old_chat_member")
|
|
if old:
|
|
if old.get("status") == 'KICKED':
|
|
Profile.erase(update["from"]["id"])
|
|
await handle_left(update)
|
|
elif update.get("new_chat_member"):
|
|
await handle_join(update)
|
|
else:
|
|
logger.info("unhandled members update")
|
|
|
|
except Exception as e:
|
|
logger.error(f"[main.my_chat_member] ERROR {e}")
|
|
logger.debug(update)
|
|
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())
|