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

Improved logic to create roles for levels by default #25

Browse Source
This commit is contained in:
Sven Heidemann 2022-11-06 19:36:15 +01:00
parent 861a847088
commit 8118d4edc3
1 changed files with 19 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
kdb-bot/src/modules/level/level_seeder.py
View File

@ -25,7 +25,7 @@ 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):
try: try:
if guild.roles.where(lambda r: r.name == level.name).count() != 0: if guild.roles.where(lambda r: r.name == level.name).count() > 0:
return 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))
@ -36,19 +36,25 @@ class LevelSeeder(DataSeederABC):
self._logger.error(__name__, f'Creating level failed', e) self._logger.error(__name__, f'Creating level failed', e)
async def seed(self): async def seed(self):
created_default = False # create levels
for guild in self._bot.guilds: for guild in self._bot.guilds:
created_default = False
if guild.roles.where(lambda r: r.name == '___ Level ___').count() == 0:
await guild.create_role(name='___ Level ___')
server = self._servers.find_server_by_discord_id(guild.id) server = self._servers.find_server_by_discord_id(guild.id)
if server is None: if server is None:
continue continue
levels = self._levels.find_levels_by_server_id(server.server_id) levels = self._levels.find_levels_by_server_id(server.server_id)
if levels is not None and levels.count() > 0: if levels is not None and levels.count() > 0:
# create levels from db
for level in levels: for level in levels:
await self._create_level(level, guild, server) await self._create_level(level, guild, server)
continue self._logger.debug(__name__, f'Seeded levels')
else:
# create default levels
for level in self._default_levels: for level in self._default_levels:
created_default = True created_default = True
await self._create_level(level, guild, server) await self._create_level(level, guild, server)
@ -56,30 +62,16 @@ class LevelSeeder(DataSeederABC):
self._logger.debug(__name__, f'Seeded default levels') self._logger.debug(__name__, f'Seeded default levels')
if created_default: if created_default:
return
for guild in self._bot.guilds:
server = self._servers.find_server_by_discord_id(guild.id)
if server is None:
continue continue
levels = self._levels.find_levels_by_server_id(server.server_id) position_above_levels = guild.roles.where(lambda r: r.name == '~~~ Level ~~~').single().position
roles: List[Role] = List(Role)
for role in guild.roles.order_by_descending(lambda r: r.position): for role in guild.roles.order_by_descending(lambda r: r.position):
if levels.where(lambda l: l.name == role.name).count() == 0: if levels.where(lambda l: l.name == role.name).count() == 0:
continue continue
roles.append(role) new_position = position_above_levels - (levels.index(levels.where(lambda l: l.name == role.name).single()) + 1)
levels = levels.order_by(lambda l: l.min_xp)
position_below_levels = roles.where(lambda r: r.name == levels.order_by(lambda l: l.min_xp).first().name).single().position - 1
for role in roles:
new_position = position_below_levels + (levels.index(levels.where(lambda l: l.name == role.name).single()) - 1)
if new_position <= 0: if new_position <= 0:
above_role: Role = roles.where(lambda r: r.position == 1).single()
await above_role.edit(position=above_role.position + 1)
new_position = 1 new_position = 1
try: try:
self._logger.debug(__name__, f'Moved {role.name} from {role.position} to {new_position}') self._logger.debug(__name__, f'Moved {role.name} from {role.position} to {new_position}')
await role.edit(position=new_position) await role.edit(position=new_position)