Advertisement
Guest User

Untitled

a guest
Jan 22nd, 2019
81
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 2.32 KB | None | 0 0
  1.  
  2. async def doJail(member):
  3.     with open(("data/jails/" + str(member.id) + ".txt"), "w") as jailFile:
  4.         rIDs = []
  5.         for role in member.roles:
  6.             if not (str(role.name).upper()) == str("@EVERYONE"):
  7.                 rIDs.append(str(role.id))
  8.  
  9.         for i in range(2):
  10.             # Make sure no roles are missed
  11.             for role in member.roles:
  12.                 print("Removed " + str(role.name) + " from " + str(member.name))
  13.                 try:
  14.                     await bot.remove_roles(member, role) # Actually take the fuckers roles
  15.                 except Exception:
  16.                     pass
  17.  
  18.         if not len(rIDs) == 0:
  19.             print(str(member.name) + " has RoleIDs : ")
  20.             print(rIDs)
  21.             jailFile.write(str("|".join(rIDs)))
  22.         else:
  23.             print(str(member.name) + " has no roles")
  24.             jailFile.write("NONE")
  25.            
  26.     await bot.add_roles(member, (discord.utils.get(member.server.roles, name=config.jailRoleName)))
  27.     return
  28.  
  29. async def doUnJail(member):
  30.     with open(("data/jails/" + str(member.id) + ".txt"), "r") as jailReadFile:
  31.         rIDs = []
  32.         content = str((jailReadFile.read()).replace("\n", ""))
  33.         rIDs = content.split("|")
  34.        
  35.     if content == "NONE":
  36.         await bot.remove_roles(member, (discord.utils.get(member.server.roles, name=str(config.jailRoleName))))
  37.         os.remove("data/jails/" + str(member.id) + ".txt")
  38.         return
  39.     else:
  40.         # Make sure no roles are missed
  41.         for i in range(len(rIDs)):
  42.             roleID = str(rIDs[i])
  43.             # Get correct role
  44.             role = discord.utils.get(member.server.roles, id=str(roleID))
  45.             print(str(role.name) + " was added to " + str(member.name))
  46.             try:
  47.                 await bot.add_roles(member, role)
  48.             except Exception:
  49.                 pass
  50.         for i in range(2):
  51.             # Actually remove this
  52.             await bot.remove_roles(member, (discord.utils.get(member.server.roles, name=config.jailRoleName)))
  53.         for i in range(2):
  54.             # MAKE SURE MEMBERS THERE AS ITS ESSENTIAL
  55.             await bot.add_roles(member, (discord.utils.get(member.server.roles, name=config.joinMemberRoleName)))
  56.         os.remove("data/jails/" + str(member.id) + ".txt")
  57.         return
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement