Merge branch 'main' into issue/147-feat-add-limit-pagination-to-events-endpoint
This commit is contained in:
commit
ae2e7320c5
@ -8,10 +8,7 @@ from ..db_engine import Result, model_to_dict, Team, DataError, DoesNotExist
|
|||||||
from ..dependencies import oauth2_scheme, valid_token
|
from ..dependencies import oauth2_scheme, valid_token
|
||||||
|
|
||||||
|
|
||||||
router = APIRouter(
|
router = APIRouter(prefix="/api/v2/results", tags=["results"])
|
||||||
prefix='/api/v2/results',
|
|
||||||
tags=['results']
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class ResultModel(pydantic.BaseModel):
|
class ResultModel(pydantic.BaseModel):
|
||||||
@ -31,15 +28,29 @@ class ResultModel(pydantic.BaseModel):
|
|||||||
game_type: str
|
game_type: str
|
||||||
|
|
||||||
|
|
||||||
@router.get('')
|
@router.get("")
|
||||||
async def get_results(
|
async def get_results(
|
||||||
away_team_id: Optional[int] = None, home_team_id: Optional[int] = None, team_one_id: Optional[int] = None,
|
away_team_id: Optional[int] = None,
|
||||||
team_two_id: Optional[int] = None, away_score_min: Optional[int] = None, away_score_max: Optional[int] = None,
|
home_team_id: Optional[int] = None,
|
||||||
home_score_min: Optional[int] = None, home_score_max: Optional[int] = None, bothscore_min: Optional[int] = None,
|
team_one_id: Optional[int] = None,
|
||||||
bothscore_max: Optional[int] = None, season: Optional[int] = None, week: Optional[int] = None,
|
team_two_id: Optional[int] = None,
|
||||||
week_start: Optional[int] = None, week_end: Optional[int] = None, ranked: Optional[bool] = None,
|
away_score_min: Optional[int] = None,
|
||||||
short_game: Optional[bool] = None, game_type: Optional[str] = None, vs_ai: Optional[bool] = None,
|
away_score_max: Optional[int] = None,
|
||||||
csv: Optional[bool] = None):
|
home_score_min: Optional[int] = None,
|
||||||
|
home_score_max: Optional[int] = None,
|
||||||
|
bothscore_min: Optional[int] = None,
|
||||||
|
bothscore_max: Optional[int] = None,
|
||||||
|
season: Optional[int] = None,
|
||||||
|
week: Optional[int] = None,
|
||||||
|
week_start: Optional[int] = None,
|
||||||
|
week_end: Optional[int] = None,
|
||||||
|
ranked: Optional[bool] = None,
|
||||||
|
short_game: Optional[bool] = None,
|
||||||
|
game_type: Optional[str] = None,
|
||||||
|
vs_ai: Optional[bool] = None,
|
||||||
|
csv: Optional[bool] = None,
|
||||||
|
limit: int = 100,
|
||||||
|
):
|
||||||
all_results = Result.select()
|
all_results = Result.select()
|
||||||
|
|
||||||
# if all_results.count() == 0:
|
# if all_results.count() == 0:
|
||||||
@ -51,28 +62,40 @@ async def get_results(
|
|||||||
this_team = Team.get_by_id(away_team_id)
|
this_team = Team.get_by_id(away_team_id)
|
||||||
all_results = all_results.where(Result.away_team == this_team)
|
all_results = all_results.where(Result.away_team == this_team)
|
||||||
except DoesNotExist:
|
except DoesNotExist:
|
||||||
raise HTTPException(status_code=404, detail=f'No team found with id {away_team_id}')
|
raise HTTPException(
|
||||||
|
status_code=404, detail=f"No team found with id {away_team_id}"
|
||||||
|
)
|
||||||
|
|
||||||
if home_team_id is not None:
|
if home_team_id is not None:
|
||||||
try:
|
try:
|
||||||
this_team = Team.get_by_id(home_team_id)
|
this_team = Team.get_by_id(home_team_id)
|
||||||
all_results = all_results.where(Result.home_team == this_team)
|
all_results = all_results.where(Result.home_team == this_team)
|
||||||
except DoesNotExist:
|
except DoesNotExist:
|
||||||
raise HTTPException(status_code=404, detail=f'No team found with id {home_team_id}')
|
raise HTTPException(
|
||||||
|
status_code=404, detail=f"No team found with id {home_team_id}"
|
||||||
|
)
|
||||||
|
|
||||||
if team_one_id is not None:
|
if team_one_id is not None:
|
||||||
try:
|
try:
|
||||||
this_team = Team.get_by_id(team_one_id)
|
this_team = Team.get_by_id(team_one_id)
|
||||||
all_results = all_results.where((Result.home_team == this_team) | (Result.away_team == this_team))
|
all_results = all_results.where(
|
||||||
|
(Result.home_team == this_team) | (Result.away_team == this_team)
|
||||||
|
)
|
||||||
except DoesNotExist:
|
except DoesNotExist:
|
||||||
raise HTTPException(status_code=404, detail=f'No team found with id {team_one_id}')
|
raise HTTPException(
|
||||||
|
status_code=404, detail=f"No team found with id {team_one_id}"
|
||||||
|
)
|
||||||
|
|
||||||
if team_two_id is not None:
|
if team_two_id is not None:
|
||||||
try:
|
try:
|
||||||
this_team = Team.get_by_id(team_two_id)
|
this_team = Team.get_by_id(team_two_id)
|
||||||
all_results = all_results.where((Result.home_team == this_team) | (Result.away_team == this_team))
|
all_results = all_results.where(
|
||||||
|
(Result.home_team == this_team) | (Result.away_team == this_team)
|
||||||
|
)
|
||||||
except DoesNotExist:
|
except DoesNotExist:
|
||||||
raise HTTPException(status_code=404, detail=f'No team found with id {team_two_id}')
|
raise HTTPException(
|
||||||
|
status_code=404, detail=f"No team found with id {team_two_id}"
|
||||||
|
)
|
||||||
|
|
||||||
if away_score_min is not None:
|
if away_score_min is not None:
|
||||||
all_results = all_results.where(Result.away_score >= away_score_min)
|
all_results = all_results.where(Result.away_score >= away_score_min)
|
||||||
@ -87,10 +110,14 @@ async def get_results(
|
|||||||
all_results = all_results.where(Result.home_score <= home_score_max)
|
all_results = all_results.where(Result.home_score <= home_score_max)
|
||||||
|
|
||||||
if bothscore_min is not None:
|
if bothscore_min is not None:
|
||||||
all_results = all_results.where((Result.home_score >= bothscore_min) & (Result.away_score >= bothscore_min))
|
all_results = all_results.where(
|
||||||
|
(Result.home_score >= bothscore_min) & (Result.away_score >= bothscore_min)
|
||||||
|
)
|
||||||
|
|
||||||
if bothscore_max is not None:
|
if bothscore_max is not None:
|
||||||
all_results = all_results.where((Result.home_score <= bothscore_max) & (Result.away_score <= bothscore_max))
|
all_results = all_results.where(
|
||||||
|
(Result.home_score <= bothscore_max) & (Result.away_score <= bothscore_max)
|
||||||
|
)
|
||||||
|
|
||||||
if season is not None:
|
if season is not None:
|
||||||
all_results = all_results.where(Result.season == season)
|
all_results = all_results.where(Result.season == season)
|
||||||
@ -114,6 +141,8 @@ async def get_results(
|
|||||||
all_results = all_results.where(Result.game_type == game_type)
|
all_results = all_results.where(Result.game_type == game_type)
|
||||||
|
|
||||||
all_results = all_results.order_by(Result.id)
|
all_results = all_results.order_by(Result.id)
|
||||||
|
limit = max(0, min(limit, 500))
|
||||||
|
all_results = all_results.limit(limit)
|
||||||
# Not functional
|
# Not functional
|
||||||
# if vs_ai is not None:
|
# if vs_ai is not None:
|
||||||
# AwayTeam = Team.alias()
|
# AwayTeam = Team.alias()
|
||||||
@ -134,60 +163,115 @@ async def get_results(
|
|||||||
# logging.info(f'Result Query:\n\n{all_results}')
|
# logging.info(f'Result Query:\n\n{all_results}')
|
||||||
|
|
||||||
if csv:
|
if csv:
|
||||||
data_list = [['id', 'away_abbrev', 'home_abbrev', 'away_score', 'home_score', 'away_tv', 'home_tv',
|
data_list = [
|
||||||
'game_type', 'season', 'week', 'short_game', 'ranked']]
|
[
|
||||||
|
"id",
|
||||||
|
"away_abbrev",
|
||||||
|
"home_abbrev",
|
||||||
|
"away_score",
|
||||||
|
"home_score",
|
||||||
|
"away_tv",
|
||||||
|
"home_tv",
|
||||||
|
"game_type",
|
||||||
|
"season",
|
||||||
|
"week",
|
||||||
|
"short_game",
|
||||||
|
"ranked",
|
||||||
|
]
|
||||||
|
]
|
||||||
for line in all_results:
|
for line in all_results:
|
||||||
data_list.append([
|
data_list.append(
|
||||||
line.id, line.away_team.abbrev, line.home_team.abbrev, line.away_score, line.home_score,
|
[
|
||||||
line.away_team_value, line.home_team_value, line.game_type if line.game_type else 'minor-league',
|
line.id,
|
||||||
line.season, line.week, line.short_game, line.ranked
|
line.away_team.abbrev,
|
||||||
])
|
line.home_team.abbrev,
|
||||||
|
line.away_score,
|
||||||
|
line.home_score,
|
||||||
|
line.away_team_value,
|
||||||
|
line.home_team_value,
|
||||||
|
line.game_type if line.game_type else "minor-league",
|
||||||
|
line.season,
|
||||||
|
line.week,
|
||||||
|
line.short_game,
|
||||||
|
line.ranked,
|
||||||
|
]
|
||||||
|
)
|
||||||
return_val = DataFrame(data_list).to_csv(header=False, index=False)
|
return_val = DataFrame(data_list).to_csv(header=False, index=False)
|
||||||
|
|
||||||
return Response(content=return_val, media_type='text/csv')
|
return Response(content=return_val, media_type="text/csv")
|
||||||
|
|
||||||
else:
|
else:
|
||||||
return_val = {'count': all_results.count(), 'results': []}
|
return_val = {"count": all_results.count(), "results": []}
|
||||||
for x in all_results:
|
for x in all_results:
|
||||||
return_val['results'].append(model_to_dict(x))
|
return_val["results"].append(model_to_dict(x))
|
||||||
|
|
||||||
return return_val
|
return return_val
|
||||||
|
|
||||||
|
|
||||||
@router.get('/{result_id}')
|
@router.get("/{result_id}")
|
||||||
async def get_one_results(result_id, csv: Optional[bool] = None):
|
async def get_one_results(result_id, csv: Optional[bool] = None):
|
||||||
try:
|
try:
|
||||||
this_result = Result.get_by_id(result_id)
|
this_result = Result.get_by_id(result_id)
|
||||||
except DoesNotExist:
|
except DoesNotExist:
|
||||||
raise HTTPException(status_code=404, detail=f'No result found with id {result_id}')
|
raise HTTPException(
|
||||||
|
status_code=404, detail=f"No result found with id {result_id}"
|
||||||
|
)
|
||||||
|
|
||||||
if csv:
|
if csv:
|
||||||
data_list = [
|
data_list = [
|
||||||
['id', 'away_abbrev', 'home_abbrev', 'away_score', 'home_score', 'away_tv', 'home_tv', 'game_type',
|
[
|
||||||
'season', 'week', 'game_type'],
|
"id",
|
||||||
[this_result.id, this_result.away_team.abbrev, this_result.away_team.abbrev, this_result.away_score,
|
"away_abbrev",
|
||||||
this_result.home_score, this_result.away_team_value, this_result.home_team_value,
|
"home_abbrev",
|
||||||
this_result.game_type if this_result.game_type else 'minor-league',
|
"away_score",
|
||||||
this_result.season, this_result.week, this_result.game_type]
|
"home_score",
|
||||||
|
"away_tv",
|
||||||
|
"home_tv",
|
||||||
|
"game_type",
|
||||||
|
"season",
|
||||||
|
"week",
|
||||||
|
"game_type",
|
||||||
|
],
|
||||||
|
[
|
||||||
|
this_result.id,
|
||||||
|
this_result.away_team.abbrev,
|
||||||
|
this_result.away_team.abbrev,
|
||||||
|
this_result.away_score,
|
||||||
|
this_result.home_score,
|
||||||
|
this_result.away_team_value,
|
||||||
|
this_result.home_team_value,
|
||||||
|
this_result.game_type if this_result.game_type else "minor-league",
|
||||||
|
this_result.season,
|
||||||
|
this_result.week,
|
||||||
|
this_result.game_type,
|
||||||
|
],
|
||||||
]
|
]
|
||||||
return_val = DataFrame(data_list).to_csv(header=False, index=False)
|
return_val = DataFrame(data_list).to_csv(header=False, index=False)
|
||||||
|
|
||||||
return Response(content=return_val, media_type='text/csv')
|
return Response(content=return_val, media_type="text/csv")
|
||||||
|
|
||||||
else:
|
else:
|
||||||
return_val = model_to_dict(this_result)
|
return_val = model_to_dict(this_result)
|
||||||
return return_val
|
return return_val
|
||||||
|
|
||||||
|
|
||||||
@router.get('/team/{team_id}')
|
@router.get("/team/{team_id}")
|
||||||
async def get_team_results(
|
async def get_team_results(
|
||||||
team_id: int, season: Optional[int] = None, week: Optional[int] = None, csv: Optional[bool] = False):
|
team_id: int,
|
||||||
all_results = Result.select().where((Result.away_team_id == team_id) | (Result.home_team_id == team_id)).order_by(Result.id)
|
season: Optional[int] = None,
|
||||||
|
week: Optional[int] = None,
|
||||||
|
csv: Optional[bool] = False,
|
||||||
|
):
|
||||||
|
all_results = (
|
||||||
|
Result.select()
|
||||||
|
.where((Result.away_team_id == team_id) | (Result.home_team_id == team_id))
|
||||||
|
.order_by(Result.id)
|
||||||
|
)
|
||||||
try:
|
try:
|
||||||
this_team = Team.get_by_id(team_id)
|
this_team = Team.get_by_id(team_id)
|
||||||
except DoesNotExist as e:
|
except DoesNotExist:
|
||||||
logging.error(f'Unknown team id {team_id} trying to pull team results')
|
logging.error(f"Unknown team id {team_id} trying to pull team results")
|
||||||
raise HTTPException(404, f'Team id {team_id} not found')
|
raise HTTPException(404, f"Team id {team_id} not found")
|
||||||
|
|
||||||
if season is not None:
|
if season is not None:
|
||||||
all_results = all_results.where(Result.season == season)
|
all_results = all_results.where(Result.season == season)
|
||||||
@ -224,31 +308,38 @@ async def get_team_results(
|
|||||||
|
|
||||||
if csv:
|
if csv:
|
||||||
data_list = [
|
data_list = [
|
||||||
['team_id', 'ranked_wins', 'ranked_losses', 'casual_wins', 'casual_losses', 'team_ranking'],
|
[
|
||||||
[team_id, r_wins, r_loss, c_wins, c_loss, this_team.ranking]
|
"team_id",
|
||||||
|
"ranked_wins",
|
||||||
|
"ranked_losses",
|
||||||
|
"casual_wins",
|
||||||
|
"casual_losses",
|
||||||
|
"team_ranking",
|
||||||
|
],
|
||||||
|
[team_id, r_wins, r_loss, c_wins, c_loss, this_team.ranking],
|
||||||
]
|
]
|
||||||
return_val = DataFrame(data_list).to_csv(header=False, index=False)
|
return_val = DataFrame(data_list).to_csv(header=False, index=False)
|
||||||
|
|
||||||
return Response(content=return_val, media_type='text/csv')
|
return Response(content=return_val, media_type="text/csv")
|
||||||
|
|
||||||
else:
|
else:
|
||||||
return_val = {
|
return_val = {
|
||||||
'team': model_to_dict(this_team),
|
"team": model_to_dict(this_team),
|
||||||
'ranked_wins': r_wins,
|
"ranked_wins": r_wins,
|
||||||
'ranked_losses': r_loss,
|
"ranked_losses": r_loss,
|
||||||
'casual_wins': c_wins,
|
"casual_wins": c_wins,
|
||||||
'casual_losses': c_loss,
|
"casual_losses": c_loss,
|
||||||
}
|
}
|
||||||
return return_val
|
return return_val
|
||||||
|
|
||||||
|
|
||||||
@router.post('')
|
@router.post("")
|
||||||
async def post_result(result: ResultModel, token: str = Depends(oauth2_scheme)):
|
async def post_result(result: ResultModel, token: str = Depends(oauth2_scheme)):
|
||||||
if not valid_token(token):
|
if not valid_token(token):
|
||||||
logging.warning('Bad Token: [REDACTED]')
|
logging.warning("Bad Token: [REDACTED]")
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
status_code=401,
|
status_code=401,
|
||||||
detail='You are not authorized to post results. This event has been logged.'
|
detail="You are not authorized to post results. This event has been logged.",
|
||||||
)
|
)
|
||||||
|
|
||||||
this_result = Result(**result.__dict__)
|
this_result = Result(**result.__dict__)
|
||||||
@ -256,11 +347,11 @@ async def post_result(result: ResultModel, token: str = Depends(oauth2_scheme)):
|
|||||||
|
|
||||||
if result.ranked:
|
if result.ranked:
|
||||||
if not result.away_team_ranking:
|
if not result.away_team_ranking:
|
||||||
error = f'Ranked game did not include away team ({result.away_team_id}) ranking.'
|
error = f"Ranked game did not include away team ({result.away_team_id}) ranking."
|
||||||
logging.error(error)
|
logging.error(error)
|
||||||
raise DataError(error)
|
raise DataError(error)
|
||||||
if not result.home_team_ranking:
|
if not result.home_team_ranking:
|
||||||
error = f'Ranked game did not include home team ({result.home_team_id}) ranking.'
|
error = f"Ranked game did not include home team ({result.home_team_id}) ranking."
|
||||||
logging.error(error)
|
logging.error(error)
|
||||||
raise DataError(error)
|
raise DataError(error)
|
||||||
|
|
||||||
@ -269,11 +360,15 @@ async def post_result(result: ResultModel, token: str = Depends(oauth2_scheme)):
|
|||||||
exp_score = 1 / (1 + (10**ratio))
|
exp_score = 1 / (1 + (10**ratio))
|
||||||
away_win = True if result.away_score > result.home_score else False
|
away_win = True if result.away_score > result.home_score else False
|
||||||
total_delta = k_value * exp_score
|
total_delta = k_value * exp_score
|
||||||
high_delta = total_delta * exp_score if exp_score > .5 else total_delta * (1 - exp_score)
|
high_delta = (
|
||||||
|
total_delta * exp_score
|
||||||
|
if exp_score > 0.5
|
||||||
|
else total_delta * (1 - exp_score)
|
||||||
|
)
|
||||||
low_delta = total_delta - high_delta
|
low_delta = total_delta - high_delta
|
||||||
|
|
||||||
# exp_score > .5 means away team is favorite
|
# exp_score > .5 means away team is favorite
|
||||||
if exp_score > .5 and away_win:
|
if exp_score > 0.5 and away_win:
|
||||||
final_delta = low_delta
|
final_delta = low_delta
|
||||||
away_delta = low_delta * 3
|
away_delta = low_delta * 3
|
||||||
home_delta = -low_delta
|
home_delta = -low_delta
|
||||||
@ -281,7 +376,7 @@ async def post_result(result: ResultModel, token: str = Depends(oauth2_scheme)):
|
|||||||
final_delta = high_delta
|
final_delta = high_delta
|
||||||
away_delta = high_delta * 3
|
away_delta = high_delta * 3
|
||||||
home_delta = -high_delta
|
home_delta = -high_delta
|
||||||
elif exp_score <= .5 and not away_win:
|
elif exp_score <= 0.5 and not away_win:
|
||||||
final_delta = low_delta
|
final_delta = low_delta
|
||||||
away_delta = -low_delta
|
away_delta = -low_delta
|
||||||
home_delta = low_delta * 3
|
home_delta = low_delta * 3
|
||||||
@ -294,18 +389,20 @@ async def post_result(result: ResultModel, token: str = Depends(oauth2_scheme)):
|
|||||||
away_delta = 0
|
away_delta = 0
|
||||||
home_delta = 0
|
home_delta = 0
|
||||||
|
|
||||||
logging.debug(f'/results ranking deltas\n\nk_value: {k_value} / ratio: {ratio} / '
|
logging.debug(
|
||||||
f'exp_score: {exp_score} / away_win: {away_win} / total_delta: {total_delta} / '
|
f"/results ranking deltas\n\nk_value: {k_value} / ratio: {ratio} / "
|
||||||
f'high_delta: {high_delta} / low_delta: {low_delta} / final_delta: {final_delta} / ')
|
f"exp_score: {exp_score} / away_win: {away_win} / total_delta: {total_delta} / "
|
||||||
|
f"high_delta: {high_delta} / low_delta: {low_delta} / final_delta: {final_delta} / "
|
||||||
|
)
|
||||||
|
|
||||||
away_team = Team.get_by_id(result.away_team_id)
|
away_team = Team.get_by_id(result.away_team_id)
|
||||||
away_team.ranking += away_delta
|
away_team.ranking += away_delta
|
||||||
away_team.save()
|
away_team.save()
|
||||||
logging.info(f'Just updated {away_team.abbrev} ranking to {away_team.ranking}')
|
logging.info(f"Just updated {away_team.abbrev} ranking to {away_team.ranking}")
|
||||||
home_team = Team.get_by_id(result.home_team_id)
|
home_team = Team.get_by_id(result.home_team_id)
|
||||||
home_team.ranking += home_delta
|
home_team.ranking += home_delta
|
||||||
home_team.save()
|
home_team.save()
|
||||||
logging.info(f'Just updated {home_team.abbrev} ranking to {home_team.ranking}')
|
logging.info(f"Just updated {home_team.abbrev} ranking to {home_team.ranking}")
|
||||||
|
|
||||||
if saved == 1:
|
if saved == 1:
|
||||||
return_val = model_to_dict(this_result)
|
return_val = model_to_dict(this_result)
|
||||||
@ -313,27 +410,38 @@ async def post_result(result: ResultModel, token: str = Depends(oauth2_scheme)):
|
|||||||
else:
|
else:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
status_code=418,
|
status_code=418,
|
||||||
detail='Well slap my ass and call me a teapot; I could not save that roster'
|
detail="Well slap my ass and call me a teapot; I could not save that roster",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@router.patch('/{result_id}')
|
@router.patch("/{result_id}")
|
||||||
async def patch_result(
|
async def patch_result(
|
||||||
result_id, away_team_id: Optional[int] = None, home_team_id: Optional[int] = None,
|
result_id,
|
||||||
away_score: Optional[int] = None, home_score: Optional[int] = None, away_team_value: Optional[int] = None,
|
away_team_id: Optional[int] = None,
|
||||||
home_team_value: Optional[int] = None, scorecard: Optional[str] = None, week: Optional[int] = None,
|
home_team_id: Optional[int] = None,
|
||||||
season: Optional[int] = None, short_game: Optional[bool] = None, game_type: Optional[str] = None,
|
away_score: Optional[int] = None,
|
||||||
token: str = Depends(oauth2_scheme)):
|
home_score: Optional[int] = None,
|
||||||
|
away_team_value: Optional[int] = None,
|
||||||
|
home_team_value: Optional[int] = None,
|
||||||
|
scorecard: Optional[str] = None,
|
||||||
|
week: Optional[int] = None,
|
||||||
|
season: Optional[int] = None,
|
||||||
|
short_game: Optional[bool] = None,
|
||||||
|
game_type: Optional[str] = None,
|
||||||
|
token: str = Depends(oauth2_scheme),
|
||||||
|
):
|
||||||
if not valid_token(token):
|
if not valid_token(token):
|
||||||
logging.warning('Bad Token: [REDACTED]')
|
logging.warning("Bad Token: [REDACTED]")
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
status_code=401,
|
status_code=401,
|
||||||
detail='You are not authorized to patch results. This event has been logged.'
|
detail="You are not authorized to patch results. This event has been logged.",
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
this_result = Result.get_by_id(result_id)
|
this_result = Result.get_by_id(result_id)
|
||||||
except DoesNotExist:
|
except DoesNotExist:
|
||||||
raise HTTPException(status_code=404, detail=f'No result found with id {result_id}')
|
raise HTTPException(
|
||||||
|
status_code=404, detail=f"No result found with id {result_id}"
|
||||||
|
)
|
||||||
|
|
||||||
if away_team_id is not None:
|
if away_team_id is not None:
|
||||||
this_result.away_team_id = away_team_id
|
this_result.away_team_id = away_team_id
|
||||||
@ -377,27 +485,32 @@ async def patch_result(
|
|||||||
else:
|
else:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
status_code=418,
|
status_code=418,
|
||||||
detail='Well slap my ass and call me a teapot; I could not save that event'
|
detail="Well slap my ass and call me a teapot; I could not save that event",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@router.delete('/{result_id}')
|
@router.delete("/{result_id}")
|
||||||
async def delete_result(result_id, token: str = Depends(oauth2_scheme)):
|
async def delete_result(result_id, token: str = Depends(oauth2_scheme)):
|
||||||
if not valid_token(token):
|
if not valid_token(token):
|
||||||
logging.warning('Bad Token: [REDACTED]')
|
logging.warning("Bad Token: [REDACTED]")
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
status_code=401,
|
status_code=401,
|
||||||
detail='You are not authorized to post results. This event has been logged.'
|
detail="You are not authorized to post results. This event has been logged.",
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
this_result = Result.get_by_id(result_id)
|
this_result = Result.get_by_id(result_id)
|
||||||
except DoesNotExist:
|
except DoesNotExist:
|
||||||
raise HTTPException(status_code=404, detail=f'No result found with id {result_id}')
|
raise HTTPException(
|
||||||
|
status_code=404, detail=f"No result found with id {result_id}"
|
||||||
|
)
|
||||||
|
|
||||||
count = this_result.delete_instance()
|
count = this_result.delete_instance()
|
||||||
|
|
||||||
if count == 1:
|
if count == 1:
|
||||||
raise HTTPException(status_code=200, detail=f'Result {result_id} has been deleted')
|
raise HTTPException(
|
||||||
|
status_code=200, detail=f"Result {result_id} has been deleted"
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
raise HTTPException(status_code=500, detail=f'Result {result_id} was not deleted')
|
raise HTTPException(
|
||||||
|
status_code=500, detail=f"Result {result_id} was not deleted"
|
||||||
|
)
|
||||||
|
|||||||
@ -8,10 +8,7 @@ from ..db_engine import StratGame, model_to_dict, fn
|
|||||||
from ..dependencies import oauth2_scheme, valid_token
|
from ..dependencies import oauth2_scheme, valid_token
|
||||||
|
|
||||||
|
|
||||||
router = APIRouter(
|
router = APIRouter(prefix="/api/v2/games", tags=["games"])
|
||||||
prefix='/api/v2/games',
|
|
||||||
tags=['games']
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class GameModel(pydantic.BaseModel):
|
class GameModel(pydantic.BaseModel):
|
||||||
@ -35,13 +32,22 @@ class GameList(pydantic.BaseModel):
|
|||||||
games: List[GameModel]
|
games: List[GameModel]
|
||||||
|
|
||||||
|
|
||||||
@router.get('')
|
@router.get("")
|
||||||
async def get_games(
|
async def get_games(
|
||||||
season: list = Query(default=None), forfeit: Optional[bool] = None, away_team_id: list = Query(default=None),
|
season: list = Query(default=None),
|
||||||
home_team_id: list = Query(default=None), team1_id: list = Query(default=None),
|
forfeit: Optional[bool] = None,
|
||||||
team2_id: list = Query(default=None), game_type: list = Query(default=None), ranked: Optional[bool] = None,
|
away_team_id: list = Query(default=None),
|
||||||
short_game: Optional[bool] = None, csv: Optional[bool] = False, short_output: bool = False,
|
home_team_id: list = Query(default=None),
|
||||||
gauntlet_id: Optional[int] = None):
|
team1_id: list = Query(default=None),
|
||||||
|
team2_id: list = Query(default=None),
|
||||||
|
game_type: list = Query(default=None),
|
||||||
|
ranked: Optional[bool] = None,
|
||||||
|
short_game: Optional[bool] = None,
|
||||||
|
csv: Optional[bool] = False,
|
||||||
|
short_output: bool = False,
|
||||||
|
gauntlet_id: Optional[int] = None,
|
||||||
|
limit: int = 100,
|
||||||
|
):
|
||||||
all_games = StratGame.select().order_by(StratGame.id)
|
all_games = StratGame.select().order_by(StratGame.id)
|
||||||
|
|
||||||
if season is not None:
|
if season is not None:
|
||||||
@ -68,49 +74,70 @@ async def get_games(
|
|||||||
if short_game is not None:
|
if short_game is not None:
|
||||||
all_games = all_games.where(StratGame.short_game == short_game)
|
all_games = all_games.where(StratGame.short_game == short_game)
|
||||||
if gauntlet_id is not None:
|
if gauntlet_id is not None:
|
||||||
all_games = all_games.where(StratGame.game_type.contains(f'gauntlet-{gauntlet_id}'))
|
all_games = all_games.where(
|
||||||
|
StratGame.game_type.contains(f"gauntlet-{gauntlet_id}")
|
||||||
|
)
|
||||||
|
|
||||||
|
all_games = all_games.limit(max(0, min(limit, 500)))
|
||||||
|
|
||||||
if csv:
|
if csv:
|
||||||
return_vals = [model_to_dict(x) for x in all_games]
|
return_vals = [model_to_dict(x) for x in all_games]
|
||||||
for x in return_vals:
|
for x in return_vals:
|
||||||
x['away_abbrev'] = x['away_team']['abbrev']
|
x["away_abbrev"] = x["away_team"]["abbrev"]
|
||||||
x['home_abbrev'] = x['home_team']['abbrev']
|
x["home_abbrev"] = x["home_team"]["abbrev"]
|
||||||
del x['away_team'], x['home_team']
|
del x["away_team"], x["home_team"]
|
||||||
|
|
||||||
output = pd.DataFrame(return_vals)[[
|
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'
|
"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 Response(content=output.to_csv(index=False), media_type="text/csv")
|
||||||
|
|
||||||
return_val = {'count': all_games.count(), 'games': [
|
return_val = {
|
||||||
model_to_dict(x, recurse=not short_output) for x in all_games
|
"count": all_games.count(),
|
||||||
]}
|
"games": [model_to_dict(x, recurse=not short_output) for x in all_games],
|
||||||
|
}
|
||||||
return return_val
|
return return_val
|
||||||
|
|
||||||
|
|
||||||
@router.get('/{game_id}')
|
@router.get("/{game_id}")
|
||||||
async def get_one_game(game_id: int):
|
async def get_one_game(game_id: int):
|
||||||
this_game = StratGame.get_or_none(StratGame.id == game_id)
|
this_game = StratGame.get_or_none(StratGame.id == game_id)
|
||||||
if not this_game:
|
if not this_game:
|
||||||
raise HTTPException(status_code=404, detail=f'StratGame ID {game_id} not found')
|
raise HTTPException(status_code=404, detail=f"StratGame ID {game_id} not found")
|
||||||
|
|
||||||
g_result = model_to_dict(this_game)
|
g_result = model_to_dict(this_game)
|
||||||
return g_result
|
return g_result
|
||||||
|
|
||||||
|
|
||||||
@router.patch('/{game_id}')
|
@router.patch("/{game_id}")
|
||||||
async def patch_game(
|
async def patch_game(
|
||||||
game_id: int, game_type: Optional[str] = None, away_score: Optional[int] = None,
|
game_id: int,
|
||||||
home_score: Optional[int] = None, token: str = Depends(oauth2_scheme)):
|
game_type: Optional[str] = None,
|
||||||
|
away_score: Optional[int] = None,
|
||||||
|
home_score: Optional[int] = None,
|
||||||
|
token: str = Depends(oauth2_scheme),
|
||||||
|
):
|
||||||
if not valid_token(token):
|
if not valid_token(token):
|
||||||
logging.warning('patch_game - Bad Token: [REDACTED]')
|
logging.warning("patch_game - Bad Token: [REDACTED]")
|
||||||
raise HTTPException(status_code=401, detail='Unauthorized')
|
raise HTTPException(status_code=401, detail="Unauthorized")
|
||||||
|
|
||||||
this_game = StratGame.get_or_none(StratGame.id == game_id)
|
this_game = StratGame.get_or_none(StratGame.id == game_id)
|
||||||
if not this_game:
|
if not this_game:
|
||||||
raise HTTPException(status_code=404, detail=f'StratGame ID {game_id} not found')
|
raise HTTPException(status_code=404, detail=f"StratGame ID {game_id} not found")
|
||||||
|
|
||||||
if away_score is not None:
|
if away_score is not None:
|
||||||
this_game.away_score = away_score
|
this_game.away_score = away_score
|
||||||
@ -123,14 +150,14 @@ async def patch_game(
|
|||||||
g_result = model_to_dict(this_game)
|
g_result = model_to_dict(this_game)
|
||||||
return g_result
|
return g_result
|
||||||
else:
|
else:
|
||||||
raise HTTPException(status_code=500, detail=f'Unable to patch game {game_id}')
|
raise HTTPException(status_code=500, detail=f"Unable to patch game {game_id}")
|
||||||
|
|
||||||
|
|
||||||
@router.post('')
|
@router.post("")
|
||||||
async def post_game(this_game: GameModel, token: str = Depends(oauth2_scheme)):
|
async def post_game(this_game: GameModel, token: str = Depends(oauth2_scheme)):
|
||||||
if not valid_token(token):
|
if not valid_token(token):
|
||||||
logging.warning('post_games - Bad Token: [REDACTED]')
|
logging.warning("post_games - Bad Token: [REDACTED]")
|
||||||
raise HTTPException(status_code=401, detail='Unauthorized')
|
raise HTTPException(status_code=401, detail="Unauthorized")
|
||||||
|
|
||||||
this_game = StratGame(**this_game.dict())
|
this_game = StratGame(**this_game.dict())
|
||||||
|
|
||||||
@ -141,25 +168,25 @@ async def post_game(this_game: GameModel, token: str = Depends(oauth2_scheme)):
|
|||||||
else:
|
else:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
status_code=418,
|
status_code=418,
|
||||||
detail='Well slap my ass and call me a teapot; I could not save that game'
|
detail="Well slap my ass and call me a teapot; I could not save that game",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@router.delete('/{game_id}')
|
@router.delete("/{game_id}")
|
||||||
async def delete_game(game_id: int, token: str = Depends(oauth2_scheme)):
|
async def delete_game(game_id: int, token: str = Depends(oauth2_scheme)):
|
||||||
if not valid_token(token):
|
if not valid_token(token):
|
||||||
logging.warning('delete_game - Bad Token: [REDACTED]')
|
logging.warning("delete_game - Bad Token: [REDACTED]")
|
||||||
raise HTTPException(status_code=401, detail='Unauthorized')
|
raise HTTPException(status_code=401, detail="Unauthorized")
|
||||||
|
|
||||||
this_game = StratGame.get_or_none(StratGame.id == game_id)
|
this_game = StratGame.get_or_none(StratGame.id == game_id)
|
||||||
if not this_game:
|
if not this_game:
|
||||||
raise HTTPException(status_code=404, detail=f'StratGame ID {game_id} not found')
|
raise HTTPException(status_code=404, detail=f"StratGame ID {game_id} not found")
|
||||||
|
|
||||||
count = this_game.delete_instance()
|
count = this_game.delete_instance()
|
||||||
|
|
||||||
if count == 1:
|
if count == 1:
|
||||||
return f'StratGame {game_id} has been deleted'
|
return f"StratGame {game_id} has been deleted"
|
||||||
else:
|
else:
|
||||||
raise HTTPException(status_code=500, detail=f'StratGame {game_id} could not be deleted')
|
raise HTTPException(
|
||||||
|
status_code=500, detail=f"StratGame {game_id} could not be deleted"
|
||||||
|
)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user