285 lines
8.0 KiB
Python
285 lines
8.0 KiB
Python
import json
|
|
import os, os.path
|
|
|
|
import requests
|
|
|
|
# =============================================================================
|
|
API_HOST = "api.smash.gg"
|
|
API_ENDPOINT = "gql/alpha"
|
|
API_SCHEME = "https"
|
|
API_URL = "{scheme}://{host}/{endpoint}".format(
|
|
scheme = API_SCHEME,
|
|
host = API_HOST,
|
|
endpoint = API_ENDPOINT,
|
|
)
|
|
|
|
# =============================================================================
|
|
# -----------------------------------------------------------------------------
|
|
class Player():
|
|
"""A Player, as registered by the smash.gg API, and their characters
|
|
choices and placement in a tournament"""
|
|
|
|
CHARACTERS = {
|
|
1272: "bowser jr",
|
|
1273: "bowser",
|
|
1275: "cloud",
|
|
1277: "daisy",
|
|
1280: "donkey kong",
|
|
1285: "falco",
|
|
1286: "fox",
|
|
1287: "ganondorf",
|
|
1290: "ice climbers",
|
|
1291: "ike",
|
|
1292: "inkling",
|
|
1293: "jigglypuff",
|
|
1294: "king dedede",
|
|
1295: "kirby",
|
|
1296: "link",
|
|
1297: "little mac",
|
|
1298: "lucario",
|
|
1299: "lucas",
|
|
1300: "lucina",
|
|
1301: "luigi",
|
|
1302: "mario",
|
|
1305: "megaman",
|
|
1307: "metaknight",
|
|
1310: "mewtwo",
|
|
1313: "ness",
|
|
1314: "olimar",
|
|
1315: "pacman",
|
|
1316: "palutena",
|
|
1317: "peach",
|
|
1318: "pichu",
|
|
1319: "pikachu",
|
|
1320: "pit",
|
|
1321: "pokemon trainer",
|
|
1323: "rob",
|
|
1324: "robin",
|
|
1325: "rosalina and luma",
|
|
1326: "roy",
|
|
1328: "samus",
|
|
1329: "sheik",
|
|
1330: "shulk",
|
|
1331: "snake",
|
|
1332: "sonic",
|
|
1333: "toon link",
|
|
1334: "villager",
|
|
1335: "wario",
|
|
1336: "wii-fit trainer",
|
|
1337: "wolf",
|
|
1338: "yoshi",
|
|
1339: "young link",
|
|
1340: "zelda",
|
|
1405: "mr game & watch",
|
|
1406: "incineroar",
|
|
1407: "king k. rool",
|
|
1408: "dark samus",
|
|
1409: "byleth",
|
|
1413: "isabelle",
|
|
1441: "piranha plant",
|
|
1453: "joker",
|
|
1532: "terry",
|
|
}
|
|
|
|
# -------------------------------------------------------------------------
|
|
def __init__(
|
|
self,
|
|
id,
|
|
prefix,
|
|
gamerTag,
|
|
placement,
|
|
seeding,
|
|
twitterHandle = None,
|
|
chars = None
|
|
):
|
|
self.id = int(id) # actually intended to store an Entrant id
|
|
self.prefix = prefix
|
|
self.gamerTag = gamerTag
|
|
self.placement = int(placement)
|
|
self.seeding = int(seeding)
|
|
self.twitterHandle = twitterHandle
|
|
|
|
if chars is None:
|
|
self.chars = {}
|
|
else:
|
|
self.chars = chars
|
|
|
|
# -------------------------------------------------------------------------
|
|
def add_character_selection(self, character, win):
|
|
try:
|
|
character = self.CHARACTERS[character]
|
|
except KeyError:
|
|
pass
|
|
|
|
try:
|
|
self.chars[character] += ( 1.01 if win else 1.00 )
|
|
# This 1.01 / 1.00 tricks should hold until the player loses 101
|
|
# matches with one character and wins 100 matches with another...
|
|
# during one tournament.
|
|
except KeyError:
|
|
self.chars[character] = ( 1.01 if win else 1.00 )
|
|
|
|
# -------------------------------------------------------------------------
|
|
def get_mains(self):
|
|
return [ cv[0] for cv in sorted(
|
|
self.chars.items(),
|
|
key = lambda cv: cv[1],
|
|
reverse = True,
|
|
) ]
|
|
|
|
# -------------------------------------------------------------------------
|
|
def conf(self):
|
|
|
|
# The char list looks like 'character1 (12.08), character2 (3.02)'
|
|
# where the number between parenthesis is the number of time the
|
|
# character was played
|
|
charslist = ", ".join(
|
|
[ "{} ({:.2f})".format(c,n) for c,n in sorted(
|
|
self.chars.items(),
|
|
key = lambda cv: cv[1],
|
|
reverse = True,
|
|
) ]
|
|
)
|
|
|
|
return """
|
|
[player {tag}]
|
|
tag: {tag}
|
|
team: {pfx}
|
|
seeding: {seed}
|
|
placement: {plc}
|
|
twitter: {twi}
|
|
characters: {charslist}""" \
|
|
.format(
|
|
tag = self.gamerTag,
|
|
pfx = self.prefix,
|
|
seed = self.seeding,
|
|
plc = self.placement,
|
|
twi = self.twitterHandle,
|
|
charslist = charslist,
|
|
)
|
|
|
|
|
|
|
|
# -----------------------------------------------------------------------------
|
|
class Tournament():
|
|
"""A Tournament, as registered by the smash.gg API"""
|
|
|
|
# -------------------------------------------------------------------------
|
|
def __init__(
|
|
self,
|
|
id,
|
|
name,
|
|
slug,
|
|
startAt,
|
|
numEntrants,
|
|
venueAddress = None,
|
|
venueName = None,
|
|
city = None,
|
|
countryCode = None,
|
|
hashtag = None,
|
|
):
|
|
self.id = id
|
|
self.name = name
|
|
self.slug = slug
|
|
self.startAt = startAt
|
|
self.numEntrants = numEntrants
|
|
self.venueAddress = venueAddress
|
|
self.venueName = venueName
|
|
self.city = city
|
|
self.countryCode = countryCode
|
|
self.hashtag = hashtag
|
|
|
|
# -------------------------------------------------------------------------
|
|
def conf(self):
|
|
return """
|
|
[Tournament]
|
|
name: {name}
|
|
date: {date}
|
|
location: {loc}
|
|
slug: {slug}
|
|
numEntrants: {nbe}""" \
|
|
.format(
|
|
id = self.id,
|
|
name = self.name,
|
|
date = self.startAt,
|
|
loc = self.venueName,
|
|
slug = self.slug,
|
|
nbe = self.numEntrants,
|
|
)
|
|
|
|
|
|
|
|
# =============================================================================
|
|
def run_query(
|
|
query_name,
|
|
variables = {},
|
|
query_dir = "queries",
|
|
query_extension = "gql",
|
|
api_url = API_URL,
|
|
log = None,
|
|
):
|
|
|
|
# Load query
|
|
query_path = os.path.join(
|
|
query_dir,
|
|
"{}.{}".format(
|
|
query_name,
|
|
query_extension,
|
|
)
|
|
)
|
|
query = ""
|
|
|
|
with open(query_path, 'r') as query_file:
|
|
query = query_file.read()
|
|
|
|
# Build payload
|
|
payload = {
|
|
"query": query,
|
|
"variables": json.dumps(variables),
|
|
}
|
|
|
|
# Build headers
|
|
headers = {
|
|
"Authorization": "Bearer {token}".format(token = args.token),
|
|
"Accept": "application/json",
|
|
"Content-Type": "application/json"
|
|
}
|
|
|
|
# Send the query
|
|
try:
|
|
log.debug("Sending query '{}' with variables:".format(query_name))
|
|
log.debug(json.dumps(variables))
|
|
except AttributeError:
|
|
pass
|
|
|
|
rv = requests.post(
|
|
API_URL,
|
|
json.dumps(payload).encode("utf-8"),
|
|
headers = headers,
|
|
proxies = {"http": args.proxy, "https": args.proxy},
|
|
)
|
|
|
|
try:
|
|
rv_json = rv.json()
|
|
except Exception as e:
|
|
log.error("HTTP request failed")
|
|
log.error(e)
|
|
log.debug(e, exc_info=True)
|
|
log.debug(rv)
|
|
|
|
# Try to return the data, or log the error and return None
|
|
try:
|
|
log.debug("query complexity : {}" \
|
|
.format(
|
|
rv_json["extensions"]["queryComplexity"],
|
|
))
|
|
return rv_json["data"]
|
|
except KeyError:
|
|
try:
|
|
log.error("GraphQL error")
|
|
log.error(rv_json)
|
|
except AttributeError:
|
|
pass
|
|
return None
|
|
|