royalnet/overwatch.py

import asyncio
import requests
import strings as s
loop = asyncio.get_event_loop()


class NotFoundException(Exception):
    pass

# Get player data
async def get_player_data(platform: str, region: str, battletag: str, **kwargs):
    print("[Overwatch] Getting player info for: {platform} {region} {battletag}".format(platform=platform,
                                                                                        region=region,
                                                                                        battletag=battletag))
    # Unofficial API requires - for discriminator numbers
    battletag = battletag.replace("#", "-")
    # GET the json unofficial API response
    r = await loop.run_in_executor(None, requests.get,
                                   'https://api.lootbox.eu/{platform}/{region}/{battletag}/profile'.format(**locals()))
    # Ensure the request is successful
    if r.status_code == 200:
        # Parse json and check for the status code inside the response
        pj = r.json()
        if "statusCode" in pj:
            if pj["statusCode"] == 404:
                raise NotFoundException("Player not found.")
            else:
                raise Exception("Unhandled API response.")
        else:
            # Success!
            return pj
    else:
        raise Exception("Unhandled API response.")


# Convert rank to a medal
def rank_to_medal(rank):
    if int(rank) == 0:
        return s.overwatch_medal_list[0]
    if int(rank) < 1500:
        return s.overwatch_medal_list[1]
    elif int(rank) < 2000:
        return s.overwatch_medal_list[2]
    elif int(rank) < 2500:
        return s.overwatch_medal_list[3]
    elif int(rank) < 3000:
        return s.overwatch_medal_list[4]
    elif int(rank) < 3500:
        return s.overwatch_medal_list[5]
    elif int(rank) < 4000:
        return s.overwatch_medal_list[6]
    else:
        # TODO: what
        raise Exception("Mi sfugge qualcosa.")


# Convert an url to a medal
def url_to_medal(rank: str):
    if rank == "https://blzgdapipro-a.akamaihd.net/game/rank-icons/season-2/rank-1.png":
        return s.overwatch_medal_list[0]
    elif rank == "https://blzgdapipro-a.akamaihd.net/game/rank-icons/season-2/rank-2.png":
        return s.overwatch_medal_list[1]
    elif rank == "https://blzgdapipro-a.akamaihd.net/game/rank-icons/season-2/rank-3.png":
        return s.overwatch_medal_list[2]
    elif rank == "https://blzgdapipro-a.akamaihd.net/game/rank-icons/season-2/rank-4.png":
        return s.overwatch_medal_list[3]
    elif rank == "https://blzgdapipro-a.akamaihd.net/game/rank-icons/season-2/rank-5.png":
        return s.overwatch_medal_list[4]
    elif rank == "https://blzgdapipro-a.akamaihd.net/game/rank-icons/season-2/rank-6.png":
        return s.overwatch_medal_list[5]
    elif rank == "https://blzgdapipro-a.akamaihd.net/game/rank-icons/season-2/rank-7.png":
        return s.overwatch_medal_list[6]
    else:
        raise NotFoundException("The medal does not exist.")