import os import discord import requests import json import logging from discord.ext import commands, tasks from discord.ext.commands import has_permissions, MissingPermissions from dotenv import load_dotenv import aiohttp import asyncio class Linking_updater(commands.Cog): def __init__(self, bot): self.bot = bot self.guild_id = int(os.getenv('guild_id')) self.crabwings_role_id = int(os.getenv('crabwings_role_id')) self.duckfeet_role_id = int(os.getenv('duckfeet_role_id')) self.elktail_role_id = int(os.getenv('elktail_role_id')) self.client_role_id = int(os.getenv('client_role_id')) self.subuser_role_id = int(os.getenv('subuser_role_id')) self.verified_role_id = int(os.getenv('verified_role_id')) @tasks.loop(minutes=10) async def linking_updater(self): logging.info("Synchronizing roles") file = open('users.json', 'r') data = json.load(file) file.close() guild = self.bot.get_guild(guild_id) i = -1 print("1") for client in data['users']: print("2") i += 1 member = guild.get_member(client['discord_id']) if member: print("3") 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: print("4") if response.status == 200: print("5") # 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=self.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=self.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=self.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=self.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=self.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']) @linking_updater.before_loop async def before_linking_updater(self): logging.info('waiting to enter loop') await bot.wait_until_ready()