Ebola-Chan/kd_discord_bot
1
0
Fork 0
forked from sh-edraft.de/sh_discord_bot
Code Issues Pull Requests Projects Releases Wiki Activity

Added stats edit command #46

Browse Source
This commit is contained in:
Sven Heidemann 2022-11-09 17:59:55 +01:00
parent 5c7db72723
commit 8cd67cd68c
5 changed files with 69 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
kdb-bot/src/bot/translation/de.json
View File

@ -231,6 +231,10 @@
"add": {
"failed": "Statistik kann nicht hinzugefügt werden :(",
"success": "Statistik wurde hinzugefügt :D"
},
"edit": {
"failed": "Statistik kann nicht bearbeitet werden :(",
"success": "Statistik wurde gespeichert :D"
}
}
},

7
kdb-bot/src/bot_core/service/message_service.py
View File

@ -126,12 +126,11 @@ class MessageService(MessageServiceABC):
return
self._logger.debug(__name__, f'Try to send message\t\t{message}\n\tto: {interaction.channel}')
msg = None
try:
if isinstance(message, discord.Embed):
msg = await interaction.response.send_message(embed=message)
await interaction.response.send_message(embed=message)
else:
msg = await interaction.response.send_message(message)
await interaction.response.send_message(message)
except Exception as e:
self._logger.error(__name__, f'Send message to channel {interaction.channel.id} failed', e)
else:
@ -146,4 +145,4 @@ class MessageService(MessageServiceABC):
if is_persistent:
return
await self.delete_message(msg, without_tracking)
await self.delete_message(await interaction.original_response(), without_tracking)

32
kdb-bot/src/modules/stats/command/stats_group.py
View File

@ -135,16 +135,42 @@ class StatsGroup(DiscordCommandABC):
@stats.command()
@commands.guild_only()
async def add(self, ctx: Context, name: str):
self._logger.debug(__name__, f'Received command stats list {ctx}')
self._logger.debug(__name__, f'Received command stats add {ctx}: {name}')
if not await self._client_utils.check_if_bot_is_ready_yet_and_respond(ctx):
return
if not self._permissions.is_member_technician(ctx.author):
await self._message_service.send_ctx_msg(ctx, self._t.transform('common.no_permission_message'))
self._logger.trace(__name__, f'Finished command stats list')
self._logger.trace(__name__, f'Finished command stats add')
return
form = AddStatisticForm(stats, name, self._message_service, self._logger, self._t)
self._logger.trace(__name__, f'Finished stats command')
self._logger.trace(__name__, f'Finished stats add command')
self._logger.trace(__name__, f'Started stats command form')
await ctx.interaction.response.send_modal(form)
@stats.command()
@commands.guild_only()
async def edit(self, ctx: Context, name: str):
self._logger.debug(__name__, f'Received command stats edit {ctx}: {name}')
if not await self._client_utils.check_if_bot_is_ready_yet_and_respond(ctx):
return
if not self._permissions.is_member_technician(ctx.author):
await self._message_service.send_ctx_msg(ctx, self._t.transform('common.no_permission_message'))
self._logger.trace(__name__, f'Finished command stats edit')
return
try:
statistic = stats.where(lambda s: s.name == name).single()
form = AddStatisticForm(stats, name, self._message_service, self._logger, self._t, code=statistic.code, description=statistic.description)
self._logger.trace(__name__, f'Finished stats edit command')
self._logger.trace(__name__, f'Started stats command form')
await ctx.interaction.response.send_modal(form)
except Exception as e:
self._logger.error(__name__, f'Cannot view statistic {name}', e)
await self._message_service.send_ctx_msg(ctx, self._t.transform('modules.stats.edit.failed'))
@edit.autocomplete('name')
async def edit_autocomplete(self, interaction: discord.Interaction, current: str) -> TList[app_commands.Choice[str]]:
return [app_commands.Choice(name=f'{statistic.name}: {statistic.description}', value=statistic.name) for statistic in stats]

8
kdb-bot/src/modules/stats/model/statistic.py
View File

@ -16,6 +16,14 @@ class Statistic:
def description(self) -> str:
return self._description
@description.setter
def description(self, value: str):
self._description = value
@property
def code(self) -> str:
return self._code
@code.setter
def code(self, value: str):
self._code = value

23
kdb-bot/src/modules/stats/ui/add_statistic_form.py
View File

@ -20,6 +20,8 @@ class AddStatisticForm(ui.Modal):
message_service: MessageServiceABC,
logger: CommandLogger,
t: TranslatePipe,
code: str = None,
description: str = None,
):
ui.Modal.__init__(self, title=name)
@ -29,7 +31,28 @@ class AddStatisticForm(ui.Modal):
self._logger = logger
self._t = t
if code is not None:
self.code.default = code
if description is not None:
self.description.default = description
async def on_submit(self, interaction: discord.Interaction):
statistic = self._stats.where(lambda s: s.name == self._name).single_or_default()
try:
if statistic is None:
self._stats.append(Statistic(self._name, self.description.value, self.code.value))
await self._message_service.send_interaction_msg(interaction, self._t.transform('modules.stats.add.success'))
return
statistic.description = self.description.value
statistic.code = self.code.value
await self._message_service.send_interaction_msg(interaction, self._t.transform('modules.stats.edit.success'))
except Exception as e:
self._logger.error(__name__, f'Save statistic {self._name} failed', e)
if statistic is None:
await self._message_service.send_interaction_msg(interaction, self._t.transform('modules.stats.add.failed'))
else:
await self._message_service.send_interaction_msg(interaction, self._t.transform('modules.stats.edit.failed'))
self._logger.trace(__name__, f'Finished stats command form')