Added technician module #44

kdb-bot/src/modules/technician/__init__.py

@@ -0,0 +1 @@
+# imports: 

kdb-bot/src/modules/technician/command/restart_command.py

@@ -0,0 +1,57 @@
+import asyncio
+
+from cpl_core.configuration import ConfigurationABC
+from cpl_discord.command import DiscordCommandABC
+from cpl_discord.service import DiscordBotServiceABC
+from cpl_translation import TranslatePipe
+from discord.ext import commands
+from discord.ext.commands import Context
+
+from bot_core.abc.client_utils_service_abc import ClientUtilsServiceABC
+from bot_core.abc.message_service_abc import MessageServiceABC
+from bot_core.configuration.bot_settings import BotSettings
+from bot_core.helper.command_checks import CommandChecks
+from bot_core.logging.command_logger import CommandLogger
+from modules.permission.abc.permission_service_abc import PermissionServiceABC
+
+
+class RestartCommand(DiscordCommandABC):
+
+    def __init__(
+            self,
+            logger: CommandLogger,
+            config: ConfigurationABC,
+            message_service: MessageServiceABC,
+            bot: DiscordBotServiceABC,
+            client_utils: ClientUtilsServiceABC,
+            translate: TranslatePipe,
+            permissions: PermissionServiceABC,
+            settings: BotSettings
+    ):
+        DiscordCommandABC.__init__(self)
+
+        self._logger = logger
+        self._config = config
+        self._message_service = message_service
+        self._bot = bot
+        self._client_utils = client_utils
+        self._t = translate
+        self._permissions = permissions
+        self._settings = settings
+
+        self._logger.trace(__name__, f'Loaded command service: {type(self).__name__}')
+
+    @commands.hybrid_command()
+    @commands.guild_only()
+    @CommandChecks.check_is_ready()
+    @CommandChecks.check_is_member_technician()
+    async def restart(self, ctx: Context):
+        self._logger.debug(__name__, f'Received command restart {ctx}')
+
+        self._config.add_configuration('IS_RESTART', 'true')
+        await self._client_utils.presence_game('common.presence.restart')
+        await self._message_service.send_ctx_msg(ctx, self._t.transform('modules.technician.restart_message'))
+        await asyncio.sleep(self._settings.wait_for_restart)
+        await self._bot.stop_async()
+
+        self._logger.trace(__name__, f'Finished restart command')

kdb-bot/src/modules/technician/command/shutdown_command.py

@@ -0,0 +1,57 @@
+import asyncio
+
+import discord
+from cpl_core.configuration import ConfigurationABC
+from cpl_discord.command import DiscordCommandABC
+from cpl_discord.service import DiscordBotServiceABC
+from cpl_translation import TranslatePipe
+from discord.ext import commands
+from discord.ext.commands import Context
+
+from bot_core.abc.client_utils_service_abc import ClientUtilsServiceABC
+from bot_core.abc.message_service_abc import MessageServiceABC
+from bot_core.configuration.bot_settings import BotSettings
+from bot_core.helper.command_checks import CommandChecks
+from bot_core.logging.command_logger import CommandLogger
+from modules.permission.abc.permission_service_abc import PermissionServiceABC
+
+
+class ShutdownCommand(DiscordCommandABC):
+
+    def __init__(
+            self,
+            logger: CommandLogger,
+            config: ConfigurationABC,
+            message_service: MessageServiceABC,
+            bot: DiscordBotServiceABC,
+            client_utils: ClientUtilsServiceABC,
+            translate: TranslatePipe,
+            permissions: PermissionServiceABC,
+            settings: BotSettings
+    ):
+        DiscordCommandABC.__init__(self)
+
+        self._logger = logger
+        self._config = config
+        self._message_service = message_service
+        self._bot = bot
+        self._client_utils = client_utils
+        self._t = translate
+        self._permissions = permissions
+        self._settings = settings
+
+        self._logger.trace(__name__, f'Loaded command service: {type(self).__name__}')
+
+    @commands.hybrid_command()
+    @commands.guild_only()
+    @CommandChecks.check_is_ready()
+    @CommandChecks.check_is_member_technician()
+    async def shutdown(self, ctx: Context):
+        self._logger.debug(__name__, f'Received command shutdown {ctx}')
+
+        await self._client_utils.presence_game('common.presence.shutdown')
+        await self._message_service.send_ctx_msg(ctx, self._t.transform('modules.technician.shutdown_message'))
+        await asyncio.sleep(self._settings.wait_for_shutdown)
+        await self._bot.stop_async()
+
+        self._logger.trace(__name__, f'Finished shutdown command')

kdb-bot/src/modules/technician/technician.json

@@ -0,0 +1,46 @@
+{
+  "ProjectSettings": {
+    "Name": "technician",
+    "Version": {
+      "Major": "0",
+      "Minor": "0",
+      "Micro": "0"
+    },
+    "Author": "",
+    "AuthorEmail": "",
+    "Description": "",
+    "LongDescription": "",
+    "URL": "",
+    "CopyrightDate": "",
+    "CopyrightName": "",
+    "LicenseName": "",
+    "LicenseDescription": "",
+    "Dependencies": [
+      "cpl-core>=2022.10.0.post7"
+    ],
+    "DevDependencies": [
+      "cpl-cli>=2022.10.0"
+    ],
+    "PythonVersion": ">=3.10.6",
+    "PythonPath": {
+      "win32": ""
+    },
+    "Classifiers": []
+  },
+  "BuildSettings": {
+    "ProjectType": "library",
+    "SourcePath": "",
+    "OutputPath": "../../dist",
+    "Main": "technician.main",
+    "EntryPoint": "technician",
+    "IncludePackageData": false,
+    "Included": [],
+    "Excluded": [
+      "*/__pycache__",
+      "*/logs",
+      "*/tests"
+    ],
+    "PackageData": {},
+    "ProjectReferences": []
+  }
+}

kdb-bot/src/modules/technician/technician_module.py

@@ -0,0 +1,27 @@
+from cpl_core.configuration import ConfigurationABC
+from cpl_core.dependency_injection import ServiceCollectionABC
+from cpl_core.environment import ApplicationEnvironmentABC
+from cpl_discord.service.discord_collection_abc import DiscordCollectionABC
+
+from bot_core.abc.module_abc import ModuleABC
+from bot_core.configuration.feature_flags_enum import FeatureFlagsEnum
+from modules.base.abc.base_helper_abc import BaseHelperABC
+from modules.technician.command.restart_command import RestartCommand
+from modules.technician.command.shutdown_command import ShutdownCommand
+from modules.base.service.base_helper_service import BaseHelperService
+
+
+class TechnicianModule(ModuleABC):
+
+    def __init__(self, dc: DiscordCollectionABC):
+        ModuleABC.__init__(self, dc, FeatureFlagsEnum.base_module)
+
+    def configure_configuration(self, config: ConfigurationABC, env: ApplicationEnvironmentABC):
+        pass
+
+    def configure_services(self, services: ServiceCollectionABC, env: ApplicationEnvironmentABC):
+        services.add_transient(BaseHelperABC, BaseHelperService)
+        # commands
+        self._dc.add_command(RestartCommand)
+        self._dc.add_command(ShutdownCommand)
+        # events