mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 11:34:18 +00:00
59 lines
No EOL
2.1 KiB
Python
59 lines
No EOL
2.1 KiB
Python
import asyncio
|
|
import aiohttp
|
|
import royalbotconfig
|
|
|
|
# https://euw.api.riotgames.com/api/lol/EUW/v1.4/summoner/52348350?api_key=RGAPI-1008c33d-b0a4-4091-8600-27022d570964
|
|
|
|
class LoLAPIError(Exception):
|
|
def __init__(self, status_code, text):
|
|
self.status_code = status_code
|
|
self.text = text
|
|
|
|
|
|
tiers = ["BRONZE", "SILVER", "GOLD", "PLATINUM", "DIAMOND", "MASTER", "CHALLENGER"]
|
|
|
|
|
|
divisions = ["I", "II", "III", "IV", "V"]
|
|
|
|
|
|
async def get_json(region, endpoint, **kwargs):
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(f"https://{region.lower()}.api.riotgames.com/api/lol/{region.upper()}{endpoint}", **kwargs) as response:
|
|
json = await response.json()
|
|
if response.status != 200:
|
|
raise LoLAPIError(response.status, f"Riot API returned {response.status}")
|
|
return json
|
|
|
|
|
|
async def get_summoner_data(region: str, summoner_id=None, summoner_name=None):
|
|
# Check for the number of arguments
|
|
if bool(summoner_id) == bool(summoner_name):
|
|
# TODO: use the correct exception
|
|
raise Exception("Invalid number of arguments specified")
|
|
params = {
|
|
"api_key": royalbotconfig.lol_token
|
|
}
|
|
if summoner_id is not None:
|
|
data = await get_json("euw", f"/v1.4/summoner/{summoner_id}", params=params)
|
|
return data[str(summoner_id)]
|
|
elif summoner_name is not None:
|
|
data = await get_json("euw", f"/v1.4/summoner/by-name/{summoner_name}", params=params)
|
|
return data[summoner_name.lower().replace(" ", "")]
|
|
|
|
|
|
async def get_rank_data(region: str, summoner_id: int):
|
|
params = {
|
|
"api_key": royalbotconfig.lol_token
|
|
}
|
|
data = await get_json("euw", f"/v2.5/league/by-summoner/{summoner_id}/entry", params=params)
|
|
soloq = None
|
|
flexq = None
|
|
ttq = None
|
|
for entry in data[str(summoner_id)]:
|
|
if entry["queue"] == "RANKED_SOLO_5x5":
|
|
soloq = entry
|
|
elif entry["queue"] == "RANKED_FLEX_SR":
|
|
flexq = entry
|
|
elif entry["queue"] == "RANKED_FLEX_TT":
|
|
ttq = entry
|
|
return soloq, flexq, ttq |