einweisung.py 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293
  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 einweisung(commands.Cog):
  7. def __init__(self, bot: discord.Bot):
  8. self.bot = bot
  9. #Command initialization
  10. @slash_command(name="einweisung", description="Freischalten eines User nach der Einweisung")
  11. async def einweisung(
  12. self,
  13. ctx,
  14. user: discord.User = Option(discord.User, "Select User"),
  15. ):
  16. if user is None:
  17. user = ctx.author
  18. elif user not in ctx.guild.members:
  19. await ctx.respond("The selected user is not a member on this Server!", ephemeral=True)
  20. return
  21. elif user == self.bot.user:
  22. await ctx.respond(f"This is me - the {self.bot.user}", ephemeral=True)
  23. return
  24. server = ctx.guild
  25. config = configparser.RawConfigParser()
  26. configFilePath = r'config.cfg'
  27. config.read_file(open(configFilePath))
  28. #Role Configuration
  29. einweisung_role_id = int(config.get('Einweisung', 'einweisung_role_id'))
  30. einweisung_role = server.get_role(einweisung_role_id)
  31. acces_role_id = int(config.get('Einweisung', 'acces_role_id'))
  32. acces_role = server.get_role(acces_role_id)
  33. department_label_role_id = int(config.get('Einweisung', 'department_label_role_id'))
  34. department_label_role = server.get_role(department_label_role_id)
  35. department1_label_id = int(config.get('Einweisung', 'department1_label_id'))
  36. department1_label = server.get_role(department1_label_id)
  37. department1_supervisor_id = int(config.get('Einweisung', 'department1_supervisor_id'))
  38. department1_role_id = int(config.get('Einweisung', 'department1_role_id'))
  39. department1_role = server.get_role(department1_role_id)
  40. department1_supervisor = server.get_role(department1_supervisor_id)
  41. department1_deputy_id = int(config.get('Einweisung', 'department1_deputy_id'))
  42. department1_deputy = server.get_role(department1_deputy_id)
  43. #department2_supervisor_id = int(config.get('Einweisung', 'department2_supervisor_id'))
  44. #department2_role_id = int(config.get('Einweisung', 'department2_role_id'))
  45. #department2_role = server.get_role(department2_role_id)
  46. #department2_supervisor = server.get_role(department2_supervisor_id)
  47. #department2_deputy_id = int(config.get('Einweisung', 'department2_deputy_id'))
  48. #department2_deputy = server.get_role(department2_deputy_id)
  49. #Command implemetation
  50. if einweisung_role is None or department1_role is None or department1_supervisor is None or acces_role is None or department1_label is None or department1_deputy is None or department_label_role is None:
  51. await ctx.respond("One or more roles are not configured properly!", ephemeral=True)
  52. return
  53. if department1_supervisor in ctx.author.roles:
  54. n = 1
  55. #elif department2_supervisor in ctx.author.roles:
  56. #n = 2
  57. else:
  58. await ctx.respond("You don't have permission to use this command!", ephemeral=True)
  59. return
  60. try:
  61. if n == 1:
  62. await user.add_roles(einweisung_role, department_label_role, department1_role, department1_deputy, department1_label, acces_role)
  63. #elif n == 2:
  64. # await user.add_roles(einweisung_role, department2_role, department2_deputy, department_label_role, acces_role)
  65. except discord.Forbidden:
  66. await ctx.respond("I don't have permission to assign roles!", ephemeral=True)
  67. return
  68. embed = discord.Embed(title="Einweisung abgeschlossen", color=0x00ff00)
  69. embed.add_field(name="User Freigeschaltet:", value=f"{user.mention} wurde erfolgreich freigeschaltet.", inline=False)
  70. await ctx.respond(embed=embed, ephemeral=True)
  71. def setup(bot: discord.Bot):
  72. bot.add_cog(einweisung(bot))