Added play_num param to /plays

This commit is contained in:
Cal Corum 2023-09-16 18:36:50 -05:00
parent 1fa2a207ee
commit b8c1096b86

View File

@ -126,7 +126,7 @@ async def get_plays(
defender_id: list = Query(default=None), runner_id: list = Query(default=None),
offense_team_id: list = Query(default=None), defense_team_id: list = Query(default=None),
hit: Optional[int] = None, double: Optional[int] = None, triple: Optional[int] = None,
homerun: Optional[int] = None,
homerun: Optional[int] = None, play_num: list = Query(default=None),
sb: Optional[int] = None, cs: Optional[int] = None, manager_id: list = Query(default=None),
run: Optional[int] = None, e_run: Optional[int] = None, rbi: list = Query(default=None),
outs: list = Query(default=None), wild_pitch: Optional[int] = None, is_final_out: Optional[bool] = None,
@ -225,6 +225,8 @@ async def get_plays(
all_plays = all_plays.where(StratPlay.wpa >= min_wpa)
if max_wpa is not None:
all_plays = all_plays.where(StratPlay.wpa <= max_wpa)
if play_num is not None:
all_plays = all_plays.where(StratPlay.play_num << play_num)
if limit > 5000:
limit = 5000