cal/major-domo-database
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Add csv support for /players

Browse Source
This commit is contained in:
Cal Corum 2023-06-21 10:04:51 -05:00
parent 8f53d83d9e
commit 1759932643
1 changed files with 37 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
app/routers_v3/players.py
View File

@ -1,10 +1,13 @@
from fastapi import APIRouter, Depends, HTTPException, Query
from fastapi import APIRouter, Depends, HTTPException, Query, Response
from typing import List, Optional
import logging
import pydantic
from pandas import DataFrame
from ..db_engine import db, Player, model_to_dict, chunked
from ..dependencies import oauth2_scheme, valid_token
from ..dependencies import oauth2_scheme, valid_token, LOG_DATA
logging.basicConfig(filename=LOG_DATA['filename'], format=LOG_DATA['format'], level=LOG_DATA['log_level'])
router = APIRouter(
prefix='/api/v3/players',
@ -44,7 +47,8 @@ class PlayerList(pydantic.BaseModel):
@router.get('')
async def get_players(
season: Optional[int], team_id: list = Query(default=None), pos: list = Query(default=None),
is_injured: Optional[bool] = None, sort: Optional[str] = None):
is_injured: Optional[bool] = None, sort: Optional[str] = None, short_output: Optional[bool] = False,
csv: Optional[bool] = False):
all_players = Player.select_season(season)
if team_id is not None:
@ -70,10 +74,36 @@ async def get_players(
elif sort == 'name-desc':
all_players = all_players.order_by(-Player.name)
return_players = {
'count': all_players.count(),
'players': [model_to_dict(x, recurse=False) for x in all_players]
}
print(f'csv: {csv}')
if csv:
player_list = [
['name', 'wara', 'image', 'image2', 'team', 'season', 'pitcher_injury', 'pos_1', 'pos_2', 'pos_3',
'pos_4', 'pos_5', 'pos_6', 'pos_7', 'pos_8', 'last_game', 'last_game2', 'il_return', 'demotion_week',
'headshot', 'vanity_card', 'strat_code', 'bbref_id', 'injury_rating']
]
for line in all_players:
player_list.append(
[
line.name, line.wara, line.image, line.image2, line.team.abbrev, line.season, line.pitcher_injury,
line.pos_1, line.pos_2, line.pos_3, line.pos_4, line.pos_5, line.pos_6, line.pos_7, line.pos_8,
line.last_game, line.last_game2, line.il_return, line.demotion_week, line.headshot,
line.vanity_card, line.strat_code.replace(",", "-_-"), line.bbref_id, line.injury_rating
]
)
return_players = {
'count': all_players.count(),
'players': DataFrame(player_list).to_csv(header=False, index=False),
'csv': True
}
db.close()
return Response(content=return_players['players'], media_type='text/csv')
else:
return_players = {
'count': all_players.count(),
'players': [model_to_dict(x, recurse=not short_output) for x in all_players]
}
db.close()
return return_players