From ab2efcde6a5fac062ace86aa1705dc47ce69f438 Mon Sep 17 00:00:00 2001
From: Rasmus Luha <rasmus.luha@gmail.com>
Date: Thu, 17 Mar 2022 00:47:22 +0200
Subject: Little Restructuring

---
 API/routers/get.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 API/routers/get.py

(limited to 'API/routers/get.py')

diff --git a/API/routers/get.py b/API/routers/get.py
new file mode 100644
index 0000000..031b469
--- /dev/null
+++ b/API/routers/get.py
@@ -0,0 +1,52 @@
+from fastapi import FastAPI, APIRouter
+import numpy as np
+import pandas as pd
+import sys
+
+sys.path.append("..")
+import utils
+
+router = APIRouter(tags=["Get requests"])
+
+@router.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(utils.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 = utils.getTeamNames()
+    team_name = team_name.capitalize()  #  Capitalizing in case url is given as lowercase
+    if team_name in teamNames.keys():
+
+        playersDf = pd.read_csv(utils.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"}
+
-- 
cgit v1.2.3