From 1c8939cf3886530fa8464aa4c0f0bafb4a435d2a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=90=D1=80=D1=82=D0=B5=D0=BC=20=D0=9F=D0=B0=D0=B2=D0=BB?= =?UTF-8?q?=D0=BE=D0=B2?= Date: Fri, 24 Nov 2023 12:51:39 +0300 Subject: [PATCH 1/2] com_reg 2 --- src/bot/bot.py | 3 ++- src/bot/plugins/command_registration.py | 20 ++++++++++++++++++++ src/depends.py | 2 ++ src/settings.py | 1 + 4 files changed, 25 insertions(+), 1 deletion(-) create mode 100644 src/bot/plugins/command_registration.py diff --git a/src/bot/bot.py b/src/bot/bot.py index 358d152..7be4b32 100644 --- a/src/bot/bot.py +++ b/src/bot/bot.py @@ -1,6 +1,7 @@ from mmpy_bot import Bot, Settings from src.bot.plugins.admin import BotAdmin +from src.bot.plugins.command_registration import CommandRegistration from src.bot.plugins.matching import Matching from src.bot.plugins.registration import Registration from src.bot.plugins.week_routine import WeekRoutine @@ -20,6 +21,6 @@ def init_bot(config: Settings_bot) -> Bot: WEBHOOK_HOST_URL=config.WEBHOOK_HOST_URL, WEBHOOK_HOST_PORT=config.WEBHOOK_HOST_PORT, ), - plugins=[Registration(), BotAdmin(), WeekRoutine(), Matching()], + plugins=[Registration(), BotAdmin(), WeekRoutine(), Matching(), CommandRegistration()], ) return bot diff --git a/src/bot/plugins/command_registration.py b/src/bot/plugins/command_registration.py new file mode 100644 index 0000000..1203c57 --- /dev/null +++ b/src/bot/plugins/command_registration.py @@ -0,0 +1,20 @@ +from dependency_injector.wiring import Provide, inject +from mmpy_bot import Plugin + +from src.depends import Container + + +class CommandRegistration(Plugin): + @inject + def on_start( + self, + settings=Provide(Container.settings), + ) -> None: + self.driver.commands.create_command( + options={ + "team_id": settings.CHANNEL_ID, + "method": "G", + "trigger": "rand", + "url": "https://www.random.org/integers/?num=1&min=1&max=100&col=1&base=10&format=plain&rnd=new&cl=w", + } + ) diff --git a/src/depends.py b/src/depends.py index 6238834..3d4cf5a 100644 --- a/src/depends.py +++ b/src/depends.py @@ -12,6 +12,8 @@ from src.endpoints import Endpoints from src.settings import Settings +# from mattermostautodriver.endpoints.commands import Commands + class Container(containers.DeclarativeContainer): # Settings diff --git a/src/settings.py b/src/settings.py index 45e6972..6f3494b 100644 --- a/src/settings.py +++ b/src/settings.py @@ -29,6 +29,7 @@ class Settings(BaseSettings): BOT_TOKEN: str BOT_TEAM: str = "" SSL_VERIFY: bool = False + CHANNEL_ID: str # logging settings LOG_FILE_LEVEL: str = "DEBUG" LOG_CONSOLE_LEVEL: str = "INFO" From bc5de3b52e04a228d9bb521d3484f67bf2f80b41 Mon Sep 17 00:00:00 2001 From: Artem Pavlov <108401543+sea-dot-ya@users.noreply.github.com> Date: Fri, 24 Nov 2023 12:56:51 +0300 Subject: [PATCH 2/2] Update depends.py --- src/depends.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/depends.py b/src/depends.py index 3d4cf5a..6238834 100644 --- a/src/depends.py +++ b/src/depends.py @@ -12,8 +12,6 @@ from src.endpoints import Endpoints from src.settings import Settings -# from mattermostautodriver.endpoints.commands import Commands - class Container(containers.DeclarativeContainer): # Settings