[HackerRank] Football Competition Winner's Goals (Python)

📖  문제

https://www.hackerrank.com/skills-verification/rest_api_intermediate

HackerRank

 

💻  코드

import requests


def getHomeTotalPages(competition, team, year):
    response = requests.get(
        "https://jsonmock.hackerrank.com/api/football_matches?competition=%s&year=%d&team1=%s&page=1" % (
        competition, year, team))
    return response.json()["total_pages"]


def getHomeTotalGoals(competition, team, year):
    total_goals = 0
    total_pages = getHomeTotalPages(competition, team, year)
    for page in range(1, total_pages + 1):
        response = requests.get(
            "https://jsonmock.hackerrank.com/api/football_matches?competition=%s&year=%d&team1=%s&page=%d" % (
            competition, year, team, page))
        for play in response.json()["data"]:
            total_goals += int(play["team1goals"])
    return (total_goals)


def getAwayTotalPages(competition, team, year):
    response = requests.get(
        "https://jsonmock.hackerrank.com/api/football_matches?competition=%s&year=%d&team2=%s&page=1" % (
        competition, year, team))
    return response.json()["total_pages"]


def getAwayTotalGoals(competition, team, year):
    total_goals = 0
    total_pages = getAwayTotalPages(competition, team, year)
    for page in range(1, total_pages + 1):
        response = requests.get(
            "https://jsonmock.hackerrank.com/api/football_matches?competition=%s&year=%d&team2=%s&page=%d" % (
            competition, year, team, page))
        for play in response.json()["data"]:
            total_goals += int(play["team2goals"])
    return (total_goals)


def getTotalGoals(competition, team, year):
    return getHomeTotalGoals(competition, team, year) + getAwayTotalGoals(competition, team, year)


def getWinner(competition, year):
    response = requests.get(
        "https://jsonmock.hackerrank.com/api/football_competitions?name=%s&year=%d" % (competition, year))
    print(response.json())
    return response.json()["data"][0]["winner"]


def getWinnerTotalGoals(competition, year):
    winner = getWinner(competition, year)
    return getTotalGoals(competition, winner, year)