Merge pull request 'Fix unbounded API queries causing worker timeouts' (#99) from bugfix/limit-caps into main
All checks were successful
Build Docker Image / build (push) Successful in 1m9s
All checks were successful
Build Docker Image / build (push) Successful in 1m9s
Reviewed-on: #99
This commit is contained in:
commit
81baa54681
@ -57,6 +57,9 @@ priv_help = (
|
|||||||
)
|
)
|
||||||
PRIVATE_IN_SCHEMA = True if priv_help == "TRUE" else False
|
PRIVATE_IN_SCHEMA = True if priv_help == "TRUE" else False
|
||||||
|
|
||||||
|
MAX_LIMIT = 500
|
||||||
|
DEFAULT_LIMIT = 200
|
||||||
|
|
||||||
|
|
||||||
def valid_token(token):
|
def valid_token(token):
|
||||||
return token == os.environ.get("API_TOKEN")
|
return token == os.environ.get("API_TOKEN")
|
||||||
|
|||||||
14
app/main.py
14
app/main.py
@ -2,6 +2,7 @@ import datetime
|
|||||||
import logging
|
import logging
|
||||||
from logging.handlers import RotatingFileHandler
|
from logging.handlers import RotatingFileHandler
|
||||||
import os
|
import os
|
||||||
|
from urllib.parse import parse_qsl, urlencode
|
||||||
|
|
||||||
from fastapi import Depends, FastAPI, Request
|
from fastapi import Depends, FastAPI, Request
|
||||||
from fastapi.openapi.docs import get_swagger_ui_html
|
from fastapi.openapi.docs import get_swagger_ui_html
|
||||||
@ -70,6 +71,19 @@ app = FastAPI(
|
|||||||
logger.info(f"Starting up now...")
|
logger.info(f"Starting up now...")
|
||||||
|
|
||||||
|
|
||||||
|
@app.middleware("http")
|
||||||
|
async def strip_empty_query_params(request: Request, call_next):
|
||||||
|
qs = request.scope.get("query_string", b"")
|
||||||
|
if qs:
|
||||||
|
pairs = parse_qsl(qs.decode(), keep_blank_values=True)
|
||||||
|
filtered = [(k, v) for k, v in pairs if v != ""]
|
||||||
|
new_qs = urlencode(filtered).encode()
|
||||||
|
request.scope["query_string"] = new_qs
|
||||||
|
if hasattr(request, "_query_params"):
|
||||||
|
del request._query_params
|
||||||
|
return await call_next(request)
|
||||||
|
|
||||||
|
|
||||||
app.include_router(current.router)
|
app.include_router(current.router)
|
||||||
app.include_router(players.router)
|
app.include_router(players.router)
|
||||||
app.include_router(results.router)
|
app.include_router(results.router)
|
||||||
|
|||||||
@ -9,6 +9,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -43,6 +45,8 @@ async def get_awards(
|
|||||||
team_id: list = Query(default=None),
|
team_id: list = Query(default=None),
|
||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
player_name: list = Query(default=None),
|
player_name: list = Query(default=None),
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
all_awards = Award.select()
|
all_awards = Award.select()
|
||||||
|
|
||||||
@ -67,8 +71,11 @@ async def get_awards(
|
|||||||
all_players = Player.select().where(fn.Lower(Player.name) << pname_list)
|
all_players = Player.select().where(fn.Lower(Player.name) << pname_list)
|
||||||
all_awards = all_awards.where(Award.player << all_players)
|
all_awards = all_awards.where(Award.player << all_players)
|
||||||
|
|
||||||
|
total_count = all_awards.count()
|
||||||
|
all_awards = all_awards.offset(offset).limit(limit)
|
||||||
|
|
||||||
return_awards = {
|
return_awards = {
|
||||||
"count": all_awards.count(),
|
"count": total_count,
|
||||||
"awards": [model_to_dict(x, recurse=not short_output) for x in all_awards],
|
"awards": [model_to_dict(x, recurse=not short_output) for x in all_awards],
|
||||||
}
|
}
|
||||||
db.close()
|
db.close()
|
||||||
|
|||||||
@ -19,6 +19,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -84,7 +86,7 @@ async def get_batstats(
|
|||||||
week_end: Optional[int] = None,
|
week_end: Optional[int] = None,
|
||||||
game_num: list = Query(default=None),
|
game_num: list = Query(default=None),
|
||||||
position: list = Query(default=None),
|
position: list = Query(default=None),
|
||||||
limit: Optional[int] = None,
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
sort: Optional[str] = None,
|
sort: Optional[str] = None,
|
||||||
short_output: Optional[bool] = True,
|
short_output: Optional[bool] = True,
|
||||||
):
|
):
|
||||||
@ -134,8 +136,7 @@ async def get_batstats(
|
|||||||
)
|
)
|
||||||
all_stats = all_stats.where((BattingStat.week >= start) & (BattingStat.week <= end))
|
all_stats = all_stats.where((BattingStat.week >= start) & (BattingStat.week <= end))
|
||||||
|
|
||||||
if limit:
|
all_stats = all_stats.limit(limit)
|
||||||
all_stats = all_stats.limit(limit)
|
|
||||||
if sort:
|
if sort:
|
||||||
if sort == "newest":
|
if sort == "newest":
|
||||||
all_stats = all_stats.order_by(-BattingStat.week, -BattingStat.game)
|
all_stats = all_stats.order_by(-BattingStat.week, -BattingStat.game)
|
||||||
@ -168,6 +169,8 @@ async def get_totalstats(
|
|||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
min_pa: Optional[int] = 1,
|
min_pa: Optional[int] = 1,
|
||||||
week: list = Query(default=None),
|
week: list = Query(default=None),
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
if sum(1 for x in [s_type, (week_start or week_end), week] if x is not None) > 1:
|
if sum(1 for x in [s_type, (week_start or week_end), week] if x is not None) > 1:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
@ -301,7 +304,10 @@ async def get_totalstats(
|
|||||||
all_players = Player.select().where(Player.id << player_id)
|
all_players = Player.select().where(Player.id << player_id)
|
||||||
all_stats = all_stats.where(BattingStat.player << all_players)
|
all_stats = all_stats.where(BattingStat.player << all_players)
|
||||||
|
|
||||||
return_stats = {"count": all_stats.count(), "stats": []}
|
total_count = all_stats.count()
|
||||||
|
all_stats = all_stats.offset(offset).limit(limit)
|
||||||
|
|
||||||
|
return_stats = {"count": total_count, "stats": []}
|
||||||
|
|
||||||
for x in all_stats:
|
for x in all_stats:
|
||||||
# Handle player field based on grouping with safe access
|
# Handle player field based on grouping with safe access
|
||||||
|
|||||||
@ -19,6 +19,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -73,7 +75,7 @@ async def get_decisions(
|
|||||||
irunners_scored: list = Query(default=None),
|
irunners_scored: list = Query(default=None),
|
||||||
game_id: list = Query(default=None),
|
game_id: list = Query(default=None),
|
||||||
player_id: list = Query(default=None),
|
player_id: list = Query(default=None),
|
||||||
limit: Optional[int] = None,
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
):
|
):
|
||||||
all_dec = Decision.select().order_by(
|
all_dec = Decision.select().order_by(
|
||||||
@ -135,10 +137,7 @@ async def get_decisions(
|
|||||||
if irunners_scored is not None:
|
if irunners_scored is not None:
|
||||||
all_dec = all_dec.where(Decision.irunners_scored << irunners_scored)
|
all_dec = all_dec.where(Decision.irunners_scored << irunners_scored)
|
||||||
|
|
||||||
if limit is not None:
|
all_dec = all_dec.limit(limit)
|
||||||
if limit < 1:
|
|
||||||
limit = 1
|
|
||||||
all_dec = all_dec.limit(limit)
|
|
||||||
|
|
||||||
return_dec = {
|
return_dec = {
|
||||||
"count": all_dec.count(),
|
"count": all_dec.count(),
|
||||||
|
|||||||
@ -9,6 +9,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -32,6 +34,8 @@ async def get_divisions(
|
|||||||
div_abbrev: Optional[str] = None,
|
div_abbrev: Optional[str] = None,
|
||||||
lg_name: Optional[str] = None,
|
lg_name: Optional[str] = None,
|
||||||
lg_abbrev: Optional[str] = None,
|
lg_abbrev: Optional[str] = None,
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
all_divisions = Division.select().where(Division.season == season)
|
all_divisions = Division.select().where(Division.season == season)
|
||||||
|
|
||||||
@ -44,8 +48,11 @@ async def get_divisions(
|
|||||||
if lg_abbrev is not None:
|
if lg_abbrev is not None:
|
||||||
all_divisions = all_divisions.where(Division.league_abbrev == lg_abbrev)
|
all_divisions = all_divisions.where(Division.league_abbrev == lg_abbrev)
|
||||||
|
|
||||||
|
total_count = all_divisions.count()
|
||||||
|
all_divisions = all_divisions.offset(offset).limit(limit)
|
||||||
|
|
||||||
return_div = {
|
return_div = {
|
||||||
"count": all_divisions.count(),
|
"count": total_count,
|
||||||
"divisions": [model_to_dict(x) for x in all_divisions],
|
"divisions": [model_to_dict(x) for x in all_divisions],
|
||||||
}
|
}
|
||||||
db.close()
|
db.close()
|
||||||
|
|||||||
@ -9,6 +9,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -34,6 +36,8 @@ async def get_draftlist(
|
|||||||
season: Optional[int],
|
season: Optional[int],
|
||||||
team_id: list = Query(default=None),
|
team_id: list = Query(default=None),
|
||||||
token: str = Depends(oauth2_scheme),
|
token: str = Depends(oauth2_scheme),
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
if not valid_token(token):
|
if not valid_token(token):
|
||||||
logger.warning(f"get_draftlist - Bad Token: {token}")
|
logger.warning(f"get_draftlist - Bad Token: {token}")
|
||||||
@ -46,7 +50,10 @@ async def get_draftlist(
|
|||||||
if team_id is not None:
|
if team_id is not None:
|
||||||
all_list = all_list.where(DraftList.team_id << team_id)
|
all_list = all_list.where(DraftList.team_id << team_id)
|
||||||
|
|
||||||
r_list = {"count": all_list.count(), "picks": [model_to_dict(x) for x in all_list]}
|
total_count = all_list.count()
|
||||||
|
all_list = all_list.offset(offset).limit(limit)
|
||||||
|
|
||||||
|
r_list = {"count": total_count, "picks": [model_to_dict(x) for x in all_list]}
|
||||||
|
|
||||||
db.close()
|
db.close()
|
||||||
return r_list
|
return r_list
|
||||||
|
|||||||
@ -9,6 +9,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -50,7 +52,7 @@ async def get_picks(
|
|||||||
overall_end: Optional[int] = None,
|
overall_end: Optional[int] = None,
|
||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
sort: Optional[str] = None,
|
sort: Optional[str] = None,
|
||||||
limit: Optional[int] = None,
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
player_id: list = Query(default=None),
|
player_id: list = Query(default=None),
|
||||||
player_taken: Optional[bool] = None,
|
player_taken: Optional[bool] = None,
|
||||||
):
|
):
|
||||||
@ -110,8 +112,7 @@ async def get_picks(
|
|||||||
all_picks = all_picks.where(DraftPick.overall <= overall_end)
|
all_picks = all_picks.where(DraftPick.overall <= overall_end)
|
||||||
if player_taken is not None:
|
if player_taken is not None:
|
||||||
all_picks = all_picks.where(DraftPick.player.is_null(not player_taken))
|
all_picks = all_picks.where(DraftPick.player.is_null(not player_taken))
|
||||||
if limit is not None:
|
all_picks = all_picks.limit(limit)
|
||||||
all_picks = all_picks.limit(limit)
|
|
||||||
|
|
||||||
if sort is not None:
|
if sort is not None:
|
||||||
if sort == "order-asc":
|
if sort == "order-asc":
|
||||||
|
|||||||
@ -3,40 +3,61 @@ from typing import List, Optional, Literal
|
|||||||
import logging
|
import logging
|
||||||
import pydantic
|
import pydantic
|
||||||
|
|
||||||
from ..db_engine import db, BattingStat, Team, Player, Current, model_to_dict, chunked, fn, per_season_weeks
|
from ..db_engine import (
|
||||||
from ..dependencies import oauth2_scheme, valid_token, handle_db_errors
|
db,
|
||||||
|
BattingStat,
|
||||||
logger = logging.getLogger('discord_app')
|
Team,
|
||||||
|
Player,
|
||||||
router = APIRouter(
|
Current,
|
||||||
prefix='/api/v3/fieldingstats',
|
model_to_dict,
|
||||||
tags=['fieldingstats']
|
chunked,
|
||||||
|
fn,
|
||||||
|
per_season_weeks,
|
||||||
|
)
|
||||||
|
from ..dependencies import (
|
||||||
|
oauth2_scheme,
|
||||||
|
valid_token,
|
||||||
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
logger = logging.getLogger("discord_app")
|
||||||
|
|
||||||
@router.get('')
|
router = APIRouter(prefix="/api/v3/fieldingstats", tags=["fieldingstats"])
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("")
|
||||||
@handle_db_errors
|
@handle_db_errors
|
||||||
async def get_fieldingstats(
|
async def get_fieldingstats(
|
||||||
season: int, s_type: Optional[str] = 'regular', team_abbrev: list = Query(default=None),
|
season: int,
|
||||||
player_name: list = Query(default=None), player_id: list = Query(default=None),
|
s_type: Optional[str] = "regular",
|
||||||
week_start: Optional[int] = None, week_end: Optional[int] = None, game_num: list = Query(default=None),
|
team_abbrev: list = Query(default=None),
|
||||||
position: list = Query(default=None), limit: Optional[int] = None, sort: Optional[str] = None,
|
player_name: list = Query(default=None),
|
||||||
short_output: Optional[bool] = True):
|
player_id: list = Query(default=None),
|
||||||
if 'post' in s_type.lower():
|
week_start: Optional[int] = None,
|
||||||
|
week_end: Optional[int] = None,
|
||||||
|
game_num: list = Query(default=None),
|
||||||
|
position: list = Query(default=None),
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
sort: Optional[str] = None,
|
||||||
|
short_output: Optional[bool] = True,
|
||||||
|
):
|
||||||
|
if "post" in s_type.lower():
|
||||||
all_stats = BattingStat.post_season(season)
|
all_stats = BattingStat.post_season(season)
|
||||||
if all_stats.count() == 0:
|
if all_stats.count() == 0:
|
||||||
db.close()
|
db.close()
|
||||||
return {'count': 0, 'stats': []}
|
return {"count": 0, "stats": []}
|
||||||
elif s_type.lower() in ['combined', 'total', 'all']:
|
elif s_type.lower() in ["combined", "total", "all"]:
|
||||||
all_stats = BattingStat.combined_season(season)
|
all_stats = BattingStat.combined_season(season)
|
||||||
if all_stats.count() == 0:
|
if all_stats.count() == 0:
|
||||||
db.close()
|
db.close()
|
||||||
return {'count': 0, 'stats': []}
|
return {"count": 0, "stats": []}
|
||||||
else:
|
else:
|
||||||
all_stats = BattingStat.regular_season(season)
|
all_stats = BattingStat.regular_season(season)
|
||||||
if all_stats.count() == 0:
|
if all_stats.count() == 0:
|
||||||
db.close()
|
db.close()
|
||||||
return {'count': 0, 'stats': []}
|
return {"count": 0, "stats": []}
|
||||||
|
|
||||||
all_stats = all_stats.where(
|
all_stats = all_stats.where(
|
||||||
(BattingStat.xch > 0) | (BattingStat.pb > 0) | (BattingStat.sbc > 0)
|
(BattingStat.xch > 0) | (BattingStat.pb > 0) | (BattingStat.sbc > 0)
|
||||||
@ -51,7 +72,9 @@ async def get_fieldingstats(
|
|||||||
if player_id:
|
if player_id:
|
||||||
all_stats = all_stats.where(BattingStat.player_id << player_id)
|
all_stats = all_stats.where(BattingStat.player_id << player_id)
|
||||||
else:
|
else:
|
||||||
p_query = Player.select_season(season).where(fn.Lower(Player.name) << [x.lower() for x in player_name])
|
p_query = Player.select_season(season).where(
|
||||||
|
fn.Lower(Player.name) << [x.lower() for x in player_name]
|
||||||
|
)
|
||||||
all_stats = all_stats.where(BattingStat.player << p_query)
|
all_stats = all_stats.where(BattingStat.player << p_query)
|
||||||
if game_num:
|
if game_num:
|
||||||
all_stats = all_stats.where(BattingStat.game == game_num)
|
all_stats = all_stats.where(BattingStat.game == game_num)
|
||||||
@ -66,72 +89,92 @@ async def get_fieldingstats(
|
|||||||
db.close()
|
db.close()
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
status_code=404,
|
status_code=404,
|
||||||
detail=f'Start week {start} is after end week {end} - cannot pull stats'
|
detail=f"Start week {start} is after end week {end} - cannot pull stats",
|
||||||
)
|
)
|
||||||
all_stats = all_stats.where(
|
all_stats = all_stats.where((BattingStat.week >= start) & (BattingStat.week <= end))
|
||||||
(BattingStat.week >= start) & (BattingStat.week <= end)
|
|
||||||
)
|
|
||||||
|
|
||||||
if limit:
|
total_count = all_stats.count()
|
||||||
all_stats = all_stats.limit(limit)
|
all_stats = all_stats.limit(limit)
|
||||||
if sort:
|
if sort:
|
||||||
if sort == 'newest':
|
if sort == "newest":
|
||||||
all_stats = all_stats.order_by(-BattingStat.week, -BattingStat.game)
|
all_stats = all_stats.order_by(-BattingStat.week, -BattingStat.game)
|
||||||
|
|
||||||
return_stats = {
|
return_stats = {
|
||||||
'count': all_stats.count(),
|
"count": total_count,
|
||||||
'stats': [{
|
"stats": [
|
||||||
'player': x.player_id if short_output else model_to_dict(x.player, recurse=False),
|
{
|
||||||
'team': x.team_id if short_output else model_to_dict(x.team, recurse=False),
|
"player": x.player_id
|
||||||
'pos': x.pos,
|
if short_output
|
||||||
'xch': x.xch,
|
else model_to_dict(x.player, recurse=False),
|
||||||
'xhit': x.xhit,
|
"team": x.team_id
|
||||||
'error': x.error,
|
if short_output
|
||||||
'pb': x.pb,
|
else model_to_dict(x.team, recurse=False),
|
||||||
'sbc': x.sbc,
|
"pos": x.pos,
|
||||||
'csc': x.csc,
|
"xch": x.xch,
|
||||||
'week': x.week,
|
"xhit": x.xhit,
|
||||||
'game': x.game,
|
"error": x.error,
|
||||||
'season': x.season
|
"pb": x.pb,
|
||||||
} for x in all_stats]
|
"sbc": x.sbc,
|
||||||
|
"csc": x.csc,
|
||||||
|
"week": x.week,
|
||||||
|
"game": x.game,
|
||||||
|
"season": x.season,
|
||||||
|
}
|
||||||
|
for x in all_stats
|
||||||
|
],
|
||||||
}
|
}
|
||||||
|
|
||||||
db.close()
|
db.close()
|
||||||
return return_stats
|
return return_stats
|
||||||
|
|
||||||
|
|
||||||
@router.get('/totals')
|
@router.get("/totals")
|
||||||
@handle_db_errors
|
@handle_db_errors
|
||||||
async def get_totalstats(
|
async def get_totalstats(
|
||||||
season: int, s_type: Literal['regular', 'post', 'total', None] = None, team_abbrev: list = Query(default=None),
|
season: int,
|
||||||
team_id: list = Query(default=None), player_name: list = Query(default=None),
|
s_type: Literal["regular", "post", "total", None] = None,
|
||||||
week_start: Optional[int] = None, week_end: Optional[int] = None, game_num: list = Query(default=None),
|
team_abbrev: list = Query(default=None),
|
||||||
position: list = Query(default=None), sort: Optional[str] = None, player_id: list = Query(default=None),
|
team_id: list = Query(default=None),
|
||||||
group_by: Literal['team', 'player', 'playerteam'] = 'player', short_output: Optional[bool] = False,
|
player_name: list = Query(default=None),
|
||||||
min_ch: Optional[int] = 1, week: list = Query(default=None)):
|
week_start: Optional[int] = None,
|
||||||
|
week_end: Optional[int] = None,
|
||||||
|
game_num: list = Query(default=None),
|
||||||
|
position: list = Query(default=None),
|
||||||
|
sort: Optional[str] = None,
|
||||||
|
player_id: list = Query(default=None),
|
||||||
|
group_by: Literal["team", "player", "playerteam"] = "player",
|
||||||
|
short_output: Optional[bool] = False,
|
||||||
|
min_ch: Optional[int] = 1,
|
||||||
|
week: list = Query(default=None),
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
|
):
|
||||||
|
|
||||||
# Build SELECT fields conditionally based on group_by to match GROUP BY exactly
|
# Build SELECT fields conditionally based on group_by to match GROUP BY exactly
|
||||||
select_fields = []
|
select_fields = []
|
||||||
|
|
||||||
if group_by == 'player':
|
if group_by == "player":
|
||||||
select_fields = [BattingStat.player, BattingStat.pos]
|
select_fields = [BattingStat.player, BattingStat.pos]
|
||||||
elif group_by == 'team':
|
elif group_by == "team":
|
||||||
select_fields = [BattingStat.team, BattingStat.pos]
|
select_fields = [BattingStat.team, BattingStat.pos]
|
||||||
elif group_by == 'playerteam':
|
elif group_by == "playerteam":
|
||||||
select_fields = [BattingStat.player, BattingStat.team, BattingStat.pos]
|
select_fields = [BattingStat.player, BattingStat.team, BattingStat.pos]
|
||||||
else:
|
else:
|
||||||
# Default case
|
# Default case
|
||||||
select_fields = [BattingStat.player, BattingStat.pos]
|
select_fields = [BattingStat.player, BattingStat.pos]
|
||||||
|
|
||||||
all_stats = (
|
all_stats = (
|
||||||
BattingStat
|
BattingStat.select(
|
||||||
.select(*select_fields,
|
*select_fields,
|
||||||
fn.SUM(BattingStat.xch).alias('sum_xch'),
|
fn.SUM(BattingStat.xch).alias("sum_xch"),
|
||||||
fn.SUM(BattingStat.xhit).alias('sum_xhit'), fn.SUM(BattingStat.error).alias('sum_error'),
|
fn.SUM(BattingStat.xhit).alias("sum_xhit"),
|
||||||
fn.SUM(BattingStat.pb).alias('sum_pb'), fn.SUM(BattingStat.sbc).alias('sum_sbc'),
|
fn.SUM(BattingStat.error).alias("sum_error"),
|
||||||
fn.SUM(BattingStat.csc).alias('sum_csc'))
|
fn.SUM(BattingStat.pb).alias("sum_pb"),
|
||||||
.where(BattingStat.season == season)
|
fn.SUM(BattingStat.sbc).alias("sum_sbc"),
|
||||||
.having(fn.SUM(BattingStat.xch) >= min_ch)
|
fn.SUM(BattingStat.csc).alias("sum_csc"),
|
||||||
|
)
|
||||||
|
.where(BattingStat.season == season)
|
||||||
|
.having(fn.SUM(BattingStat.xch) >= min_ch)
|
||||||
)
|
)
|
||||||
|
|
||||||
if True in [s_type is not None, week_start is not None, week_end is not None]:
|
if True in [s_type is not None, week_start is not None, week_end is not None]:
|
||||||
@ -141,16 +184,20 @@ async def get_totalstats(
|
|||||||
elif week_start is not None or week_end is not None:
|
elif week_start is not None or week_end is not None:
|
||||||
if week_start is None or week_end is None:
|
if week_start is None or week_end is None:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
status_code=400, detail='Both week_start and week_end must be included if either is used.'
|
status_code=400,
|
||||||
|
detail="Both week_start and week_end must be included if either is used.",
|
||||||
|
)
|
||||||
|
weeks["start"] = week_start
|
||||||
|
if week_end < weeks["start"]:
|
||||||
|
raise HTTPException(
|
||||||
|
status_code=400,
|
||||||
|
detail="week_end must be greater than or equal to week_start",
|
||||||
)
|
)
|
||||||
weeks['start'] = week_start
|
|
||||||
if week_end < weeks['start']:
|
|
||||||
raise HTTPException(status_code=400, detail='week_end must be greater than or equal to week_start')
|
|
||||||
else:
|
else:
|
||||||
weeks['end'] = week_end
|
weeks["end"] = week_end
|
||||||
|
|
||||||
all_stats = all_stats.where(
|
all_stats = all_stats.where(
|
||||||
(BattingStat.week >= weeks['start']) & (BattingStat.week <= weeks['end'])
|
(BattingStat.week >= weeks["start"]) & (BattingStat.week <= weeks["end"])
|
||||||
)
|
)
|
||||||
|
|
||||||
elif week is not None:
|
elif week is not None:
|
||||||
@ -161,14 +208,20 @@ async def get_totalstats(
|
|||||||
if position is not None:
|
if position is not None:
|
||||||
p_list = [x.upper() for x in position]
|
p_list = [x.upper() for x in position]
|
||||||
all_players = Player.select().where(
|
all_players = Player.select().where(
|
||||||
(Player.pos_1 << p_list) | (Player.pos_2 << p_list) | (Player.pos_3 << p_list) | (Player.pos_4 << p_list) |
|
(Player.pos_1 << p_list)
|
||||||
(Player.pos_5 << p_list) | (Player.pos_6 << p_list) | (Player.pos_7 << p_list) | (Player.pos_8 << p_list)
|
| (Player.pos_2 << p_list)
|
||||||
|
| (Player.pos_3 << p_list)
|
||||||
|
| (Player.pos_4 << p_list)
|
||||||
|
| (Player.pos_5 << p_list)
|
||||||
|
| (Player.pos_6 << p_list)
|
||||||
|
| (Player.pos_7 << p_list)
|
||||||
|
| (Player.pos_8 << p_list)
|
||||||
)
|
)
|
||||||
all_stats = all_stats.where(BattingStat.player << all_players)
|
all_stats = all_stats.where(BattingStat.player << all_players)
|
||||||
if sort is not None:
|
if sort is not None:
|
||||||
if sort == 'player':
|
if sort == "player":
|
||||||
all_stats = all_stats.order_by(BattingStat.player)
|
all_stats = all_stats.order_by(BattingStat.player)
|
||||||
elif sort == 'team':
|
elif sort == "team":
|
||||||
all_stats = all_stats.order_by(BattingStat.team)
|
all_stats = all_stats.order_by(BattingStat.team)
|
||||||
if group_by is not None:
|
if group_by is not None:
|
||||||
# Use the same fields for GROUP BY as we used for SELECT
|
# Use the same fields for GROUP BY as we used for SELECT
|
||||||
@ -177,47 +230,57 @@ async def get_totalstats(
|
|||||||
all_teams = Team.select().where(Team.id << team_id)
|
all_teams = Team.select().where(Team.id << team_id)
|
||||||
all_stats = all_stats.where(BattingStat.team << all_teams)
|
all_stats = all_stats.where(BattingStat.team << all_teams)
|
||||||
elif team_abbrev is not None:
|
elif team_abbrev is not None:
|
||||||
all_teams = Team.select().where(fn.Lower(Team.abbrev) << [x.lower() for x in team_abbrev])
|
all_teams = Team.select().where(
|
||||||
|
fn.Lower(Team.abbrev) << [x.lower() for x in team_abbrev]
|
||||||
|
)
|
||||||
all_stats = all_stats.where(BattingStat.team << all_teams)
|
all_stats = all_stats.where(BattingStat.team << all_teams)
|
||||||
|
|
||||||
if player_name is not None:
|
if player_name is not None:
|
||||||
all_players = Player.select().where(fn.Lower(Player.name) << [x.lower() for x in player_name])
|
all_players = Player.select().where(
|
||||||
|
fn.Lower(Player.name) << [x.lower() for x in player_name]
|
||||||
|
)
|
||||||
all_stats = all_stats.where(BattingStat.player << all_players)
|
all_stats = all_stats.where(BattingStat.player << all_players)
|
||||||
elif player_id is not None:
|
elif player_id is not None:
|
||||||
all_players = Player.select().where(Player.id << player_id)
|
all_players = Player.select().where(Player.id << player_id)
|
||||||
all_stats = all_stats.where(BattingStat.player << all_players)
|
all_stats = all_stats.where(BattingStat.player << all_players)
|
||||||
|
|
||||||
return_stats = {
|
total_count = all_stats.count()
|
||||||
'count': 0,
|
all_stats = all_stats.offset(offset).limit(limit)
|
||||||
'stats': []
|
|
||||||
}
|
return_stats = {"count": total_count, "stats": []}
|
||||||
|
|
||||||
for x in all_stats:
|
for x in all_stats:
|
||||||
if x.sum_xch + x.sum_sbc <= 0:
|
if x.sum_xch + x.sum_sbc <= 0:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# Handle player field based on grouping with safe access
|
|
||||||
this_player = 'TOT'
|
|
||||||
if 'player' in group_by and hasattr(x, 'player'):
|
|
||||||
this_player = x.player_id if short_output else model_to_dict(x.player, recurse=False)
|
|
||||||
|
|
||||||
# Handle team field based on grouping with safe access
|
# Handle player field based on grouping with safe access
|
||||||
this_team = 'TOT'
|
this_player = "TOT"
|
||||||
if 'team' in group_by and hasattr(x, 'team'):
|
if "player" in group_by and hasattr(x, "player"):
|
||||||
this_team = x.team_id if short_output else model_to_dict(x.team, recurse=False)
|
this_player = (
|
||||||
|
x.player_id if short_output else model_to_dict(x.player, recurse=False)
|
||||||
return_stats['stats'].append({
|
)
|
||||||
'player': this_player,
|
|
||||||
'team': this_team,
|
# Handle team field based on grouping with safe access
|
||||||
'pos': x.pos,
|
this_team = "TOT"
|
||||||
'xch': x.sum_xch,
|
if "team" in group_by and hasattr(x, "team"):
|
||||||
'xhit': x.sum_xhit,
|
this_team = (
|
||||||
'error': x.sum_error,
|
x.team_id if short_output else model_to_dict(x.team, recurse=False)
|
||||||
'pb': x.sum_pb,
|
)
|
||||||
'sbc': x.sum_sbc,
|
|
||||||
'csc': x.sum_csc
|
return_stats["stats"].append(
|
||||||
})
|
{
|
||||||
|
"player": this_player,
|
||||||
return_stats['count'] = len(return_stats['stats'])
|
"team": this_team,
|
||||||
|
"pos": x.pos,
|
||||||
|
"xch": x.sum_xch,
|
||||||
|
"xhit": x.sum_xhit,
|
||||||
|
"error": x.sum_error,
|
||||||
|
"pb": x.sum_pb,
|
||||||
|
"sbc": x.sum_sbc,
|
||||||
|
"csc": x.sum_csc,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
return_stats["count"] = len(return_stats["stats"])
|
||||||
db.close()
|
db.close()
|
||||||
return return_stats
|
return return_stats
|
||||||
|
|||||||
@ -9,6 +9,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -38,6 +40,8 @@ async def get_injuries(
|
|||||||
is_active: bool = None,
|
is_active: bool = None,
|
||||||
short_output: bool = False,
|
short_output: bool = False,
|
||||||
sort: Optional[str] = "start-asc",
|
sort: Optional[str] = "start-asc",
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
all_injuries = Injury.select()
|
all_injuries = Injury.select()
|
||||||
|
|
||||||
@ -64,8 +68,11 @@ async def get_injuries(
|
|||||||
elif sort == "start-desc":
|
elif sort == "start-desc":
|
||||||
all_injuries = all_injuries.order_by(-Injury.start_week, -Injury.start_game)
|
all_injuries = all_injuries.order_by(-Injury.start_week, -Injury.start_game)
|
||||||
|
|
||||||
|
total_count = all_injuries.count()
|
||||||
|
all_injuries = all_injuries.offset(offset).limit(limit)
|
||||||
|
|
||||||
return_injuries = {
|
return_injuries = {
|
||||||
"count": all_injuries.count(),
|
"count": total_count,
|
||||||
"injuries": [model_to_dict(x, recurse=not short_output) for x in all_injuries],
|
"injuries": [model_to_dict(x, recurse=not short_output) for x in all_injuries],
|
||||||
}
|
}
|
||||||
db.close()
|
db.close()
|
||||||
|
|||||||
@ -9,6 +9,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -34,6 +36,8 @@ async def get_keepers(
|
|||||||
team_id: list = Query(default=None),
|
team_id: list = Query(default=None),
|
||||||
player_id: list = Query(default=None),
|
player_id: list = Query(default=None),
|
||||||
short_output: bool = False,
|
short_output: bool = False,
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
all_keepers = Keeper.select()
|
all_keepers = Keeper.select()
|
||||||
|
|
||||||
@ -44,8 +48,11 @@ async def get_keepers(
|
|||||||
if player_id is not None:
|
if player_id is not None:
|
||||||
all_keepers = all_keepers.where(Keeper.player_id << player_id)
|
all_keepers = all_keepers.where(Keeper.player_id << player_id)
|
||||||
|
|
||||||
|
total_count = all_keepers.count()
|
||||||
|
all_keepers = all_keepers.offset(offset).limit(limit)
|
||||||
|
|
||||||
return_keepers = {
|
return_keepers = {
|
||||||
"count": all_keepers.count(),
|
"count": total_count,
|
||||||
"keepers": [model_to_dict(x, recurse=not short_output) for x in all_keepers],
|
"keepers": [model_to_dict(x, recurse=not short_output) for x in all_keepers],
|
||||||
}
|
}
|
||||||
db.close()
|
db.close()
|
||||||
|
|||||||
@ -9,6 +9,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -29,6 +31,8 @@ async def get_managers(
|
|||||||
name: list = Query(default=None),
|
name: list = Query(default=None),
|
||||||
active: Optional[bool] = None,
|
active: Optional[bool] = None,
|
||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
if active is not None:
|
if active is not None:
|
||||||
current = Current.latest()
|
current = Current.latest()
|
||||||
@ -61,7 +65,9 @@ async def get_managers(
|
|||||||
i_mgr.append(z)
|
i_mgr.append(z)
|
||||||
final_mgrs = [model_to_dict(y, recurse=not short_output) for y in i_mgr]
|
final_mgrs = [model_to_dict(y, recurse=not short_output) for y in i_mgr]
|
||||||
|
|
||||||
return_managers = {"count": len(final_mgrs), "managers": final_mgrs}
|
total_count = len(final_mgrs)
|
||||||
|
final_mgrs = final_mgrs[offset : offset + limit]
|
||||||
|
return_managers = {"count": total_count, "managers": final_mgrs}
|
||||||
|
|
||||||
else:
|
else:
|
||||||
all_managers = Manager.select()
|
all_managers = Manager.select()
|
||||||
@ -69,8 +75,10 @@ async def get_managers(
|
|||||||
name_list = [x.lower() for x in name]
|
name_list = [x.lower() for x in name]
|
||||||
all_managers = all_managers.where(fn.Lower(Manager.name) << name_list)
|
all_managers = all_managers.where(fn.Lower(Manager.name) << name_list)
|
||||||
|
|
||||||
|
total_count = all_managers.count()
|
||||||
|
all_managers = all_managers.offset(offset).limit(limit)
|
||||||
return_managers = {
|
return_managers = {
|
||||||
"count": all_managers.count(),
|
"count": total_count,
|
||||||
"managers": [
|
"managers": [
|
||||||
model_to_dict(x, recurse=not short_output) for x in all_managers
|
model_to_dict(x, recurse=not short_output) for x in all_managers
|
||||||
],
|
],
|
||||||
|
|||||||
@ -19,6 +19,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -68,7 +70,7 @@ async def get_pitstats(
|
|||||||
week_start: Optional[int] = None,
|
week_start: Optional[int] = None,
|
||||||
week_end: Optional[int] = None,
|
week_end: Optional[int] = None,
|
||||||
game_num: list = Query(default=None),
|
game_num: list = Query(default=None),
|
||||||
limit: Optional[int] = None,
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
ip_min: Optional[float] = None,
|
ip_min: Optional[float] = None,
|
||||||
sort: Optional[str] = None,
|
sort: Optional[str] = None,
|
||||||
short_output: Optional[bool] = True,
|
short_output: Optional[bool] = True,
|
||||||
@ -121,8 +123,7 @@ async def get_pitstats(
|
|||||||
(PitchingStat.week >= start) & (PitchingStat.week <= end)
|
(PitchingStat.week >= start) & (PitchingStat.week <= end)
|
||||||
)
|
)
|
||||||
|
|
||||||
if limit:
|
all_stats = all_stats.limit(limit)
|
||||||
all_stats = all_stats.limit(limit)
|
|
||||||
if sort:
|
if sort:
|
||||||
if sort == "newest":
|
if sort == "newest":
|
||||||
all_stats = all_stats.order_by(-PitchingStat.week, -PitchingStat.game)
|
all_stats = all_stats.order_by(-PitchingStat.week, -PitchingStat.game)
|
||||||
@ -154,6 +155,8 @@ async def get_totalstats(
|
|||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
group_by: Literal["team", "player", "playerteam"] = "player",
|
group_by: Literal["team", "player", "playerteam"] = "player",
|
||||||
week: list = Query(default=None),
|
week: list = Query(default=None),
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
if sum(1 for x in [s_type, (week_start or week_end), week] if x is not None) > 1:
|
if sum(1 for x in [s_type, (week_start or week_end), week] if x is not None) > 1:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
@ -259,7 +262,10 @@ async def get_totalstats(
|
|||||||
all_players = Player.select().where(Player.id << player_id)
|
all_players = Player.select().where(Player.id << player_id)
|
||||||
all_stats = all_stats.where(PitchingStat.player << all_players)
|
all_stats = all_stats.where(PitchingStat.player << all_players)
|
||||||
|
|
||||||
return_stats = {"count": all_stats.count(), "stats": []}
|
total_count = all_stats.count()
|
||||||
|
all_stats = all_stats.offset(offset).limit(limit)
|
||||||
|
|
||||||
|
return_stats = {"count": total_count, "stats": []}
|
||||||
|
|
||||||
for x in all_stats:
|
for x in all_stats:
|
||||||
# Handle player field based on grouping with safe access
|
# Handle player field based on grouping with safe access
|
||||||
|
|||||||
@ -6,7 +6,13 @@ Thin HTTP layer using PlayerService for business logic.
|
|||||||
from fastapi import APIRouter, Query, Response, Depends
|
from fastapi import APIRouter, Query, Response, Depends
|
||||||
from typing import Optional, List
|
from typing import Optional, List
|
||||||
|
|
||||||
from ..dependencies import oauth2_scheme, cache_result, handle_db_errors
|
from ..dependencies import (
|
||||||
|
oauth2_scheme,
|
||||||
|
cache_result,
|
||||||
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
|
)
|
||||||
from ..services.base import BaseService
|
from ..services.base import BaseService
|
||||||
from ..services.player_service import PlayerService
|
from ..services.player_service import PlayerService
|
||||||
|
|
||||||
@ -24,9 +30,7 @@ async def get_players(
|
|||||||
strat_code: list = Query(default=None),
|
strat_code: list = Query(default=None),
|
||||||
is_injured: Optional[bool] = None,
|
is_injured: Optional[bool] = None,
|
||||||
sort: Optional[str] = None,
|
sort: Optional[str] = None,
|
||||||
limit: Optional[int] = Query(
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
default=None, ge=1, description="Maximum number of results to return"
|
|
||||||
),
|
|
||||||
offset: Optional[int] = Query(
|
offset: Optional[int] = Query(
|
||||||
default=None, ge=0, description="Number of results to skip for pagination"
|
default=None, ge=0, description="Number of results to skip for pagination"
|
||||||
),
|
),
|
||||||
|
|||||||
@ -9,6 +9,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -42,6 +44,8 @@ async def get_results(
|
|||||||
away_abbrev: list = Query(default=None),
|
away_abbrev: list = Query(default=None),
|
||||||
home_abbrev: list = Query(default=None),
|
home_abbrev: list = Query(default=None),
|
||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
all_results = Result.select_season(season)
|
all_results = Result.select_season(season)
|
||||||
|
|
||||||
@ -74,8 +78,11 @@ async def get_results(
|
|||||||
if week_end is not None:
|
if week_end is not None:
|
||||||
all_results = all_results.where(Result.week <= week_end)
|
all_results = all_results.where(Result.week <= week_end)
|
||||||
|
|
||||||
|
total_count = all_results.count()
|
||||||
|
all_results = all_results.offset(offset).limit(limit)
|
||||||
|
|
||||||
return_results = {
|
return_results = {
|
||||||
"count": all_results.count(),
|
"count": total_count,
|
||||||
"results": [model_to_dict(x, recurse=not short_output) for x in all_results],
|
"results": [model_to_dict(x, recurse=not short_output) for x in all_results],
|
||||||
}
|
}
|
||||||
db.close()
|
db.close()
|
||||||
|
|||||||
@ -12,6 +12,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -44,6 +46,8 @@ async def get_players(
|
|||||||
key_mlbam: list = Query(default=None),
|
key_mlbam: list = Query(default=None),
|
||||||
sort: Optional[str] = None,
|
sort: Optional[str] = None,
|
||||||
csv: Optional[bool] = False,
|
csv: Optional[bool] = False,
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
all_players = SbaPlayer.select()
|
all_players = SbaPlayer.select()
|
||||||
|
|
||||||
@ -101,8 +105,11 @@ async def get_players(
|
|||||||
db.close()
|
db.close()
|
||||||
return Response(content=return_val, media_type="text/csv")
|
return Response(content=return_val, media_type="text/csv")
|
||||||
|
|
||||||
|
total_count = all_players.count()
|
||||||
|
all_players = all_players.offset(offset).limit(limit)
|
||||||
|
|
||||||
return_val = {
|
return_val = {
|
||||||
"count": all_players.count(),
|
"count": total_count,
|
||||||
"players": [model_to_dict(x) for x in all_players],
|
"players": [model_to_dict(x) for x in all_players],
|
||||||
}
|
}
|
||||||
db.close()
|
db.close()
|
||||||
|
|||||||
@ -9,6 +9,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -38,6 +40,8 @@ async def get_schedules(
|
|||||||
week_start: Optional[int] = None,
|
week_start: Optional[int] = None,
|
||||||
week_end: Optional[int] = None,
|
week_end: Optional[int] = None,
|
||||||
short_output: Optional[bool] = True,
|
short_output: Optional[bool] = True,
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
all_sched = Schedule.select_season(season)
|
all_sched = Schedule.select_season(season)
|
||||||
|
|
||||||
@ -69,8 +73,11 @@ async def get_schedules(
|
|||||||
|
|
||||||
all_sched = all_sched.order_by(Schedule.id)
|
all_sched = all_sched.order_by(Schedule.id)
|
||||||
|
|
||||||
|
total_count = all_sched.count()
|
||||||
|
all_sched = all_sched.offset(offset).limit(limit)
|
||||||
|
|
||||||
return_sched = {
|
return_sched = {
|
||||||
"count": all_sched.count(),
|
"count": total_count,
|
||||||
"schedules": [model_to_dict(x, recurse=not short_output) for x in all_sched],
|
"schedules": [model_to_dict(x, recurse=not short_output) for x in all_sched],
|
||||||
}
|
}
|
||||||
db.close()
|
db.close()
|
||||||
|
|||||||
@ -8,6 +8,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -23,6 +25,8 @@ async def get_standings(
|
|||||||
league_abbrev: Optional[str] = None,
|
league_abbrev: Optional[str] = None,
|
||||||
division_abbrev: Optional[str] = None,
|
division_abbrev: Optional[str] = None,
|
||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
standings = Standings.select_season(season)
|
standings = Standings.select_season(season)
|
||||||
|
|
||||||
@ -57,8 +61,11 @@ async def get_standings(
|
|||||||
div_teams = [x for x in standings]
|
div_teams = [x for x in standings]
|
||||||
div_teams.sort(key=lambda team: win_pct(team), reverse=True)
|
div_teams.sort(key=lambda team: win_pct(team), reverse=True)
|
||||||
|
|
||||||
|
total_count = len(div_teams)
|
||||||
|
div_teams = div_teams[offset : offset + limit]
|
||||||
|
|
||||||
return_standings = {
|
return_standings = {
|
||||||
"count": len(div_teams),
|
"count": total_count,
|
||||||
"standings": [model_to_dict(x, recurse=not short_output) for x in div_teams],
|
"standings": [model_to_dict(x, recurse=not short_output) for x in div_teams],
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -13,6 +13,8 @@ from ..dependencies import (
|
|||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
update_season_batting_stats,
|
update_season_batting_stats,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -59,6 +61,8 @@ async def get_games(
|
|||||||
division_id: Optional[int] = None,
|
division_id: Optional[int] = None,
|
||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
sort: Optional[str] = None,
|
sort: Optional[str] = None,
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
) -> Any:
|
) -> Any:
|
||||||
all_games = StratGame.select()
|
all_games = StratGame.select()
|
||||||
|
|
||||||
@ -119,8 +123,11 @@ async def get_games(
|
|||||||
StratGame.season, StratGame.week, StratGame.game_num
|
StratGame.season, StratGame.week, StratGame.game_num
|
||||||
)
|
)
|
||||||
|
|
||||||
|
total_count = all_games.count()
|
||||||
|
all_games = all_games.offset(offset).limit(limit)
|
||||||
|
|
||||||
return_games = {
|
return_games = {
|
||||||
"count": all_games.count(),
|
"count": total_count,
|
||||||
"games": [model_to_dict(x, recurse=not short_output) for x in all_games],
|
"games": [model_to_dict(x, recurse=not short_output) for x in all_games],
|
||||||
}
|
}
|
||||||
db.close()
|
db.close()
|
||||||
|
|||||||
@ -13,7 +13,13 @@ from ...db_engine import (
|
|||||||
fn,
|
fn,
|
||||||
model_to_dict,
|
model_to_dict,
|
||||||
)
|
)
|
||||||
from ...dependencies import add_cache_headers, cache_result, handle_db_errors
|
from ...dependencies import (
|
||||||
|
add_cache_headers,
|
||||||
|
cache_result,
|
||||||
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
|
)
|
||||||
from .common import build_season_games
|
from .common import build_season_games
|
||||||
|
|
||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
@ -52,7 +58,7 @@ async def get_batting_totals(
|
|||||||
risp: Optional[bool] = None,
|
risp: Optional[bool] = None,
|
||||||
inning: list = Query(default=None),
|
inning: list = Query(default=None),
|
||||||
sort: Optional[str] = None,
|
sort: Optional[str] = None,
|
||||||
limit: Optional[int] = 200,
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
page_num: Optional[int] = 1,
|
page_num: Optional[int] = 1,
|
||||||
week_start: Optional[int] = None,
|
week_start: Optional[int] = None,
|
||||||
@ -423,8 +429,6 @@ async def get_batting_totals(
|
|||||||
run_plays = run_plays.order_by(StratPlay.game.asc())
|
run_plays = run_plays.order_by(StratPlay.game.asc())
|
||||||
# For other group_by values, skip game_id/play_num sorting since they're not in GROUP BY
|
# For other group_by values, skip game_id/play_num sorting since they're not in GROUP BY
|
||||||
|
|
||||||
if limit < 1:
|
|
||||||
limit = 1
|
|
||||||
bat_plays = bat_plays.paginate(page_num, limit)
|
bat_plays = bat_plays.paginate(page_num, limit)
|
||||||
|
|
||||||
logger.info(f"bat_plays query: {bat_plays}")
|
logger.info(f"bat_plays query: {bat_plays}")
|
||||||
|
|||||||
@ -13,7 +13,13 @@ from ...db_engine import (
|
|||||||
fn,
|
fn,
|
||||||
SQL,
|
SQL,
|
||||||
)
|
)
|
||||||
from ...dependencies import handle_db_errors, add_cache_headers, cache_result
|
from ...dependencies import (
|
||||||
|
handle_db_errors,
|
||||||
|
add_cache_headers,
|
||||||
|
cache_result,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
|
)
|
||||||
from .common import build_season_games
|
from .common import build_season_games
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -51,7 +57,7 @@ async def get_fielding_totals(
|
|||||||
team_id: list = Query(default=None),
|
team_id: list = Query(default=None),
|
||||||
manager_id: list = Query(default=None),
|
manager_id: list = Query(default=None),
|
||||||
sort: Optional[str] = None,
|
sort: Optional[str] = None,
|
||||||
limit: Optional[int] = 200,
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
page_num: Optional[int] = 1,
|
page_num: Optional[int] = 1,
|
||||||
):
|
):
|
||||||
@ -237,8 +243,6 @@ async def get_fielding_totals(
|
|||||||
def_plays = def_plays.order_by(StratPlay.game.asc())
|
def_plays = def_plays.order_by(StratPlay.game.asc())
|
||||||
# For other group_by values, skip game_id/play_num sorting since they're not in GROUP BY
|
# For other group_by values, skip game_id/play_num sorting since they're not in GROUP BY
|
||||||
|
|
||||||
if limit < 1:
|
|
||||||
limit = 1
|
|
||||||
def_plays = def_plays.paginate(page_num, limit)
|
def_plays = def_plays.paginate(page_num, limit)
|
||||||
|
|
||||||
logger.info(f"def_plays query: {def_plays}")
|
logger.info(f"def_plays query: {def_plays}")
|
||||||
|
|||||||
@ -16,7 +16,13 @@ from ...db_engine import (
|
|||||||
SQL,
|
SQL,
|
||||||
complex_data_to_csv,
|
complex_data_to_csv,
|
||||||
)
|
)
|
||||||
from ...dependencies import handle_db_errors, add_cache_headers, cache_result
|
from ...dependencies import (
|
||||||
|
handle_db_errors,
|
||||||
|
add_cache_headers,
|
||||||
|
cache_result,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
|
)
|
||||||
from .common import build_season_games
|
from .common import build_season_games
|
||||||
|
|
||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
@ -51,7 +57,7 @@ async def get_pitching_totals(
|
|||||||
risp: Optional[bool] = None,
|
risp: Optional[bool] = None,
|
||||||
inning: list = Query(default=None),
|
inning: list = Query(default=None),
|
||||||
sort: Optional[str] = None,
|
sort: Optional[str] = None,
|
||||||
limit: Optional[int] = 200,
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
csv: Optional[bool] = False,
|
csv: Optional[bool] = False,
|
||||||
page_num: Optional[int] = 1,
|
page_num: Optional[int] = 1,
|
||||||
@ -164,8 +170,6 @@ async def get_pitching_totals(
|
|||||||
if group_by in ["playergame", "teamgame"]:
|
if group_by in ["playergame", "teamgame"]:
|
||||||
pitch_plays = pitch_plays.order_by(StratPlay.game.asc())
|
pitch_plays = pitch_plays.order_by(StratPlay.game.asc())
|
||||||
|
|
||||||
if limit < 1:
|
|
||||||
limit = 1
|
|
||||||
pitch_plays = pitch_plays.paginate(page_num, limit)
|
pitch_plays = pitch_plays.paginate(page_num, limit)
|
||||||
|
|
||||||
# Execute the Peewee query
|
# Execute the Peewee query
|
||||||
|
|||||||
@ -16,6 +16,8 @@ from ...dependencies import (
|
|||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
add_cache_headers,
|
add_cache_headers,
|
||||||
cache_result,
|
cache_result,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -70,7 +72,7 @@ async def get_plays(
|
|||||||
pitcher_team_id: list = Query(default=None),
|
pitcher_team_id: list = Query(default=None),
|
||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
sort: Optional[str] = None,
|
sort: Optional[str] = None,
|
||||||
limit: Optional[int] = 200,
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
page_num: Optional[int] = 1,
|
page_num: Optional[int] = 1,
|
||||||
s_type: Literal["regular", "post", "total", None] = None,
|
s_type: Literal["regular", "post", "total", None] = None,
|
||||||
):
|
):
|
||||||
@ -185,8 +187,6 @@ async def get_plays(
|
|||||||
season_games = season_games.where(StratGame.week > 18)
|
season_games = season_games.where(StratGame.week > 18)
|
||||||
all_plays = all_plays.where(StratPlay.game << season_games)
|
all_plays = all_plays.where(StratPlay.game << season_games)
|
||||||
|
|
||||||
if limit < 1:
|
|
||||||
limit = 1
|
|
||||||
bat_plays = all_plays.paginate(page_num, limit)
|
bat_plays = all_plays.paginate(page_num, limit)
|
||||||
|
|
||||||
if sort == "wpa-desc":
|
if sort == "wpa-desc":
|
||||||
|
|||||||
@ -11,6 +11,8 @@ from ..dependencies import (
|
|||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
cache_result,
|
cache_result,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
from ..services.base import BaseService
|
from ..services.base import BaseService
|
||||||
from ..services.team_service import TeamService
|
from ..services.team_service import TeamService
|
||||||
|
|||||||
@ -10,6 +10,8 @@ from ..dependencies import (
|
|||||||
valid_token,
|
valid_token,
|
||||||
PRIVATE_IN_SCHEMA,
|
PRIVATE_IN_SCHEMA,
|
||||||
handle_db_errors,
|
handle_db_errors,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -36,7 +38,7 @@ class TransactionList(pydantic.BaseModel):
|
|||||||
@router.get("")
|
@router.get("")
|
||||||
@handle_db_errors
|
@handle_db_errors
|
||||||
async def get_transactions(
|
async def get_transactions(
|
||||||
season,
|
season: int,
|
||||||
team_abbrev: list = Query(default=None),
|
team_abbrev: list = Query(default=None),
|
||||||
week_start: Optional[int] = 0,
|
week_start: Optional[int] = 0,
|
||||||
week_end: Optional[int] = None,
|
week_end: Optional[int] = None,
|
||||||
@ -45,8 +47,9 @@ async def get_transactions(
|
|||||||
player_name: list = Query(default=None),
|
player_name: list = Query(default=None),
|
||||||
player_id: list = Query(default=None),
|
player_id: list = Query(default=None),
|
||||||
move_id: Optional[str] = None,
|
move_id: Optional[str] = None,
|
||||||
is_trade: Optional[bool] = None,
|
|
||||||
short_output: Optional[bool] = False,
|
short_output: Optional[bool] = False,
|
||||||
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
|
offset: int = Query(default=0, ge=0),
|
||||||
):
|
):
|
||||||
if season:
|
if season:
|
||||||
transactions = Transaction.select_season(season)
|
transactions = Transaction.select_season(season)
|
||||||
@ -84,15 +87,15 @@ async def get_transactions(
|
|||||||
else:
|
else:
|
||||||
transactions = transactions.where(Transaction.frozen == 0)
|
transactions = transactions.where(Transaction.frozen == 0)
|
||||||
|
|
||||||
if is_trade is not None:
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=501, detail="The is_trade parameter is not implemented, yet"
|
|
||||||
)
|
|
||||||
|
|
||||||
transactions = transactions.order_by(-Transaction.week, Transaction.moveid)
|
transactions = transactions.order_by(-Transaction.week, Transaction.moveid)
|
||||||
|
|
||||||
|
total_count = transactions.count()
|
||||||
|
transactions = transactions.offset(offset).limit(limit)
|
||||||
|
|
||||||
return_trans = {
|
return_trans = {
|
||||||
"count": transactions.count(),
|
"count": total_count,
|
||||||
|
"limit": limit,
|
||||||
|
"offset": offset,
|
||||||
"transactions": [
|
"transactions": [
|
||||||
model_to_dict(x, recurse=not short_output) for x in transactions
|
model_to_dict(x, recurse=not short_output) for x in transactions
|
||||||
],
|
],
|
||||||
|
|||||||
@ -26,6 +26,8 @@ from ..dependencies import (
|
|||||||
update_season_batting_stats,
|
update_season_batting_stats,
|
||||||
update_season_pitching_stats,
|
update_season_pitching_stats,
|
||||||
get_cache_stats,
|
get_cache_stats,
|
||||||
|
MAX_LIMIT,
|
||||||
|
DEFAULT_LIMIT,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger("discord_app")
|
logger = logging.getLogger("discord_app")
|
||||||
@ -72,7 +74,7 @@ async def get_season_batting_stats(
|
|||||||
"cs",
|
"cs",
|
||||||
] = "woba", # Sort field
|
] = "woba", # Sort field
|
||||||
sort_order: Literal["asc", "desc"] = "desc", # asc or desc
|
sort_order: Literal["asc", "desc"] = "desc", # asc or desc
|
||||||
limit: Optional[int] = 200,
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
offset: int = 0,
|
offset: int = 0,
|
||||||
csv: Optional[bool] = False,
|
csv: Optional[bool] = False,
|
||||||
):
|
):
|
||||||
@ -218,7 +220,7 @@ async def get_season_pitching_stats(
|
|||||||
"re24",
|
"re24",
|
||||||
] = "era", # Sort field
|
] = "era", # Sort field
|
||||||
sort_order: Literal["asc", "desc"] = "asc", # asc or desc (asc default for ERA)
|
sort_order: Literal["asc", "desc"] = "asc", # asc or desc (asc default for ERA)
|
||||||
limit: Optional[int] = 200,
|
limit: int = Query(default=DEFAULT_LIMIT, ge=1, le=MAX_LIMIT),
|
||||||
offset: int = 0,
|
offset: int = 0,
|
||||||
csv: Optional[bool] = False,
|
csv: Optional[bool] = False,
|
||||||
):
|
):
|
||||||
|
|||||||
154
tests/unit/test_query_limits.py
Normal file
154
tests/unit/test_query_limits.py
Normal file
@ -0,0 +1,154 @@
|
|||||||
|
"""
|
||||||
|
Tests for query limit/offset parameter validation and middleware behavior.
|
||||||
|
|
||||||
|
Verifies that:
|
||||||
|
- FastAPI enforces MAX_LIMIT cap (returns 422 for limit > 500)
|
||||||
|
- FastAPI enforces ge=1 on limit (returns 422 for limit=0 or limit=-1)
|
||||||
|
- Transactions endpoint returns limit/offset keys in the response
|
||||||
|
- strip_empty_query_params middleware treats ?param= as absent
|
||||||
|
|
||||||
|
These tests exercise FastAPI parameter validation which fires before any
|
||||||
|
handler code runs, so most tests don't require a live DB connection.
|
||||||
|
|
||||||
|
The app imports redis and psycopg2 at module level, so we mock those
|
||||||
|
system-level packages before importing app.main.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import sys
|
||||||
|
import pytest
|
||||||
|
from unittest.mock import MagicMock, patch
|
||||||
|
|
||||||
|
# ---------------------------------------------------------------------------
|
||||||
|
# Stub out C-extension / system packages that aren't installed in the test
|
||||||
|
# environment before any app code is imported.
|
||||||
|
# ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
_redis_stub = MagicMock()
|
||||||
|
_redis_stub.Redis = MagicMock(return_value=MagicMock(ping=MagicMock(return_value=True)))
|
||||||
|
sys.modules.setdefault("redis", _redis_stub)
|
||||||
|
|
||||||
|
_psycopg2_stub = MagicMock()
|
||||||
|
sys.modules.setdefault("psycopg2", _psycopg2_stub)
|
||||||
|
|
||||||
|
_playhouse_pool_stub = MagicMock()
|
||||||
|
sys.modules.setdefault("playhouse.pool", _playhouse_pool_stub)
|
||||||
|
_playhouse_pool_stub.PooledPostgresqlDatabase = MagicMock()
|
||||||
|
|
||||||
|
_pandas_stub = MagicMock()
|
||||||
|
sys.modules.setdefault("pandas", _pandas_stub)
|
||||||
|
_pandas_stub.DataFrame = MagicMock()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(scope="module")
|
||||||
|
def client():
|
||||||
|
"""
|
||||||
|
TestClient with the Peewee db object mocked so the app can be imported
|
||||||
|
without a running PostgreSQL instance. FastAPI validates query params
|
||||||
|
before calling handler code, so 422 responses don't need a real DB.
|
||||||
|
"""
|
||||||
|
mock_db = MagicMock()
|
||||||
|
mock_db.is_closed.return_value = False
|
||||||
|
mock_db.connect.return_value = None
|
||||||
|
mock_db.close.return_value = None
|
||||||
|
|
||||||
|
with patch("app.db_engine.db", mock_db):
|
||||||
|
from fastapi.testclient import TestClient
|
||||||
|
from app.main import app
|
||||||
|
|
||||||
|
with TestClient(app, raise_server_exceptions=False) as c:
|
||||||
|
yield c
|
||||||
|
|
||||||
|
|
||||||
|
def test_limit_exceeds_max_returns_422(client):
|
||||||
|
"""
|
||||||
|
GET /api/v3/decisions with limit=1000 should return 422.
|
||||||
|
|
||||||
|
MAX_LIMIT is 500; the decisions endpoint declares
|
||||||
|
limit: int = Query(ge=1, le=MAX_LIMIT), so FastAPI rejects values > 500
|
||||||
|
before any handler code runs.
|
||||||
|
"""
|
||||||
|
response = client.get("/api/v3/decisions?limit=1000")
|
||||||
|
assert response.status_code == 422
|
||||||
|
|
||||||
|
|
||||||
|
def test_limit_zero_returns_422(client):
|
||||||
|
"""
|
||||||
|
GET /api/v3/decisions with limit=0 should return 422.
|
||||||
|
|
||||||
|
Query(ge=1) rejects zero values.
|
||||||
|
"""
|
||||||
|
response = client.get("/api/v3/decisions?limit=0")
|
||||||
|
assert response.status_code == 422
|
||||||
|
|
||||||
|
|
||||||
|
def test_limit_negative_returns_422(client):
|
||||||
|
"""
|
||||||
|
GET /api/v3/decisions with limit=-1 should return 422.
|
||||||
|
|
||||||
|
Query(ge=1) rejects negative values.
|
||||||
|
"""
|
||||||
|
response = client.get("/api/v3/decisions?limit=-1")
|
||||||
|
assert response.status_code == 422
|
||||||
|
|
||||||
|
|
||||||
|
def test_transactions_has_limit_in_response(client):
|
||||||
|
"""
|
||||||
|
GET /api/v3/transactions?season=12 should include 'limit' and 'offset'
|
||||||
|
keys in the JSON response body.
|
||||||
|
|
||||||
|
The transactions endpoint was updated to return pagination metadata
|
||||||
|
alongside results so callers know the applied page size.
|
||||||
|
"""
|
||||||
|
mock_qs = MagicMock()
|
||||||
|
mock_qs.count.return_value = 0
|
||||||
|
mock_qs.where.return_value = mock_qs
|
||||||
|
mock_qs.order_by.return_value = mock_qs
|
||||||
|
mock_qs.offset.return_value = mock_qs
|
||||||
|
mock_qs.limit.return_value = mock_qs
|
||||||
|
mock_qs.__iter__ = MagicMock(return_value=iter([]))
|
||||||
|
|
||||||
|
with (
|
||||||
|
patch("app.routers_v3.transactions.Transaction") as mock_txn,
|
||||||
|
patch("app.routers_v3.transactions.Team") as mock_team,
|
||||||
|
patch("app.routers_v3.transactions.Player") as mock_player,
|
||||||
|
):
|
||||||
|
mock_txn.select_season.return_value = mock_qs
|
||||||
|
mock_txn.select.return_value = mock_qs
|
||||||
|
mock_team.select.return_value = mock_qs
|
||||||
|
mock_player.select.return_value = mock_qs
|
||||||
|
|
||||||
|
response = client.get("/api/v3/transactions?season=12")
|
||||||
|
|
||||||
|
# If the mock is sufficient the response is 200 with pagination keys;
|
||||||
|
# if some DB path still fires we at least confirm limit param is accepted.
|
||||||
|
assert response.status_code != 422
|
||||||
|
if response.status_code == 200:
|
||||||
|
data = response.json()
|
||||||
|
assert "limit" in data, "Response missing 'limit' key"
|
||||||
|
assert "offset" in data, "Response missing 'offset' key"
|
||||||
|
|
||||||
|
|
||||||
|
def test_empty_string_param_stripped(client):
|
||||||
|
"""
|
||||||
|
Query params with an empty string value should be treated as absent.
|
||||||
|
|
||||||
|
The strip_empty_query_params middleware rewrites the query string before
|
||||||
|
FastAPI parses it, so ?league_abbrev= is removed entirely rather than
|
||||||
|
forwarded as an empty string to the handler.
|
||||||
|
|
||||||
|
Expected: the request is accepted (not 422) and the empty param is ignored.
|
||||||
|
"""
|
||||||
|
mock_qs = MagicMock()
|
||||||
|
mock_qs.count.return_value = 0
|
||||||
|
mock_qs.where.return_value = mock_qs
|
||||||
|
mock_qs.__iter__ = MagicMock(return_value=iter([]))
|
||||||
|
|
||||||
|
with patch("app.routers_v3.standings.Standings") as mock_standings:
|
||||||
|
mock_standings.select_season.return_value = mock_qs
|
||||||
|
|
||||||
|
# ?league_abbrev= should be stripped → treated as absent (None), not ""
|
||||||
|
response = client.get("/api/v3/standings?season=12&league_abbrev=")
|
||||||
|
|
||||||
|
assert response.status_code != 422, (
|
||||||
|
"Empty string query param caused a 422 — middleware may not be stripping it"
|
||||||
|
)
|
||||||
Loading…
Reference in New Issue
Block a user