Player Tracking Stats

nba_py/__init__.py

@@ -1,8 +1,7 @@
-from datetime import datetime, timedelta
-import os
-
 from requests import get
+from datetime import datetime, timedelta
 from nba_py.constants import League
+import os
 
 HAS_PANDAS = True
 try:
@@ -22,13 +21,10 @@
 # Constants
 TODAY = datetime.today()
 BASE_URL = 'http://stats.nba.com/stats/{endpoint}'
-HEADERS = {
-    'user-agent': ('Mozilla/5.0 (Windows NT 6.2; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/57.0.2987.133 Safari/537.36'),  # noqa: E501
-    'Dnt': ('1'),
-    'Accept-Encoding': ('gzip, deflate, sdch'),
-    'Accept-Language': ('en'),
-    'origin': ('http://stats.nba.com')
-    }
+HEADERS = {'user-agent': ('Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) '
+                          'AppleWebKit/537.36 (KHTML, like Gecko) '
+                          'Chrome/45.0.2454.101 Safari/537.36'),
+          }
 
 
 def _api_scrape(json_inp, ndx):

nba_py/constants.py

@@ -1,10 +1,4 @@
-from datetime import datetime
-
-_curr_year = datetime.now().year
-if datetime.now().month > 6:
-    CURRENT_SEASON = str(_curr_year) + "-" + str(_curr_year + 1)[2:]
-else:
-    CURRENT_SEASON = str(_curr_year - 1) + "-" + str(_curr_year)[2:]
+CURRENT_SEASON = '2017-18'
 
 TEAMS = {
     'ATL': {
@@ -419,10 +413,19 @@ class MeasureType:
     Usage = 'Usage'
     Default = Base
 
-
 class PtMeasureType:
     SpeedDistance = 'SpeedDistance'
-
+    Drives = 'Drives'
+    Defense = 'Defense'
+    CatchShoot = 'CatchShoot'
+    Passing = 'Passing'
+    Possessions = 'Possessions'
+    PullUpShot = 'PullUpShot'
+    Rebounding = 'Rebounding'
+    Efficiency = 'Efficiency'
+    ElbowTouch = 'ElbowTouch'
+    PostTouch = 'PostTouch'
+    PaintTouch = 'PaintTouch'
 
 class GroupQuantity:
     Default = 5
@@ -614,8 +617,7 @@ class StatCategory:
 
 
 class ContextMeasure:
-    # Not sure if this is mapped correctly. Source:
-    # https://github.com/bradleyfay/NBAStats
+    # Not sure if this is mapped correctly. Source: https://github.com/bradleyfay/NBAStats
     FGM = 'FGM'
     FGA = 'FGA'
     FG_PCT = 'FG_PCT'

nba_py/draftcombine.py

@@ -1,14 +1,11 @@
 from nba_py import _api_scrape, _get_json
-from nba_py import constants
+from nba_py.constants import *
 
 
 class Summary:
     _endpoint = 'draftcombinestats'
 
-    def __init__(
-            self,
-            league_id=constants.League.Default,
-            season=constants.CURRENT_SEASON):
+    def __init__(self, league_id=League.Default, season=CURRENT_SEASON):
         self.json = _get_json(endpoint=self._endpoint,
                               params={'LeagueID': league_id,
                                       'SeasonYear': season})
@@ -20,10 +17,7 @@ def overall(self):
 class DrillResults:
     _endpoint = 'draftcombinedrillresults'
 
-    def __init__(
-            self,
-            league_id=constants.League.Default,
-            season=constants.CURRENT_SEASON):
+    def __init__(self, league_id=League.Default, season=CURRENT_SEASON):
         self.json = _get_json(endpoint=self._endpoint,
                               params={'LeagueID': league_id,
                                       'SeasonYear': season})
@@ -35,10 +29,7 @@ def overall(self):
 class SpotShooting:
     _endpoint = 'draftcombinespotshooting'
 
-    def __init__(
-            self,
-            league_id=constants.League.Default,
-            season=constants.CURRENT_SEASON):
+    def __init__(self, league_id=League.Default, season=CURRENT_SEASON):
         self.json = _get_json(endpoint=self._endpoint,
                               params={'LeagueID': league_id,
                                       'SeasonYear': season})

nba_py/game.py

@@ -1,19 +1,19 @@
 from nba_py import _api_scrape, _get_json
-from nba_py import constants
+from nba_py.constants import *
 
 
 class BoxscoreSummary:
     _endpoint = 'boxscoresummaryv2'
 
     def __init__(self,
                  game_id,
-                 season=constants.CURRENT_SEASON,
-                 season_type=constants.SeasonType.Default,
-                 range_type=constants.RangeType.Default,
-                 start_period=constants.StartPeriod.Default,
-                 end_period=constants.EndPeriod.Default,
-                 start_range=constants.StartRange.Default,
-                 end_range=constants.EndRange.Default):
+                 season=CURRENT_SEASON,
+                 season_type=SeasonType.Default,
+                 range_type=RangeType.Default,
+                 start_period=StartPeriod.Default,
+                 end_period=EndPeriod.Default,
+                 start_range=StartRange.Default,
+                 end_range=EndRange.Default):
         self.json = _get_json(endpoint=self._endpoint,
                               params={'GameID': game_id,
                                       'Season': season,
@@ -51,18 +51,17 @@ def season_series(self):
     def available_video(self):
         return _api_scrape(self.json, 8)
 
-
 class _BaseBoxcore:
 
     def __init__(self,
                  game_id,
-                 season=constants.CURRENT_SEASON,
-                 season_type=constants.SeasonType.Default,
-                 range_type=constants.RangeType.Default,
-                 start_period=constants.StartPeriod.Default,
-                 end_period=constants.EndPeriod.Default,
-                 start_range=constants.StartRange.Default,
-                 end_range=constants.EndRange.Default):
+                 season=CURRENT_SEASON,
+                 season_type=SeasonType.Default,
+                 range_type=RangeType.Default,
+                 start_period=StartPeriod.Default,
+                 end_period=EndPeriod.Default,
+                 start_range=StartRange.Default,
+                 end_range=EndRange.Default):
         self.json = _get_json(endpoint=self._endpoint,
                               params={'GameID': game_id,
                                       'Season': season,
@@ -154,8 +153,8 @@ class PlayByPlay:
 
     def __init__(self,
                  game_id,
-                 start_period=constants.StartPeriod.Default,
-                 end_period=constants.EndPeriod.Default):
+                 start_period=StartPeriod.Default,
+                 end_period=EndPeriod.Default):
         self.json = _get_json(endpoint=self._endpoint,
                               params={'GameID': game_id,
                                       'StartPeriod': start_period,
@@ -166,8 +165,10 @@ def info(self):
 
     def available_video(self):
         return _api_scrape(self.json, 1)
-
-
+
+
+
+
 class HustleStats:
     _endpoint = 'hustlestatsboxscore'