From d8810b1b25ed56c587c714b7521249beef0d9337 Mon Sep 17 00:00:00 2001 From: Purpur <43528123+Pemigrade@users.noreply.github.com> Date: Sun, 10 Jan 2021 06:15:50 -0600 Subject: [PATCH] Update linking_updater.py --- cogs/linking_updater.py | 111 +++------------------------------------- 1 file changed, 7 insertions(+), 104 deletions(-) diff --git a/cogs/linking_updater.py b/cogs/linking_updater.py index 42a90b7..e3cde37 100644 --- a/cogs/linking_updater.py +++ b/cogs/linking_updater.py @@ -1,120 +1,23 @@ -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(seconds=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']) - + print("Hello2") + @linking_updater.before_loop async def before_linking_updater(self): - logging.info('waiting to enter loop') + print("Hello") await self.bot.wait_until_ready() def setup(bot): bot.add_cog(Linking_updater(bot)) + +linking_updater.start()