demotion.py 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. import discord
  2. from discord.ext import commands
  3. from discord.commands import Option
  4. from discord.commands import slash_command
  5. import configparser
  6. class demotion(commands.Cog):
  7. def __init__(self, bot: discord.Bot):
  8. self.bot = bot
  9. #Command initialization
  10. @slash_command(name="demotion", description= "Demote a department member to the previous rank")
  11. async def demotion(
  12. self,
  13. ctx,
  14. user: str = Option(discord.User, "Select User", required=True),
  15. ):
  16. if user not in ctx.guild.members:
  17. await ctx.respond("The selected user is not a member on this Server!", ephemeral=True)
  18. return
  19. elif user == self.bot.user:
  20. await ctx.respond(f"This is me - the {self.bot.user}", ephemeral=True)
  21. return
  22. elif user == ctx.author:
  23. await ctx.respond("You cannot promote yourself!", ephemeral=True)
  24. return
  25. server = ctx.guild
  26. config = configparser.RawConfigParser()
  27. configFilePath = r'config.cfg'
  28. config.read_file(open(configFilePath))
  29. #Role Configuration
  30. department1_ranks_ids = config.get('Role Management', 'department1_ranks').split(', ')
  31. department1_ranks = [ctx.guild.get_role(int(role_id)) for role_id in department1_ranks_ids]
  32. department1_supervisor_role_id = int(config.get('Einweisung', 'department1_supervisor_id'))
  33. department1_supervisor_role = server.get_role(department1_supervisor_role_id)
  34. #department2_ranks_ids = config.get('Role Management', 'department2_ranks').split(', ')
  35. #department2_ranks = [ctx.guild.get_role(int(role_id)) for role_id in department2_ranks_ids]
  36. #department2_supervisor_role_id = int(config.get('Einweisung', 'department2_supervisor_id'))
  37. #department2_supervisor_role = server.get_role(department2_supervisor_role_id)
  38. #Command implementation
  39. if ctx.author.roles.contains(department1_supervisor_role):
  40. ranks = department1_ranks
  41. #elif ctx.author.roles.contains(department2_supervisor_role):
  42. #ranks = department2_ranks
  43. else:
  44. await ctx.respond("You do not have permission to use this command!", ephemeral=True)
  45. return
  46. user_roles = user.roles
  47. user_rank = None
  48. for rank in ranks:
  49. if rank in user_roles:
  50. user_rank = rank
  51. break
  52. if user_rank is None:
  53. await ctx.respond("The selected user does not have a rank that can be demoted!", ephemeral=True)
  54. return
  55. current_rank_index = ranks.index(user_rank)
  56. if current_rank_index == 0:
  57. await ctx.respond("The selected user is already at the lowest rank!", ephemeral=True)
  58. return
  59. new_rank = ranks[current_rank_index - 1]
  60. await user.remove_roles(user_rank)
  61. await user.add_roles(new_rank)
  62. await ctx.respond(f"{user.mention} has been demoted to {new_rank.name}!", ephemeral=True)
  63. def setup(bot: discord.Bot):
  64. bot.add_cog(demotion(bot))