Merge pull request '0.3 - level down (#28)' (#98) from #28 into 0.3

Reviewed-on: sh-edraft.de/kd_discord_bot#98
Reviewed-by: Ebola-Chan <nick.jungmann@gmail.com>
Closes #28
This commit is contained in:
Sven Heidemann 2022-11-09 10:03:21 +01:00
commit 054266bfa0
4 changed files with 80 additions and 35 deletions

View File

@ -8,7 +8,7 @@
"LoggingSettings": { "LoggingSettings": {
"Path": "logs/", "Path": "logs/",
"Filename": "bot.log", "Filename": "bot.log",
"ConsoleLogLevel": "TRACE", "ConsoleLogLevel": "DEBUG",
"FileLogLevel": "TRACE" "FileLogLevel": "TRACE"
}, },
"BotLoggingSettings": { "BotLoggingSettings": {
@ -76,7 +76,12 @@
}, },
"BootLog": { "BootLog": {
"910199451145076828": { "910199451145076828": {
"LoginMessageChannelId": "910199452915093588" "LoginMessageChannelId": 910199452915093588
}
},
"Level": {
"910199451145076828": {
"ChangedLevelNotificationChannelId": 910199452667637892
} }
}, },
"Permission": { "Permission": {

View File

@ -175,9 +175,6 @@
}, },
"level": { "level": {
"new_level_message": "<@{}> ist nun Level {}", "new_level_message": "<@{}> ist nun Level {}",
"seeding_started": "Levelsystem wird neu geladen.",
"seeding_failed": "Levelsystem konnte nicht neu geladen werden.",
"seeding_finished": "Levelsystem wurde erfolgreich neu geladen.",
"error": { "error": {
"nothing_found": "Keine Level Einträge gefunden.", "nothing_found": "Keine Level Einträge gefunden.",
"level_with_name_already_exists": "Ein Level mit dem Namen {} existiert bereits!", "level_with_name_already_exists": "Ein Level mit dem Namen {} existiert bereits!",
@ -191,13 +188,22 @@
"permission_int": "Berechtigungen" "permission_int": "Berechtigungen"
}, },
"create": { "create": {
"created": "Level {} mit Berechtigungen {} wurde erstellt." "created": "Level {} mit Berechtigungen {} wurde erstellt.",
"seeding_started": "Levelsystem wird neu geladen.",
"seeding_failed": "Levelsystem konnte nicht neu geladen werden.",
"seeding_finished": "Levelsystem wurde Erfolgreich neu geladen."
}, },
"remove": { "remove": {
"success": "Level {} wurde entfernt :D", "success": "Level {} wurde entfernt :D",
"error": { "error": {
"not_found": "Level {} nicht gefunden!" "not_found": "Level {} nicht gefunden!"
} }
},
"down": {
"already_first": "{} hat bereits das erste Level.",
"success": "{} wurde auf Level {} runtergesetzt :)",
"failed": "{} konnte nicht runtergesetzt werden :(",
"created": "Level {} mit Berechtigungen {} wurde erstellt."
} }
}, },
"database": {}, "database": {},

View File

@ -15,6 +15,7 @@ from bot_core.abc.message_service_abc import MessageServiceABC
from bot_core.logging.command_logger import CommandLogger from bot_core.logging.command_logger import CommandLogger
from bot_data.abc.level_repository_abc import LevelRepositoryABC from bot_data.abc.level_repository_abc import LevelRepositoryABC
from bot_data.abc.server_repository_abc import ServerRepositoryABC from bot_data.abc.server_repository_abc import ServerRepositoryABC
from bot_data.abc.user_repository_abc import UserRepositoryABC
from bot_data.model.level import Level from bot_data.model.level import Level
from modules.level.level_seeder import LevelSeeder from modules.level.level_seeder import LevelSeeder
from modules.level.service.level_service import LevelService from modules.level.service.level_service import LevelService
@ -31,9 +32,10 @@ class LevelGroup(DiscordCommandABC):
client_utils: ClientUtilsServiceABC, client_utils: ClientUtilsServiceABC,
permission_service: PermissionServiceABC, permission_service: PermissionServiceABC,
translate: TranslatePipe, translate: TranslatePipe,
db_context: DatabaseContextABC, db: DatabaseContextABC,
levels: LevelRepositoryABC, levels: LevelRepositoryABC,
servers: ServerRepositoryABC, servers: ServerRepositoryABC,
users: UserRepositoryABC,
level_service: LevelService, level_service: LevelService,
level_seeder: LevelSeeder, level_seeder: LevelSeeder,
): ):
@ -45,25 +47,16 @@ class LevelGroup(DiscordCommandABC):
self._client_utils = client_utils self._client_utils = client_utils
self._permissions = permission_service self._permissions = permission_service
self._t = translate self._t = translate
self._db_context = db_context self._db = db
self._levels = levels self._levels = levels
self._servers = servers self._servers = servers
self._users = users
self._level_service = level_service self._level_service = level_service
self._level_seeder = level_seeder self._level_seeder = level_seeder
self._logger.trace(__name__, f'Loaded command service: {type(self).__name__}') self._logger.trace(__name__, f'Loaded command service: {type(self).__name__}')
async def _seed_levels(self, channel: discord.TextChannel):
# send message to ctx.channel because send_ctx_msg resolves ctx
try:
await self._message_service.send_channel_message(channel, self._t.transform('modules.level.seeding_started'))
await self._level_seeder.seed()
await self._message_service.send_channel_message(channel, self._t.transform('modules.level.seeding_finished'))
except Exception as e:
self._logger.error(__name__, f'Level seeding failed', e)
await self._message_service.send_channel_message(channel, self._t.transform('modules.level.seeding_failed'))
@commands.hybrid_group() @commands.hybrid_group()
@commands.guild_only() @commands.guild_only()
async def level(self, ctx: Context): async def level(self, ctx: Context):
@ -115,14 +108,6 @@ class LevelGroup(DiscordCommandABC):
async def create(self, ctx: Context, name: str, color: str, min_xp: int, permissions: int): async def create(self, ctx: Context, name: str, color: str, min_xp: int, permissions: int):
self._logger.debug(__name__, f'Received command level create {ctx}') self._logger.debug(__name__, f'Received command level create {ctx}')
if not await self._client_utils.check_if_bot_is_ready_yet_and_respond(ctx):
return
if not self._permissions.is_member_admin(ctx.author):
await self._message_service.send_ctx_msg(ctx, self._t.transform('common.no_permission_message'))
self._logger.trace(__name__, f'Finished command level remove')
return
try: try:
color = hex(discord.Colour.from_str(color).value) color = hex(discord.Colour.from_str(color).value)
except Exception as e: except Exception as e:
@ -135,6 +120,9 @@ class LevelGroup(DiscordCommandABC):
self._logger.error(__name__, f'Error parsing permissions {permissions}', e) self._logger.error(__name__, f'Error parsing permissions {permissions}', e)
return return
if not await self._client_utils.check_if_bot_is_ready_yet_and_respond(ctx):
return
if ctx.guild is None: if ctx.guild is None:
return return
@ -152,13 +140,21 @@ class LevelGroup(DiscordCommandABC):
else: else:
try: try:
self._levels.add_level(level) self._levels.add_level(level)
self._db_context.save_changes() self._db.save_changes()
self._logger.info(__name__, f'Saved level {name} with color {color}, min_xp {min_xp} and permissions {permissions}') self._logger.info(__name__, f'Saved level {name} with color {color}, min_xp {min_xp} and permissions {permissions}')
except Exception as e: except Exception as e:
self._logger.error(__name__, f'Could not save level {name} with color {color}, min_xp {min_xp} and permissions {permissions}', e) self._logger.error(__name__, f'Could not save level {name} with color {color}, min_xp {min_xp} and permissions {permissions}', e)
else: else:
await self._message_service.send_ctx_msg(ctx, self._t.transform('modules.level.create.created').format(name, permissions)) await self._message_service.send_ctx_msg(ctx, self._t.transform('modules.level.create.created').format(name, permissions))
await self._seed_levels(ctx.channel)
# send message to ctx.channel because send_ctx_msg resolves ctx
try:
await self._message_service.send_channel_message(ctx.channel, self._t.transform('modules.level.create.seeding_started'))
await self._level_seeder.seed()
await self._message_service.send_channel_message(ctx.channel, self._t.transform('modules.level.create.seeding_finished'))
except Exception as e:
self._logger.error(__name__, f'Level seeding failed', e)
await self._message_service.send_channel_message(ctx.channel, self._t.transform('modules.level.create.seeding_failed'))
self._logger.trace(__name__, f'Finished command level create') self._logger.trace(__name__, f'Finished command level create')
@ -219,7 +215,7 @@ class LevelGroup(DiscordCommandABC):
try: try:
self._levels.delete_level(level_from_db) self._levels.delete_level(level_from_db)
self._db_context.save_changes() self._db.save_changes()
guild: Guild = self._bot.guilds.where(lambda g: g == ctx.guild).single() guild: Guild = self._bot.guilds.where(lambda g: g == ctx.guild).single()
role: Role = guild.roles.where(lambda r: r.name == level).single_or_default() role: Role = guild.roles.where(lambda r: r.name == level).single_or_default()
if role is not None: if role is not None:
@ -238,3 +234,41 @@ class LevelGroup(DiscordCommandABC):
server = self._servers.get_server_by_discord_id(interaction.guild.id) server = self._servers.get_server_by_discord_id(interaction.guild.id)
levels = self._levels.get_levels_by_server_id(server.server_id).select(lambda l: l.name) levels = self._levels.get_levels_by_server_id(server.server_id).select(lambda l: l.name)
return [app_commands.Choice(name=level, value=level) for level in levels] return [app_commands.Choice(name=level, value=level) for level in levels]
@level.command()
@commands.guild_only()
async def down(self, ctx: Context, member: discord.Member):
self._logger.debug(__name__, f'Received command level down {ctx} {member}')
if not await self._client_utils.check_if_bot_is_ready_yet_and_respond(ctx):
return
if not self._permissions.is_member_moderator(ctx.author):
await self._message_service.send_ctx_msg(ctx, self._t.transform('common.no_permission_message'))
self._logger.trace(__name__, f'Finished command level list')
return
if ctx.guild is None:
return
server = self._servers.get_server_by_discord_id(ctx.guild.id)
user = self._users.get_user_by_discord_id_and_server_id(member.id, server.server_id)
level = self._level_service.get_level(user)
levels = self._levels.get_levels_by_server_id(server.server_id).order_by(lambda l: l.min_xp)
if level == levels.first():
await self._message_service.send_ctx_msg(ctx, self._t.transform('modules.level.down.already_first').format(member.name))
self._logger.trace(__name__, f'Finished command level list')
return
try:
new_level = levels.where(lambda l: l.min_xp < level.min_xp).last()
user.xp = new_level.min_xp
self._users.update_user(user)
self._db.save_changes()
await self._message_service.send_ctx_msg(ctx, self._t.transform('modules.level.down.success').format(member.name, new_level.name))
await self._level_service.set_level(user)
except Exception as e:
self._logger.error(__name__, f'Cannot level down {member.name} with level {level.name}', e)
await self._message_service.send_ctx_msg(ctx, self._t.transform('modules.level.down.failed').format(member.name))
self._logger.trace(__name__, f'Finished command level down')

