def test_filter_data():
leagues = [{
"league": "epl",
"players": 600
}, {
"league": "la_liga",
"players": 300
}, {
"league": "bundesliga",
"players": 400
}, {
"league": "serie_a",
"players": 500
}, {
"league": "ligue_1",
"players": 600
}, {
"league": "rfpl",
"players": 700
}]
filtered_leagues = filter_data(leagues, {"league": "epl"})
assert filtered_leagues == [{"league": "epl", "players": 600}]
filtered_leagues = filter_data(leagues, {"players": 600})
assert filtered_leagues == [{
"league": "epl",
"players": 600
}, {
"league": "ligue_1",
"players": 600
}]
async def get_league_results(
self, league_name, season, options=None, **kwargs):
"""Returns a list containing information about all the results
(matches) played by the teams in the given league in the given season.
:param league_name: The league's name.
:type league_name: str
:param season: The season.
:type season: str or int
:param options: Options to filter the data by, defaults to None.
:param options: dict, optional
:return: A list of the results as seen on Understat's league overview.
:rtype: list
"""
url = LEAGUE_URL.format(to_league_name(league_name), season)
dates_data = await get_data(self.session, url, "datesData")
results = [r for r in dates_data if r["isResult"]]
if options:
kwargs = options
filtered_data = filter_data(results, kwargs)
return filtered_data
async def get_league_players(self,
league_name,
season,
options=None,
**kwargs):
"""Returns a list containing information about all the players in
the given league in the given season.
:param league_name: The league's name.
:type league_name: str
:param season: The season.
:param options: Options to filter the data by, defaults to None.
:param options: dict, optional
:type season: str or int
:return: A list of the players as seen on Understat's league overview.
:rtype: list
"""
url = LEAGUE_URL.format(to_league_name(league_name), season)
players_data = await get_data(self.session, url, "playersData")
if options:
kwargs = options
filtered_data = filter_data(players_data, kwargs)
return filtered_data
async def get_team_players(self,
team_name,
season,
options=None,
**kwargs):
"""Returns a team's player statistics in the given season.
:param team_name: A team's name.
:type team_name: str
:param season: The season.
:type season: int or str
:param options: Options to filter the data by, defaults to None.
:param options: dict, optional
:return: List of the team's players' statistics in the given season.
:rtype: list
"""
url = TEAM_URL.format(team_name.replace(" ", "_"), season)
players_data = await get_data(self.session, url, "playersData")
if options:
kwargs = options
filtered_data = filter_data(players_data, kwargs)
return filtered_data
async def get_team_fixtures(self,
team_name,
season,
options=None,
**kwargs):
"""Returns a team's upcoming fixtures in the given season.
:param team_name: A team's name.
:type team_name: str
:param season: The season.
:type season: int or str
:param options: Options to filter the data by, defaults to None.
:param options: dict, optional
:return: List of the team's upcoming fixtures in the given season.
:rtype: list
"""
url = TEAM_URL.format(team_name.replace(" ", "_"), season)
dates_data = await get_data(self.session, url, "datesData")
fixtures = [f for f in dates_data if not f["isResult"]]
if options:
kwargs = options
filtered_data = filter_data(fixtures, kwargs)
return filtered_data
def get_league_fixtures(
self, league_name, season, options=None, **kwargs):
"""Returns a list containing information about all the upcoming
fixtures of the given league in the given season.
:param league_name: The league's name.
:type league_name: str
:param season: The season.
:type season: str or int
:param options: Options to filter the data by, defaults to None.
:param options: dict, optional
:return: A list of the fixtures as seen on Understat's league overview.
:rtype: list
"""
url = LEAGUE_URL.format(to_league_name(league_name), season)
dates_data = get_data(url, "datesData")
fixtures = [f for f in dates_data if not f["isResult"]]
if options:
kwargs = options
filtered_data = filter_data(fixtures, kwargs)
return filtered_data
async def get_stats(self, options=None, **kwargs):
"""Returns a list containing stats of every league, grouped by month.
:param options: Options to filter the data by, defaults to None.
:param options: dict, optional
:return: List of dictionaries.
:rtype: list
"""
stats = await get_data(self.session, BASE_URL, "statData")
if options:
kwargs = options
filtered_data = filter_data(stats, kwargs)
return filtered_data
async def get_player_matches(self, player_id, options=None, **kwargs):
"""Returns the player with the given ID's matches data.
:param player_id: The player's Understat ID.
:type player_id: int or str
:param options: Options to filter the data by, defaults to None.
:param options: dict, optional
:return: List of the player's matches data.
:rtype: list
"""
url = PLAYER_URL.format(player_id)
matches_data = await get_data(self.session, url, "matchesData")
if options:
kwargs = options
filtered_data = filter_data(matches_data, kwargs)
return filtered_data
def get_player_shots(self, player_id, options=None, **kwargs):
"""Returns the player with the given ID's shot data.
:param player_id: The player's Understat ID.
:type player_id: int or str
:param options: Options to filter the data by, defaults to None.
:param options: dict, optional
:return: List of the player's shot data.
:rtype: list
"""
url = PLAYER_URL.format(player_id)
shots_data = get_data(url, "shotsData")
if options:
kwargs = options
filtered_data = filter_data(shots_data, kwargs)
return filtered_data
async def get_match_shots(self, match_id, options=None, **kwargs):
"""Returns a dictionary containing information about shots taken by
the players in the given match.
:param fixture_id: A match's ID.
:type fixture_id: int
:param options: Options to filter the data by, defaults to None.
:param options: dict, optional
:return: Dictionary containing information about the players who played
in the match.
:rtype: dict
"""
url = MATCH_URL.format(match_id)
players_data = await get_data(self.session, url, "shotsData")
if options:
kwargs = options
filtered_data = filter_data(players_data, kwargs)
return filtered_data
def get_teams(self, league_name, season, options=None, **kwargs):
"""Returns a list containing information about all the teams in
the given league in the given season.
:param league_name: The league's name.
:type league_name: str
:param season: The season.
:param options: Options to filter the data by, defaults to None.
:param options: dict, optional
:type season: str or int
:return: A list of the league's table as seen on Understat's
league overview.
:rtype: list
"""
url = LEAGUE_URL.format(to_league_name(league_name), season)
teams_data = get_data(url, "teamsData")
if options:
kwargs = options
filtered_data = filter_data(list(teams_data.values()), kwargs)
return filtered_data
def get_team_results(
self, team_name, season, options=None, **kwargs):
"""Returns a team's results in the given season.
:param team_name: A team's name.
:type team_name: str
:param season: The season.
:type season: int or str
:param options: Options to filter the data by, defaults to None.
:param options: dict, optional
:return: List of the team's results in the given season.
:rtype: list
"""
url = TEAM_URL.format(team_name.replace(" ", "_"), season)
dates_data = get_data(url, "datesData")
results = [r for r in dates_data if r["isResult"]]
if options:
kwargs = options
filtered_data = filter_data(results, kwargs)
return filtered_data
