TimingsBot/cogs/linking_updater.py

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()
        
def setup(bot):
    bot.add_cog(Linking_updater(bot))