From 5337d4838d1b9d0774cdb8755f76549111944316 Mon Sep 17 00:00:00 2001 From: abdullaev388 Date: Sat, 19 Feb 2022 16:02:14 +0500 Subject: [PATCH] asyncio_middlewares.py was created && BaseMiddleware class was replaced to asyncio_middlewares.py --- .../middleware/flooding_middleware.py | 2 +- examples/asynchronous_telebot/middleware/i18n.py | 3 ++- telebot/asyncio_handler_backends.py | 15 --------------- telebot/asyncio_middlewares.py | 15 +++++++++++++++ 4 files changed, 18 insertions(+), 17 deletions(-) create mode 100644 telebot/asyncio_middlewares.py diff --git a/examples/asynchronous_telebot/middleware/flooding_middleware.py b/examples/asynchronous_telebot/middleware/flooding_middleware.py index b8f589e..761190d 100644 --- a/examples/asynchronous_telebot/middleware/flooding_middleware.py +++ b/examples/asynchronous_telebot/middleware/flooding_middleware.py @@ -1,6 +1,6 @@ # Just a little example of middleware handlers -from telebot.asyncio_handler_backends import BaseMiddleware +from telebot.asyncio_middlewares import BaseMiddleware from telebot.async_telebot import AsyncTeleBot from telebot.async_telebot import CancelUpdate bot = AsyncTeleBot('TOKEN') diff --git a/examples/asynchronous_telebot/middleware/i18n.py b/examples/asynchronous_telebot/middleware/i18n.py index 81281bc..cbbd82b 100644 --- a/examples/asynchronous_telebot/middleware/i18n.py +++ b/examples/asynchronous_telebot/middleware/i18n.py @@ -7,6 +7,7 @@ # But this example just to show the work of middlewares. import telebot +import telebot.asyncio_middlewares from telebot.async_telebot import AsyncTeleBot from telebot import asyncio_handler_backends import logging @@ -27,7 +28,7 @@ TRANSLATIONS = { bot = AsyncTeleBot('TOKEN') -class LanguageMiddleware(asyncio_handler_backends.BaseMiddleware): +class LanguageMiddleware(telebot.asyncio_middlewares.BaseMiddleware): def __init__(self): self.update_types = ['message'] # Update types that will be handled by this middleware. async def pre_process(self, message, data): diff --git a/telebot/asyncio_handler_backends.py b/telebot/asyncio_handler_backends.py index 1784481..51ed603 100644 --- a/telebot/asyncio_handler_backends.py +++ b/telebot/asyncio_handler_backends.py @@ -1,18 +1,3 @@ -class BaseMiddleware: - """ - Base class for middleware. - - Your middlewares should be inherited from this class. - """ - def __init__(self): - pass - - async def pre_process(self, message, data): - raise NotImplementedError - async def post_process(self, message, data, exception): - raise NotImplementedError - - class State: def __init__(self) -> None: self.name = None diff --git a/telebot/asyncio_middlewares.py b/telebot/asyncio_middlewares.py new file mode 100644 index 0000000..cf33280 --- /dev/null +++ b/telebot/asyncio_middlewares.py @@ -0,0 +1,15 @@ +class BaseMiddleware: + """ + Base class for middleware. + + Your middlewares should be inherited from this class. + """ + + def __init__(self): + pass + + async def pre_process(self, message, data): + raise NotImplementedError + + async def post_process(self, message, data, exception): + raise NotImplementedError