version-0.2.0

This commit is contained in:
2024-01-06 14:25:35 +03:00
parent bc05b93c47
commit 908529b5bb
26 changed files with 185 additions and 153 deletions

126
main.py Normal file
View 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) # Выберите подходящий порт