View File

@ -29,14 +29,14 @@ class LevelSeeder(DataSeederABC):
async def _create_level(self, level: Level, guild: Guild, server: Server): async def _create_level(self, level: Level, guild: Guild, server: Server):
level.server = server level.server = server
try: try:
if guild.roles.where(lambda r: r.name == level.name).first_or_default() is not None: if guild.roles.where(lambda r: r.name == level.name).first_or_default() is None:
return
await guild.create_role(name=level.name, colour=Colour(int(level.color, 16)), hoist=False, mentionable=True, permissions=Permissions(level.permissions)) await guild.create_role(name=level.name, colour=Colour(int(level.color, 16)), hoist=False, mentionable=True, permissions=Permissions(level.permissions))
self._logger.info(__name__, f'Created level {level.name}') self._logger.debug(__name__, f'Created role {level.name}')
if self._levels.find_levels_by_server_id(server.server_id).where(lambda l: l == level).first_or_default() is not None: levels = self._levels.find_levels_by_server_id(server.server_id)
if levels is None or levels.where(lambda l: l == level).first_or_default() is None:
self._levels.add_level(level) self._levels.add_level(level)
self._logger.debug(__name__, f'Saved level {level.name}')
except discord.errors.Forbidden as e: except discord.errors.Forbidden as e:
self._logger.error(__name__, f'Creating level failed', e) self._logger.error(__name__, f'Creating level failed', e)
level.permissions = 0 level.permissions = 0