pull some data from API, fix some datatype compats

Former-commit-id: 03ec0dd8fd

src/cli/pull.py

@@ -0,0 +1,66 @@
+import requests 
+import json
+from exceptions import APIError
+
+def load_config(path):
+  with open(path, "r") as f:
+    return json.load(f)
+
+url = "https://titanscouting.epochml.org"
+config_tra = load_config("config.json")
+trakey = config_tra['persistent']['key']['tra']
+
+def get_team_competition():
+  endpoint = '/api/fetchTeamCompetition'
+  params = {
+    "CLIENT_ID": trakey['CLIENT_ID'],
+    "CLIENT_SECRET": trakey['CLIENT_SECRET']
+  }
+  response = requests.request("GET", url + endpoint, params=params)
+  json = response.json()
+  if json['success']:
+    return json['competition']
+  else:
+    raise APIError(json, endpoint)
+
+def get_team():
+  endpoint = '/api/fetchTeamCompetition'
+  params = {
+    "CLIENT_ID": trakey['CLIENT_ID'],
+    "CLIENT_SECRET": trakey['CLIENT_SECRET']
+  }
+  response = requests.request("GET", url + endpoint, params=params)
+  json = response.json()
+  if json['success']:
+    return json['team']
+  else:
+    raise APIError(json, endpoint)
+
+def get_team_match_data(competition, team_num):
+  endpoint = '/api/fetchAllTeamMatchData'
+  params = {
+    "competition": competition,
+    "teamScouted": team_num,
+    "CLIENT_ID": trakey['CLIENT_ID'],
+    "CLIENT_SECRET": trakey['CLIENT_SECRET']
+  }
+  response = requests.request("GET", url + endpoint, params=params)
+  json = response.json()
+  if json['success']:
+    return json['data'][team_num]
+  else:
+    raise APIError(json, endpoint)
+
+def get_teams_at_competition(competition):
+    endpoint = '/api/fetchAllTeamNicknamesAtCompetition'
+    params = {
+      "competition": competition,
+      "CLIENT_ID": trakey['CLIENT_ID'],
+      "CLIENT_SECRET": trakey['CLIENT_SECRET']
+    }
+    response = requests.request("GET", url + endpoint, params=params)
+    json = response.json()
+    if json['success']:
+      return list(json['data'].keys())
+    else:
+      raise APIError(json, endpoint)