mmr.py

import aiohttp
import discord
from mogi_objects import Player

headers = {'Content-type': 'application/json'}


class LoungeData:
    def __init__(self):
        self._data = None

    async def lounge_api_full(self):
        async with aiohttp.ClientSession() as session:
            async with session.get("https://www.mk8dx-lounge.com/api/player/list") as response:
                if response.status == 200:
                    _data_full = await response.json()
                    self._data = [player for player in _data_full['players'] if "discordId" in player]
    
    def data(self):
        return self._data

lounge_data = LoungeData()

async def mk8dx_150cc_mmr(url, members):
    base_url = url + '/api/player?'
    players = []
    timeout = aiohttp.ClientTimeout(total=10)
    try:
        async with aiohttp.ClientSession(
            timeout=timeout,
            auth=aiohttp.BasicAuth(
                "username", "password")) as session:
            for member in members:
                request_text = f"discordId={member.id}"
                request_url = base_url + request_text
                async with session.get(request_url, headers=headers) as resp:
                    if resp.status != 200:
                        players.append(None)
                        continue
                    player_data = await resp.json()
                    if 'mmr' not in player_data.keys():
                        players.append(
                            Player(member, player_data['name'], None))
                        continue
                    players.append(
                        Player(member, player_data['name'], player_data['mmr']))
    except:
        print(f"Fetch for player {members[0]} has failed.", flush=True)
    return players


async def get_mmr_from_discord_id(discord_id):
    base_url = "https://www.mk8dx-lounge.com" + '/api/player?'
    timeout = aiohttp.ClientTimeout(total=10)
    async with aiohttp.ClientSession(
        timeout=timeout,
        auth=aiohttp.BasicAuth(
            "username", "password")) as session:
        request_text = f"discordId={discord_id}"
        request_url = base_url + request_text
        async with session.get(request_url, headers=headers) as resp:
            if resp.status != 200:
                return "Player does not exist"
            player_data = await resp.json()
            if 'mmr' not in player_data.keys():
                return "Player has no mmr"
            return player_data['mmr']


async def get_mmr(config, members):
    return await mk8dx_150cc_mmr(config, members)


async def mk8dx_150cc_fc(config, name):
    base_url = config["url"] + '/api/player?'
    request_url = base_url + f'name={name}'
    timeout = aiohttp.ClientTimeout(total=15)
    async with aiohttp.ClientSession(timeout=timeout) as session:
        async with session.get(request_url, headers=headers) as resp:
            if resp.status != 200:
                return None
            player_data = await resp.json()
            if 'switchFc' not in player_data.keys():
                return None
            return player_data['switchFc']