Add short_output in /results

This commit is contained in:
Cal Corum 2023-06-21 10:05:33 -05:00
parent 5d7780a920
commit 449dafcfae

View File

@ -31,7 +31,8 @@ class ResultList(pydantic.BaseModel):
async def get_results(
season: int, team_abbrev: list = Query(default=None), week_start: Optional[int] = None,
week_end: Optional[int] = None, game_num: list = Query(default=None),
away_abbrev: list = Query(default=None), home_abbrev: list = Query(default=None)):
away_abbrev: list = Query(default=None), home_abbrev: list = Query(default=None),
short_output: Optional[bool] = False):
all_results = Result.select_season(season)
if team_abbrev is not None:
@ -65,7 +66,7 @@ async def get_results(
return_results = {
'count': all_results.count(),
'results': [model_to_dict(x, recurse=False) for x in all_results]
'results': [model_to_dict(x, recurse=not short_output) for x in all_results]
}
db.close()
return return_results