Skip to content

Commit

Permalink
Отказ от ConfigMiddleware в пользу присвоения переменной объекту bot
Browse files Browse the repository at this point in the history
  • Loading branch information
MasterGroosha committed May 18, 2021
1 parent 174cefd commit 91a5a1d
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 28 deletions.
5 changes: 1 addition & 4 deletions bot/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@
from aiogram import Bot, Dispatcher
from aiogram.types import BotCommand

from bot.middlewares.config import ConfigMiddleware
from bot.handlers.usermode import register_usermode_handlers
from bot.handlers.adminmode import register_adminmode_handlers
from bot.handlers.common import register_common_handlers
Expand Down Expand Up @@ -42,16 +41,14 @@ async def main():
raise ValueError(f'Идентификатор "{admin_chat_id}" не является числом. Бот не может быть запущен.')

bot = Bot(token=token)
bot["admin_chat_id"] = admin_chat_id # Добавление айдишника к объекту bot
dp = Dispatcher(bot)

# Регистрация хэндлеров
register_adminmode_handlers(dp, admin_chat_id)
register_common_handlers(dp)
register_usermode_handlers(dp)

# Регистрация мидлвари
dp.middleware.setup(ConfigMiddleware(admin_chat_id))

# Регистрация /-команд в интерфейсе
await set_bot_commands(bot)

Expand Down
12 changes: 6 additions & 6 deletions bot/handlers/usermode.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,32 +14,32 @@ async def _send_expiring_notification(message: types.Message):
await msg.delete()


async def text_message(message: types.Message, admin_chat: int):
async def text_message(message: types.Message):
"""
Хэндлер на текстовые сообщения от пользователя
:param message: сообщение от пользователя для админа(-ов)
:param admin_chat: ID чата для админа(-ов)
"""
if len(message.text) > 4000:
return await message.reply("К сожалению, длина этого сообщения превышает допустимый размер. "
"Пожалуйста, сократи свою мысль и попробуй ещё раз.")
await message.bot.send_message(admin_chat, message.html_text + f"\n\n#id{message.from_user.id}", parse_mode="HTML")
admin_chat_id = message.bot.get("admin_chat_id")
await message.bot.send_message(admin_chat_id, message.html_text + f"\n\n#id{message.from_user.id}", parse_mode="HTML")
await create_task(_send_expiring_notification(message))


async def supported_media(message: types.Message, admin_chat: int):
async def supported_media(message: types.Message):
"""
Хэндлер на медиафайлы от пользователя.
Поддерживаются только типы, к которым можно добавить подпись (полный список см. в регистраторе внизу)
:param message: медиафайл от пользователя
:param admin_chat: ID чата для админа(-ов)
"""
if message.caption and len(message.caption) > 1000:
return await message.reply("К сожалению, длина подписи медиафайла превышает допустимый размер. "
"Пожалуйста, сократи свою мысль и попробуй ещё раз.")
await message.copy_to(admin_chat,
admin_chat_id = message.bot.get("admin_chat_id")
await message.copy_to(admin_chat_id,
caption=((message.caption or "") + f"\n\n#id{message.from_user.id}"),
parse_mode="HTML")
await create_task(_send_expiring_notification(message))
Expand Down
Empty file removed bot/middlewares/__init__.py
Empty file.
18 changes: 0 additions & 18 deletions bot/middlewares/config.py

This file was deleted.

0 comments on commit 91a5a1d

Please sign in to comment.