diff options
Diffstat (limited to 'api/routers')
| -rw-r--r-- | api/routers/__init__.py | 1 | ||||
| -rw-r--r-- | api/routers/get.py | 42 |
2 files changed, 26 insertions, 17 deletions
diff --git a/api/routers/__init__.py b/api/routers/__init__.py index e69de29..085b949 100644 --- a/api/routers/__init__.py +++ b/api/routers/__init__.py @@ -0,0 +1 @@ +from . import get diff --git a/api/routers/get.py b/api/routers/get.py index 7842cc3..7fd74ac 100644 --- a/api/routers/get.py +++ b/api/routers/get.py @@ -1,4 +1,4 @@ -from fastapi import FastAPI, APIRouter +from fastapi import APIRouter import numpy as np import pandas as pd @@ -6,8 +6,8 @@ router = APIRouter(tags=["Get requests"]) # Making relative paths. On windows slashes would be backwards -# Though, it is probably not a good way. And the API has to be activated -# from the project dir, for the paths to match. +# It is probably not a good way. +# API has to be activated from the project dir for the paths to match. relPathTeams = "AllAboutData/Data/NBAteams.csv" relPathPlayers = "AllAboutData/Data/Players/" @@ -22,31 +22,39 @@ def getTeamNames(): teamsDf = pd.read_csv(relPathTeams).to_dict() team_names = {} for el in range(len(teamsDf["name"])): - team_names.update({teamsDf["name"][el] : teamsDf["full_name"][el]}) + team_names.update({teamsDf["name"][el]: teamsDf["full_name"][el]}) return team_names @router.get("/teams") def getTeams(): + ''' + Returns all the teams of NBA, converted to json, + where each item represents one team. + ''' - # Converting the data and converting into dict to send + # Reading the data and converting into dict to send teamsDf = pd.read_csv(relPathTeams).to_dict() teams_dict = {} for el in range(len(teamsDf["name"])): teams_dict.update({el+1 : - {"Abbreviation" : teamsDf["abbreviation"][el], - "Name" : teamsDf["name"][el], - "FullName" : teamsDf["full_name"][el], - "City" : teamsDf["city"][el], - "Conference" : teamsDf["conference"][el], - "Division" : teamsDf["division"][el]} }) + {"Abbreviation": teamsDf["abbreviation"][el], + "Name": teamsDf["name"][el], + "FullName": teamsDf["full_name"][el], + "City": teamsDf["city"][el], + "Conference": teamsDf["conference"][el], + "Division": teamsDf["division"][el]} }) return teams_dict @router.get("/players/{team_name}") def getPlayers(team_name: str): + ''' + Returns players of requested team, converted to json, + where each item represents one player. + ''' teamNames = getTeamNames() team_name = team_name.capitalize() # Capitalizing in case url is given as lowercase @@ -58,14 +66,14 @@ def getPlayers(team_name: str): players_dict = {} for el in range(len(playersDf["first_name"])): players_dict.update({el+1 : - {"first_name" : playersDf["first_name"][el], - "last_name" : playersDf["last_name"][el], - "position" : playersDf["position"][el], - "height_feet" : playersDf["height_feet"][el], - "height_inches" : playersDf["height_inches"][el]} }) + {"first_name": playersDf["first_name"][el], + "last_name": playersDf["last_name"][el], + "position": playersDf["position"][el], + "height_feet": playersDf["height_feet"][el], + "height_inches": playersDf["height_inches"][el]} }) return players_dict else: - return {"Error 404" : "Team name Not Found"} + return {"Error 404": "Team name Not Found"} |