blob: 7842cc356c45cd77927030cc1ecbed6c19ea411d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
from fastapi import FastAPI, APIRouter
import numpy as np
import pandas as pd
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.
relPathTeams = "AllAboutData/Data/NBAteams.csv"
relPathPlayers = "AllAboutData/Data/Players/"
def getTeamNames():
'''
Returns dictionary with fetched teams' names as keys
and full_names as values.
'''
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]})
return team_names
@router.get("/teams")
def getTeams():
# Converting 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]} })
return teams_dict
@router.get("/players/{team_name}")
def getPlayers(team_name: str):
teamNames = getTeamNames()
team_name = team_name.capitalize() # Capitalizing in case url is given as lowercase
if team_name in teamNames.keys():
playersDf = pd.read_csv(relPathPlayers+teamNames[team_name]+".csv")
playersDf = playersDf.replace({np.nan:None}) # For Json Nan must be replaced
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]} })
return players_dict
else:
return {"Error 404" : "Team name Not Found"}
|
