diff --git a/app/routers_v2/scouting.py b/app/routers_v2/scouting.py index 1b727a5..124392f 100644 --- a/app/routers_v2/scouting.py +++ b/app/routers_v2/scouting.py @@ -6,19 +6,15 @@ from ..db_engine import Player from ..dependencies import oauth2_scheme, valid_token from ..player_scouting import get_player_ids - -router = APIRouter( - prefix='/api/v2/scouting', - tags=['scouting'] -) +router = APIRouter(prefix="/api/v2/scouting", tags=["scouting"]) class BattingFiles(pydantic.BaseModel): - vl_basic: str = 'vl-basic.csv' - vl_rate: str = 'vl-rate.csv' - vr_basic: str = 'vr-basic.csv' - vr_rate: str = 'vr-rate.csv' - running: str = 'running.csv' + vl_basic: str = "vl-basic.csv" + vl_rate: str = "vl-rate.csv" + vr_basic: str = "vr-basic.csv" + vr_rate: str = "vr-rate.csv" + running: str = "running.csv" # def csv_file_to_dataframe(filename: str) -> pd.DataFrame | None: @@ -28,45 +24,46 @@ class BattingFiles(pydantic.BaseModel): # for row in reader: -@router.get('/playerkeys') +@router.get("/playerkeys") async def get_player_keys(player_id: list = Query(default=None)): all_keys = [] for x in player_id: this_player = Player.get_or_none(Player.player_id == x) if this_player is not None: - this_keys = get_player_ids(this_player.bbref_id, id_type='bbref') + this_keys = get_player_ids(this_player.bbref_id, id_type="bbref") if this_keys is not None: all_keys.append(this_keys) - return_val = {'count': len(all_keys), 'keys': [ - dict(x) for x in all_keys - ]} + return_val = {"count": len(all_keys), "keys": [dict(x) for x in all_keys]} return return_val -@router.post('/live-update/batting') -def live_update_batting(files: BattingFiles, cardset_id: int, token: str = Depends(oauth2_scheme)): +@router.post("/live-update/batting") +def live_update_batting( + files: BattingFiles, cardset_id: int, token: str = Depends(oauth2_scheme) +): if not valid_token(token): - logging.warning('Bad Token: [REDACTED]') + logging.warning("Bad Token: [REDACTED]") raise HTTPException( - status_code=401, - detail='You are not authorized to initiate live updates.' + status_code=401, detail="You are not authorized to initiate live updates." ) - data = {} # : { 'vL': [combined vl stat data], 'vR': [combined vr stat data] } + data = ( + {} + ) # : { 'vL': [combined vl stat data], 'vR': [combined vr stat data] } for row in files.vl_basic: - if row['pa'] >= 20: - data[row['fgid']]['vL'] = row + if row["pa"] >= 20: + data[row["fgid"]]["vL"] = row for row in files.vl_rate: - if row['fgid'] in data.keys(): - data[row['fgid']]['vL'].extend(row) + if row["fgid"] in data.keys(): + data[row["fgid"]]["vL"].extend(row) for row in files.vr_basic: - if row['pa'] >= 40 and row['fgid'] in data.keys(): - data[row['fgid']]['vR'] = row + if row["pa"] >= 40 and row["fgid"] in data.keys(): + data[row["fgid"]]["vR"] = row for row in files.vr_rate: - if row['fgid'] in data.keys(): - data[row['fgid']]['vR'].extend(row) + if row["fgid"] in data.keys(): + data[row["fgid"]]["vR"].extend(row) for x in data.items(): pass @@ -76,15 +73,3 @@ def live_update_batting(files: BattingFiles, cardset_id: int, token: str = Depen # Read running stats and create/update BattingCard object return files.dict() - - -@router.post('/live-update/pitching') -def live_update_pitching(files: BattingFiles, token: str = Depends(oauth2_scheme)): - if not valid_token(token): - logging.warning('Bad Token: [REDACTED]') - raise HTTPException( - status_code=401, - detail='You are not authorized to initiate live updates.' - ) - - return files.dict()