forked from sh-edraft.de/sh_discord_bot
Added deploy command
This commit is contained in:
parent
4b08c99817
commit
b753601979
1
.gitignore
vendored
1
.gitignore
vendored
@ -140,6 +140,7 @@ cython_debug/
|
||||
|
||||
# custom vs code
|
||||
PythonImportHelper-v2-Completion.json
|
||||
deploy/
|
||||
|
||||
# idea
|
||||
.idea/
|
@ -28,7 +28,8 @@
|
||||
"Technicians": [
|
||||
240160344557879316,
|
||||
236592458664902657
|
||||
]
|
||||
],
|
||||
"DeployFilesPath": "../../deploy"
|
||||
},
|
||||
"Base": {
|
||||
"910199451145076828": {
|
||||
|
@ -25,7 +25,8 @@
|
||||
},
|
||||
"Technicians": [
|
||||
240160344557879316
|
||||
]
|
||||
],
|
||||
"DeployFilesPath": "../../deploy"
|
||||
},
|
||||
"Base": {
|
||||
"910199451145076828": {
|
||||
|
@ -27,7 +27,8 @@
|
||||
"Technicians": [
|
||||
240160344557879316,
|
||||
236592458664902657
|
||||
]
|
||||
],
|
||||
"DeployFilesPath": "../../deploy"
|
||||
},
|
||||
"Base": {
|
||||
"650366049023295514": {
|
||||
|
@ -24,7 +24,8 @@
|
||||
"Technicians": [
|
||||
240160344557879316,
|
||||
236592458664902657
|
||||
]
|
||||
],
|
||||
"DeployFilesPath": "../../deploy"
|
||||
},
|
||||
"Base": {
|
||||
"910199451145076828": {
|
||||
|
@ -5,6 +5,7 @@ from cpl_core.environment import ApplicationEnvironmentABC
|
||||
from cpl_discord import get_discord_collection
|
||||
from cpl_discord.discord_event_types_enum import DiscordEventTypesEnum
|
||||
|
||||
from modules.admin.command.deploy_command import DeployCommand
|
||||
from modules.admin.command.restart_command import RestartCommand
|
||||
from modules.admin.command.shutdown_command import ShutdownCommand
|
||||
from modules.base.command.afk_command import AFKCommand
|
||||
@ -39,6 +40,7 @@ class StartupDiscordExtension(StartupExtensionABC):
|
||||
# admin
|
||||
dc.add_command(RestartCommand)
|
||||
dc.add_command(ShutdownCommand)
|
||||
dc.add_command(DeployCommand)
|
||||
# moderator
|
||||
dc.add_command(PurgeCommand)
|
||||
# simple
|
||||
|
@ -14,6 +14,7 @@ class BotSettings(ConfigurationModelABC):
|
||||
|
||||
self._servers: List[ServerSettings] = List()
|
||||
self._technicians: list[int] = []
|
||||
self._deploy_file_path = './'
|
||||
|
||||
@property
|
||||
def servers(self) -> List[ServerSettings]:
|
||||
@ -23,10 +24,16 @@ class BotSettings(ConfigurationModelABC):
|
||||
def technicians(self) -> list[int]:
|
||||
return self._technicians
|
||||
|
||||
@property
|
||||
def deploy_file_path(self) -> str:
|
||||
return self._deploy_file_path
|
||||
|
||||
def from_dict(self, settings: dict):
|
||||
try:
|
||||
self._technicians = settings["Technicians"]
|
||||
self._deploy_file_path = settings["DeployFilesPath"]
|
||||
settings.pop("Technicians")
|
||||
settings.pop("DeployFilesPath")
|
||||
servers = List(ServerSettings)
|
||||
for s in settings:
|
||||
st = ServerSettings()
|
||||
|
@ -96,7 +96,7 @@ class MessageService(MessageServiceABC):
|
||||
self._logger.error(__name__, f'Send message to channel {ctx.channel.id} failed', e)
|
||||
else:
|
||||
self._logger.info(__name__, f'Sent message to channel {ctx.channel.id}')
|
||||
if not without_tracking:
|
||||
if not without_tracking and ctx.guild is not None:
|
||||
self._clients.append_sent_message_count(self._bot.user.id, ctx.guild.id, 1)
|
||||
self._db.save_changes()
|
||||
|
||||
|
86
src/modules/admin/command/deploy_command.py
Normal file
86
src/modules/admin/command/deploy_command.py
Normal file
@ -0,0 +1,86 @@
|
||||
import os
|
||||
import shutil
|
||||
import zipfile
|
||||
from io import BytesIO
|
||||
|
||||
import requests
|
||||
from cpl_core.configuration import ConfigurationABC
|
||||
from cpl_core.logging import LoggerABC
|
||||
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 modules.permission.abc.permission_service_abc import PermissionServiceABC
|
||||
|
||||
|
||||
class DeployCommand(DiscordCommandABC):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
logger: LoggerABC,
|
||||
config: ConfigurationABC,
|
||||
message_service: MessageServiceABC,
|
||||
bot: DiscordBotServiceABC,
|
||||
client_utils: ClientUtilsServiceABC,
|
||||
translate: TranslatePipe,
|
||||
bot_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._bot_settings = bot_settings
|
||||
|
||||
self._logger.trace(__name__, f'Loaded command service: {type(self).__name__}')
|
||||
|
||||
@commands.command()
|
||||
@commands.dm_only()
|
||||
async def deploy(self, ctx: Context, old_version: str):
|
||||
self._logger.debug(__name__, f'Received command deploy {ctx}')
|
||||
|
||||
if ctx.author.id not in self._bot_settings.technicians:
|
||||
await self._message_service.send_ctx_msg(ctx, self._t.transform('common.no_permission_message'))
|
||||
self._logger.trace(__name__, f'Finished deploy command')
|
||||
return
|
||||
|
||||
if len(ctx.message.attachments) > 1:
|
||||
raise IndexError(self._t.transform('common.errors.too_many_arguments'))
|
||||
|
||||
if len(ctx.message.attachments) < 1:
|
||||
raise IndexError(self._t.transform('common.errors.missing_required_argument'))
|
||||
|
||||
try:
|
||||
attachment_url = ctx.message.attachments[0].url
|
||||
file_request = requests.get(attachment_url)
|
||||
except Exception as e:
|
||||
self._logger.error(__name__, f'An error occurred downloading the zip file', e)
|
||||
await self._message_service.send_ctx_msg(ctx, self._t.transform('common.command_error'))
|
||||
return
|
||||
|
||||
try:
|
||||
self._logger.debug(__name__, f'Deploying files to {self._bot_settings.deploy_file_path}/latest')
|
||||
deploy_path = f'{self._bot_settings.deploy_file_path}/latest'
|
||||
deploy_old_path = f'{self._bot_settings.deploy_file_path}/{old_version}'
|
||||
|
||||
if os.path.exists(deploy_path):
|
||||
self._logger.debug(__name__, f'Moving files of {self._bot_settings.deploy_file_path}/latest to {self._bot_settings.deploy_file_path}/{old_version}')
|
||||
shutil.move(deploy_path, deploy_old_path)
|
||||
os.makedirs(deploy_path)
|
||||
|
||||
file = zipfile.ZipFile(BytesIO(file_request.content))
|
||||
file.extractall(deploy_path)
|
||||
except Exception as e:
|
||||
self._logger.error(__name__, f'An error occurred extracting the zip file', e)
|
||||
await self._message_service.send_ctx_msg(ctx, self._t.transform('common.command_error'))
|
||||
return
|
||||
|
||||
self._logger.trace(__name__, f'Finished deploy command')
|
@ -1,4 +1,5 @@
|
||||
import datetime
|
||||
import traceback
|
||||
import uuid
|
||||
|
||||
from cpl_core.time import TimeFormatSettings
|
||||
|
Loading…
Reference in New Issue
Block a user