added OwnerMiddleware

This commit is contained in:
Alexey Sokolov
2022-08-23 00:42:30 +03:00
parent 4aaed03655
commit 3753f4c6ef
4 changed files with 22 additions and 2 deletions

View File

@@ -76,4 +76,4 @@ def create_tables_if_not_exists(dbfile: str) -> None:
connection.commit()
cursor.close()
connection.close()
logging.warning('Database created!')
logging.info('Database created!')

View File

@@ -4,12 +4,14 @@ from aiogram.contrib.middlewares.logging import LoggingMiddleware
from config import Config
from .commands import register_handlers
from .middlewares import OwnerMiddleware
config = Config()
telegram_bot = Bot(token=config.teletoken)
telegram_dispatcher = Dispatcher(telegram_bot)
telegram_dispatcher.middleware.setup(LoggingMiddleware())
telegram_dispatcher.middleware.setup(OwnerMiddleware())
register_handlers(telegram_dispatcher)

View File

@@ -2,7 +2,6 @@ from dataclasses import dataclass
import typing
from aiogram import Dispatcher
from aiogram.types import Message
from .test import telegram_test

19
telegram/middlewares.py Normal file
View File

@@ -0,0 +1,19 @@
import logging
from aiogram.types import Message
from aiogram.dispatcher.middlewares import BaseMiddleware
from config import Config
class OwnerMiddleware(BaseMiddleware):
"""
Ignore commands from non-owner.
"""
def __init__(self):
self.bot_owner = Config().bot_owner
super(OwnerMiddleware, self).__init__()
async def on_pre_process_message(self, message: Message, _):
if message.chat.id != self.bot_owner:
message.text = ''