blob: 3816fab0854e793818761fd42981183913e4fc23 (
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
|
from fastapi import FastAPI
import pandas as pd
import numpy as np
app = FastAPI()
# Making relative paths for the data, on windows slashes would have to be turned around.
# It is probably not the best way.
relPathTeams = "../AllAboutData/Data/NBAteams.csv"
relPathPlayers = "../AllAboutData/Data/Players/"
# func which return dict with team names as keys and full team names as values
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]})
return team_names
@app.get("/teams")
def getTeams():
# Getting the data from csv files and then converting into dict to be send out on get request
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
@app.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")
print(playersDf)
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"}
|
