cal/paper-dynasty-database
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update /games and /plays for sheets import

Browse Source
This commit is contained in:
Cal Corum 2023-10-22 16:41:32 -05:00
parent 23181d55e3
commit 74a037b3fc
2 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
app/routers_v2/stratgame.py
View File

@ -76,12 +76,17 @@ async def get_games(
if csv:
return_vals = [model_to_dict(x) for x in all_games]
for x in return_vals:
x['away_team_name'] = x['away_team']['lname']
x['home_team_name'] = x['home_team']['lname']
x['away_abbrev'] = x['away_team']['abbrev']
x['home_abbrev'] = x['home_team']['abbrev']
del x['away_team'], x['home_team']
db.close()
return Response(content=pd.DataFrame(return_vals).to_csv(index=False), media_type='text/csv')
output = pd.DataFrame(return_vals)[[
'id', 'away_abbrev', 'home_abbrev', 'away_score', 'home_score', 'away_team_value', 'home_team_value',
'game_type', 'season', 'week', 'short_game', 'ranked'
]]
return Response(content=output.to_csv(index=False), media_type='text/csv')
return_val = {'count': all_games.count(), 'games': [
model_to_dict(x, recurse=not short_output) for x in all_games

8
app/routers_v2/stratplays.py
View File

@ -327,7 +327,7 @@ async def get_batting_totals(
group_by: Literal['team', 'player', 'playerteam', 'playergame', 'teamgame', 'league', 'gametype'] = 'player',
min_pa: Optional[int] = 1, team_id: list = Query(default=None), inning: list = Query(default=None),
obc: list = Query(default=None), risp: Optional[bool] = None, game_type: list = Query(default=None),
page_num: Optional[int] = 1, sort: Optional[str] = None, limit: Optional[int] = 500,
page_num: Optional[int] = 1, sort: Optional[str] = 'pa-desc', limit: Optional[int] = 500,
short_output: Optional[bool] = False, csv: Optional[bool] = False):
season_games = StratGame.select()
if season is not None:
@ -386,7 +386,7 @@ async def get_batting_totals(
)
if player_id is not None:
all_players = Player.select().where(Player.id << player_id)
all_players = Player.select().where(Player.player_id << player_id)
bat_plays = bat_plays.where(StratPlay.batter << all_players)
run_plays = run_plays.where(StratPlay.runner << all_players)
if team_id is not None:
@ -583,7 +583,7 @@ async def get_pitching_totals(
group_by: Literal['team', 'player', 'playerteam', 'playergame', 'teamgame', 'league', 'gametype'] = 'player',
min_pa: Optional[int] = 1, team_id: list = Query(default=None), manager_id: list = Query(default=None),
obc: list = Query(default=None), risp: Optional[bool] = None, inning: list = Query(default=None),
page_num: Optional[int] = 1, game_type: list = Query(default=None), sort: Optional[str] = None,
page_num: Optional[int] = 1, game_type: list = Query(default=None), sort: Optional[str] = 'ip-desc',
limit: Optional[int] = 500, short_output: Optional[bool] = False, csv: Optional[bool] = False):
season_games = StratGame.select()
if season is not None:
@ -650,7 +650,7 @@ async def get_pitching_totals(
)
if player_id is not None:
all_players = Player.select().where(Player.id << player_id)
all_players = Player.select().where(Player.player_id << player_id)
pit_plays = pit_plays.where(StratPlay.pitcher << all_players)
if team_id is not None:
all_teams = Team.select().where(Team.id << team_id)