Jon Solow
commited on
Commit
·
13801e8
1
Parent(s):
7fd5387
Implement query for league schedule from pfr
Browse files
src/queries/pfr/league_schedule.py
CHANGED
@@ -10,7 +10,7 @@ def get_full_schedule(season_int: str | int) -> pd.DataFrame:
|
|
10 |
return df
|
11 |
|
12 |
|
13 |
-
def get_week_team_time_map(df_schedule: pd.DataFrame) -> dict[int, dict[str,
|
14 |
min_week = 1
|
15 |
max_week = 23
|
16 |
week_team_time_map: dict[int, dict[str, str]] = {k: {} for k in range(min_week, max_week + 1)}
|
@@ -21,6 +21,6 @@ def get_week_team_time_map(df_schedule: pd.DataFrame) -> dict[int, dict[str, str
|
|
21 |
return week_team_time_map
|
22 |
|
23 |
|
24 |
-
def get_season_time_map(season_int: str | int) -> dict[int, dict[str,
|
25 |
df_schedule = get_full_schedule(season_int)
|
26 |
return get_week_team_time_map(df_schedule)
|
|
|
10 |
return df
|
11 |
|
12 |
|
13 |
+
def get_week_team_time_map(df_schedule: pd.DataFrame) -> dict[int, dict[str, pd.Timestamp]]:
|
14 |
min_week = 1
|
15 |
max_week = 23
|
16 |
week_team_time_map: dict[int, dict[str, str]] = {k: {} for k in range(min_week, max_week + 1)}
|
|
|
21 |
return week_team_time_map
|
22 |
|
23 |
|
24 |
+
def get_season_time_map(season_int: str | int) -> dict[int, dict[str, pd.Timestamp]]:
|
25 |
df_schedule = get_full_schedule(season_int)
|
26 |
return get_week_team_time_map(df_schedule)
|