Chief Delphi - View Single Post

sdangelo · #6 05-05-2016, 06:35 PM

Quote:

Originally Posted by Nathan Streeter

It would be interesting to see which CMP would have more teams from the 2016 CMP elims...

Spoiler for Code:

Code:

import json
import urllib2

BASE_URL = 'http://www.thebluealliance.com/api/v2/'
APP_HEADER = 'X-TBA-App-Id'
APP_ID = 'sdangelo:splitchamps:v0.1'

event_codes = ['arc', 'cars', 'carv', 'cur', 'gal', 'hop', 'new', 'tes']

def fetch_endpoint(endpoint):
    #Checks Blue Alliance for data
    full_url = BASE_URL + endpoint
    url = urllib2.Request(full_url, headers={APP_HEADER: APP_ID, 'User-agent': 'Mozilla/5.0'})
    response = urllib2.urlopen(url)
    return json.loads(response.read())

def get_teams(event_key):
    #builds list of all teams
    teams = []
    event = fetch_endpoint("event/2016"+ str(event_key))
    for alliance in event["alliances"]:
        for team in alliance["picks"]:
            teams.append(str(team))
    return teams

def get_locations(teams_list):
    stl = []
    hous = []
    allTeams = []
    for team in teams_list:
        data = fetch_endpoint("team/" + team)
        if data["country_name"] == "USA":
            s = data["region"]
            if s == "Maine" or s == "Vermont" or s == "New Hampshire" or s == "Massachusetts" or\
               s == "Connecticut" or s == "Rhode Island" or s == "New York" or s == "Pennsylvania"\
               or s == "New Jersey" or s == "Delaware" or s == "Maryland" or s == "District of Columbia"\
               or s == "Virginia" or s == "West Virginia" or s == "Kentucky" or s == "Ohio" or\
               s == "Michigan" or s == "Indiana" or s == "Illinois" or s == "Wisconsin" or\
               s == "Minnesota" or s == "Iowa" or s == "Missouri" or s == "Kansas" or\
               s == "Nebraska" or s == "South Dakota" or s == "North Dakota":
                stl.append(data["team_number"])
            else:
                hous.append(data["team_number"])
        elif data["country_name"] == "Canada":
            p = data["region"]
            if p == "Ontario" or p == "Quebec" or p == "Newfoundland and Laborador"\
               or p == "Manitoba" or p == "Nunavit":
                stl.append(data["team_number"])
            else:
                hous.append(data["team_number"])
        else:
            c = data["country_name"]
            if c == "Mexico" or c == "Haiti" or c == "Dominican Republic" or\
               c == "Ecuador" or c == "Columbia" or c == "Peru" or c == "Chile"\
               or c == "Brazil" or c == "South Africa" or c == "Israel" or\
               c == "Egypt" or c == "Jordan" or c == "Lebanon" or c == "Turkey"\
               or c == "Kuwait" or c == "United Arab Emirates" or c == "Singapore"\
               or c == "Malaysia" or c == "Australia" or c == "New Zealand" or\
               c == "Philippines" or c == "China":
                hous.append(data["team_number"])
            else:
                stl.append(data["team_number"])
    allTeams.append(stl)
    allTeams.append(hous)
    return allTeams

if __name__ == "__main__":
    stLouis = []
    houston = []
    for event in event_codes:
        teamsList = get_teams(event)
        locList = get_locations(teamsList)
        for team in locList[0]:
            stLouis.append(team)
        for team in locList[1]:
            houston.append(team)
    print "Number of elims teams in St. Louis: " + str(len(stLouis))
    print "Number of elims teams in Houston: " + str(len(houston)) + "\n"

    stLouis.sort()
    houston.sort()
    print "St. Louis teams:"
    print stLouis
    print "Houston teams:"
    print houston

Results:
Number of elims teams in St. Louis: 174
Number of elims teams in Houston: 82

Spoiler for St. Louis Teams:

Spoiler for Houston Teams:

If anyone sees any teams in the wrong list or code errors, let me know and I'll try to fix it. Québec may not be working correctly as my Python editor couldn't handle the 'é'.