From 30f5d04effcef78b9560844af588219851df9b5d Mon Sep 17 00:00:00 2001 From: Purpur <43528123+Pemigrade@users.noreply.github.com> Date: Sun, 10 Jan 2021 06:23:06 -0600 Subject: [PATCH] Update bot.py --- bot.py | 86 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++- 1 file changed, 85 insertions(+), 1 deletion(-) diff --git a/bot.py b/bot.py index e3036cd..f9883dd 100644 --- a/bot.py +++ b/bot.py @@ -76,7 +76,6 @@ async def on_message(message): await timings.analyze_timings(message) await bot.process_commands(message) - @bot.event async def on_raw_reaction_add(payload): global verification_message @@ -134,6 +133,89 @@ async def react(ctx, url, reaction): @tasks.loop(minutes=10) async def updater(): + logging.info("Synchronizing roles") + file = open('users.json', 'r') + data = json.load(file) + file.close() + guild = bot.get_guild(guild_id) + i = -1 + for client in data['users']: + i += 1 + member = guild.get_member(client['discord_id']) + if member: + api_key = client['client_api_key'] + url = "https://panel.birdflop.com/api/client" + headers = { + 'Accept': 'application/json', + 'Authorization': 'Bearer ' + api_key, + } + async with aiohttp.ClientSession() as session: + async with session.get(url, headers=headers) as response: + if response.status == 200: + # Formats response for servers in JSON format + servers_json_response = await response.json() + + user_client = False + user_subuser = False + user_crabwings = False + user_duckfeet = False + user_elktail = False + for server in servers_json_response['data']: + server_owner = server['attributes']['server_owner'] + if server_owner == True: + user_client = True + elif server_owner == False: + user_subuser = True + server_node = server['attributes']['node'] + if server_node == "Crabwings - NYC": + user_crabwings = True + elif server_node == "Duckfeet - EU": + user_duckfeet = True + elif server_node == "Elktail - EU": + user_elktail = True + role = discord.utils.get(guild.roles, id=client_role_id) + if user_client == True: + await member.add_roles(role) + else: + await member.remove_roles(role) + role = discord.utils.get(guild.roles, id=subuser_role_id) + if user_subuser == True: + await member.add_roles(role) + else: + await member.remove_roles(role) + role = discord.utils.get(guild.roles, id=crabwings_role_id) + if user_crabwings == True: + await member.add_roles(role) + else: + await member.remove_roles(role) + role = discord.utils.get(guild.roles, id=duckfeet_role_id) + if user_duckfeet == True: + await member.add_roles(role) + else: + await member.remove_roles(role) + role = discord.utils.get(guild.roles, id=elktail_role_id) + if user_elktail == True: + await member.add_roles(role) + else: + await member.remove_roles(role) + else: + data['users'].pop(i) + json_dumps = json.dumps(data, indent=2) + file = open('users.json', 'w') + file.write(json_dumps) + file.close() + await member.edit(roles=[]) + logging.info("removed discord_id " + str(client['discord_id']) + " with client_id " + str( + client['client_id']) + " and INVALID client_api_key " + client['client_api_key']) + else: + data['users'].pop(i) + json_dumps = json.dumps(data, indent=2) + file = open('users.json', 'w') + file.write(json_dumps) + file.close() + logging.info("removed discord_id " + str(client['discord_id']) + " with client_id " + str( + client['client_id']) + " and client_api_key " + client['client_api_key']) + # Update backups logging.info('Ensuring backups') url = "https://panel.birdflop.com/api/application/servers" @@ -264,6 +346,8 @@ for file_name in os.listdir('./cogs'): bot.load_extension(f'cogs.{file_name[:-3]}') updater.start() +linking_updater = bot.get_cog('Linking_updater') +await linking_updater.linking_updater.start() bot.run(token) # full name: message.author.name + "#" + str(message.author.discriminator) + " (" + str(message.author.id) + ")"