added OwnerMiddleware
This commit is contained in:
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