Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class LabClientCheck(commands.Bot):
- def __init__(self, *args, **kwargs):
- kwargs["command_prefix"] = "!"
- self.role_id = None if "role_id" not in kwargs else kwargs["role_id"]
- self.server_id = None if "server_id" not in kwargs else kwargs["server_id"]
- super().__init__(*args, **kwargs)
- async def on_ready(self):
- self.bg_task = self.loop.create_task(self.labtask())
- async def labtask(self):
- await self.wait_until_ready()
- global config, api
- print(self.server_id)
- guild = self.get_guild(int(self.server_id))
- print(guild)
- role = guild.get_role(int(self.role_id))
- if isinstance(guild, discord.Guild):
- role = guild.get_role(int(self.role_id))
- if role is None:
- print("Could not retrieve role information.")
- sys.exit(255)
- print(role)
- for key in config:
- user = config[key]
- id = key
- status = api.get_user_online(user)
- print(user, id, status)
- if isinstance(role, discord.Role):
- print("Retrieving guild info?")
- #member = guild.get_member(id)
- member = discord.utils.get(guild.members, id=int(id))
- role = discord.utils.get(guild.roles, id=int(self.role_id))
- print("Check?", member)
- if status:
- print("Removed " + user + " from role.")
- await member.remove_roles([role], "Auto: Detected login from lab.")
- else:
- print("Test")
- res = await member.add_roles(role, "Auto: Detected logout from lab.", atomic=True)
- print("Added " + user + " to role.")
- #else:
- # print(role, member)
- sys.exit(0)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement