minor comment fixes

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"}