version-0.2.0
This commit is contained in:
126
main.py
Normal file
126
main.py
Normal file
@@ -0,0 +1,126 @@
|
||||
import asyncio
|
||||
import logging
|
||||
import signal
|
||||
import sys
|
||||
from aiohttp import web
|
||||
from bot.api 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 storage.profile import Profile
|
||||
from bot.state import State
|
||||
from bot.config import FEEDBACK_CHAT_ID
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
state = State()
|
||||
|
||||
|
||||
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 == ChatMemberStatus.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(request):
|
||||
# storage revalidation
|
||||
await handle_startup()
|
||||
|
||||
# Получение обновлений
|
||||
updates = await request.json()
|
||||
|
||||
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"])
|
||||
|
||||
return web.Response()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
logging.basicConfig(level=logging.INFO, stream=sys.stdout)
|
||||
|
||||
app = web.Application()
|
||||
app.router.add_post('/your_bot_token', main) # Замените 'your_bot_token' на реальный токен вашего бота
|
||||
web.run_app(app, host='0.0.0.0', port=3000) # Выберите подходящий порт
|
Reference in New Issue
Block a user