added OwnerMiddleware
This commit is contained in:
@@ -76,4 +76,4 @@ def create_tables_if_not_exists(dbfile: str) -> None:
|
|||||||
connection.commit()
|
connection.commit()
|
||||||
cursor.close()
|
cursor.close()
|
||||||
connection.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 config import Config
|
||||||
|
|
||||||
from .commands import register_handlers
|
from .commands import register_handlers
|
||||||
|
from .middlewares import OwnerMiddleware
|
||||||
|
|
||||||
|
|
||||||
config = Config()
|
config = Config()
|
||||||
telegram_bot = Bot(token=config.teletoken)
|
telegram_bot = Bot(token=config.teletoken)
|
||||||
telegram_dispatcher = Dispatcher(telegram_bot)
|
telegram_dispatcher = Dispatcher(telegram_bot)
|
||||||
telegram_dispatcher.middleware.setup(LoggingMiddleware())
|
telegram_dispatcher.middleware.setup(LoggingMiddleware())
|
||||||
|
telegram_dispatcher.middleware.setup(OwnerMiddleware())
|
||||||
register_handlers(telegram_dispatcher)
|
register_handlers(telegram_dispatcher)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -2,7 +2,6 @@ from dataclasses import dataclass
|
|||||||
import typing
|
import typing
|
||||||
|
|
||||||
from aiogram import Dispatcher
|
from aiogram import Dispatcher
|
||||||
from aiogram.types import Message
|
|
||||||
|
|
||||||
from .test import telegram_test
|
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