added OwnerMiddleware
This commit is contained in:
@@ -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!')
|
||||
@@ -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)
|
||||
|
||||
|
||||
|
||||
@@ -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
19
telegram/middlewares.py
Normal 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 = ''
|
||||
Reference in New Issue
Block a user