sh-edraft.de/sh_discord_bot
4
1
Fork 1
Code Issues 21 Pull Requests Actions Packages Projects 3 Releases 27 Wiki Activity

Fixed level assignment #25

Browse Source
This commit is contained in:
Sven Heidemann 2022-11-06 21:54:21 +01:00
parent 164671a07c
commit 6e39154c75
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
kdb-bot/src/modules/level/service/level_service.py
View File

@ -1,3 +1,5 @@
import asyncio
import discord import discord
from cpl_core.database.context import DatabaseContextABC from cpl_core.database.context import DatabaseContextABC
from cpl_core.logging import LoggerABC from cpl_core.logging import LoggerABC
@ -31,12 +33,11 @@ class LevelService:
def get_level(self, user: User) -> Level: def get_level(self, user: User) -> Level:
levels = self._levels.get_levels_by_server_id(user.server.server_id).order_by(lambda l: l.min_xp) levels = self._levels.get_levels_by_server_id(user.server.server_id).order_by(lambda l: l.min_xp)
return levels.where(lambda l: user.xp >= l.min_xp).first() return levels.where(lambda l: user.xp >= l.min_xp).last()
async def set_level(self, user: User): async def set_level(self, user: User):
level_names = self._levels.get_levels_by_server_id(user.server.server_id).select(lambda l: l.name) level_names = self._levels.get_levels_by_server_id(user.server.server_id).select(lambda l: l.name)
guild: Guild = self._bot.guilds.where(lambda g: g.id == user.server.discord_server_id).single() guild: Guild = self._bot.guilds.where(lambda g: g.id == user.server.discord_server_id).single()
level_role: Role = guild.roles.where(lambda r: r.name == self.get_level(user).name).single()
member: Member = guild.members.where(lambda m: m.id == user.discord_id).single() member: Member = guild.members.where(lambda m: m.id == user.discord_id).single()
for role in member.roles: for role in member.roles:
if role.name not in level_names.to_list(): if role.name not in level_names.to_list():
@ -49,6 +50,7 @@ class LevelService:
except Exception as e: except Exception as e:
self._logger.error(__name__, f'Removing role {role.name} from {member.name} failed!', e) self._logger.error(__name__, f'Removing role {role.name} from {member.name} failed!', e)
level_role: Role = guild.roles.where(lambda r: r.name == self.get_level(user).name).single()
if level_role in member.roles: if level_role in member.roles:
return return
try: try: