blob: c67c95b007074cfc9bdbda7ebe28a61ed01a2f2b (
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
72
73
74
75
76
77
78
79
80
81
82
|
from fastapi import FastAPI
import pandas as pd
import numpy as np
import os
import sys
sys.path.append("--")
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("/")
def getIndex():
return {"Message" : "Hello!"}
#@app.get("/sync")
#def syncPlayers(): # Currnetly only works for Unix type systems, which is not good
# os.system("rm " + relPathPlayers + "*")
# getData.getPlayerData(getData.url, getData.headers)
@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"}
|
