Update linking_updater.py

cogs/linking_updater.py

@@ -1,23 +1,121 @@
+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'))
+        linking_updater.start()
 
     @tasks.loop(seconds=10)
     async def linking_updater(self):
-        print("Hello2")
-        
+        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):
-        print("Hello")
+        logging.info('waiting to enter loop')
         await self.bot.wait_until_ready()
         
 def setup(bot):
     bot.add_cog(Linking_updater(bot))
-
-linking_updater.start()