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 = ''