From 063e25e578dc30cd2647cf90880cea6b9bb080f9 Mon Sep 17 00:00:00 2001 From: def Date: Tue, 10 Jan 2023 17:29:58 +0400 Subject: [PATCH] feat: abstract method run --- mirrortea/abstract_source_platform/abstact_source_platform.py | 4 ++++ mirrortea/abstract_source_platform/telegram.py | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/mirrortea/abstract_source_platform/abstact_source_platform.py b/mirrortea/abstract_source_platform/abstact_source_platform.py index f12d8b3..a087541 100644 --- a/mirrortea/abstract_source_platform/abstact_source_platform.py +++ b/mirrortea/abstract_source_platform/abstact_source_platform.py @@ -7,3 +7,7 @@ class AbstractSourcePlatform(ABC): @abstractmethod def get_user_information(self) -> User: """Init tables in database""" + + @abstractmethod + def run(self) -> None: + """Platform launch""" diff --git a/mirrortea/abstract_source_platform/telegram.py b/mirrortea/abstract_source_platform/telegram.py index 0cb3916..5c4e383 100644 --- a/mirrortea/abstract_source_platform/telegram.py +++ b/mirrortea/abstract_source_platform/telegram.py @@ -14,7 +14,7 @@ class Telegram(AbstractSourcePlatform): self.dispatcher = telegram.Dispatcher(bot=self.bot) self.dispatcher.register_message_handler(self.on_message) - async def run(self): + async def run(self) -> None: await self.dispatcher.start_polling() async def on_message(self, msg):