wsba-hockey 1.2.4__py3-none-any.whl → 1.2.5__py3-none-any.whl

wsba_hockey/tools/columns.py

@@ -38,6 +38,7 @@ def col_map():
             'periodDescriptor.maxRegulationPeriods':'period_max_regulation',
             'periodDescriptor.periodType':'period_type',
             'gameOutcome.lastPeriodType':'period_type_last',
+            'gameOutcome.otPeriods':'ot_periods',
             'seriesUrl':'series_url',
             'seriesStatus.round':'series_round',
             'seriesStatus.seriesAbbrev':'series_abbr',
@@ -48,7 +49,11 @@ def col_map():
             'seriesStatus.topSeedTeamAbbrev':'top_seed_team_abbr',
             'seriesStatus.bottomSeedTeamAbbrev':'bottom_seed_team_abbr',
             'seriesStatus.topSeedWins':'top_seed_wins',
-            'seriesStatus.bottomSeedWins':'bottom_seed_wins'
+            'seriesStatus.bottomSeedWins':'bottom_seed_wins',
+            'clock.timeRemaining':'period_time_remaining',
+            'clock.secondsRemaining':'period_seconds_remaining',
+            'clock.running':'period_clock_running',
+            'clock.inIntermission':'game_in_intermission'
         },
         'season_info':{
             'id':'season',

wsba_hockey/wsba_main.py

@@ -85,7 +85,7 @@ CONVERT_TEAM_ABBR = {'L.A':'LAK',
                      'T.B':'TBL',
                      'PHX':'ARI'}
 
-PER_SIXTY = ['Fi','xGi','Gi','A1','A2','P1','P','Si','OZF','NZF','DZF','FF','FA','xGF','xGA','GF','GA','SF','SA','CF','CA','HF','HA','Give','Take','Penl','Penl2','Penl5','Draw','Block','GSAx']
+PER_SIXTY = ['Fi','xGi','Gi','A1','A2','P1','P','Si','OZF','NZF','DZF','FF','FA','xGF','xGA','GF','GA','SF','SA','CF','CA','HF','HA','Give','Take','Penl','Penl2','Penl5','Draw','PIM','Block','GSAx']
 
 #Some games in the API are specifically known to cause errors in scraping.
 #This list is updated as frequently as necessary
@@ -289,7 +289,7 @@ def nhl_scrape_schedule(season:int, start:str = '', end:str = ''):
             A DataFrame containing the schedule data for the specified season and date range.
     """
 
-    api = "https://api-web.nhle.com/v1/schedule/"
+    api = "https://api-web.nhle.com/v1/score/"
 
     #If either start or end are blank then find start and endpoints for specified season
     if start == '' or end == '':
@@ -325,7 +325,7 @@ def nhl_scrape_schedule(season:int, start:str = '', end:str = ''):
         print(f'Scraping games on {str(inc)[:10]}...')
         
         get = rs.get(f'{api}{str(inc)[:10]}').json()
-        gameWeek = pd.json_normalize(list(pd.json_normalize(get['gameWeek'])['games'])[0])
+        gameWeek = pd.json_normalize(get['games']).drop(columns=['goals'],errors='ignore')
         
         #Return nothing if there's nothing
         if gameWeek.empty:
@@ -382,35 +382,54 @@ def nhl_scrape_season(season:int, split_shifts:bool = False, season_types:list[i
     """
      
     #Determine whether to use schedule data in repository or to scrape
+    local_failed = False
+
     if local:
-        load = pd.read_csv(local_path)
-        load['date'] = pd.to_datetime(load['date'])
-
-        if start == '' or end == '':
-            season_data = rs.get('https://api.nhle.com/stats/rest/en/season').json()['data']
-            season_data = [s for s in season_data if s['id'] == season][0]
-            start = season_data['startDate'][0:10]
-            end = season_data['endDate'][0:10]
-            
+        try:
+            load = pd.read_csv(local_path)
+            load['game_date'] = pd.to_datetime(load['game_date'])
+
+            if start == '' or end == '':
+                season_data = rs.get('https://api.nhle.com/stats/rest/en/season').json()['data']
+                season_data = [s for s in season_data if s['id'] == season][0]
+                
+                season_start = season_data['startDate'][0:10]
+                season_end = season_data['endDate'][0:10]
+
+            else:   
+                season_start = f'{(str(season)[0:4] if int(start[0:2])>=9 else str(season)[4:8])}-{start[0:2]}-{start[3:5]}'
+                season_end =  f'{(str(season)[0:4] if int(end[0:2])>=9 else str(season)[4:8])}-{end[0:2]}-{end[3:5]}'
+
             form = '%Y-%m-%d'
 
             #Create datetime values from dates
-            start = datetime.strptime(start,form)
-            end = datetime.strptime(end,form)
-
-        else:   
-            start = f'{(str(season)[0:4] if int(start[0:2])>=9 else str(season)[4:8])}-{start[0:2]}-{start[3:5]}'
-            end =  f'{(str(season)[0:4] if int(end[0:2])>=9 else str(season)[4:8])}-{end[0:2]}-{end[3:5]}'
+            start_date = datetime.strptime(season_start,form)
+            end_date = datetime.strptime(season_end,form)
+
+            load = load.loc[(load['season']==season)&
+                            (load['season_type'].isin(season_types))&
+                            (load['game_date']>=start_date)&(load['game_date']<=end_date)&
+                            (load['game_schedule_state']=='OK')&
+                            (load['game_state']!='FUT')
+                            ]
             
+            game_ids = load['game_id'].to_list()
+        except KeyError:
+            #If loading games locally fails then force a scrape
+            local_failed = True
+            print('Loading games locally has failed.  Loading schedule data with a scrape...')
+    else:
+        local_failed = True
+
+    if local_failed:
+        load = nhl_scrape_schedule(season,start,end)
         load = load.loc[(load['season']==season)&
                         (load['season_type'].isin(season_types))&
-                        (load['game_date']>=start)&(load['game_date']<=end)]
+                        (load['game_schedule_state']=='OK')&
+                        (load['game_state']!='FUT')
+                        ]
         
         game_ids = load['game_id'].to_list()
-    else:
-        load = nhl_scrape_schedule(season,start,end)
-        load = load.loc[(load['season']==season)&(load['season_type'].isin(season_types))]
-        game_ids = load['game_id'].to_list()
 
     #If no games found, terminate the process
     if not game_ids:

{wsba_hockey-1.2.4.dist-info → wsba_hockey-1.2.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wsba_hockey
-Version: 1.2.4
+Version: 1.2.5
 Summary: WeakSide Breakout's complete Python package of access to hockey data, primairly including the scraping of National Hockey League schedule, play-by-play, and shifts information.
 Author-email: Owen Singh <owenbksingh@gmail.com>
 Project-URL: Homepage, https://github.com/owensingh38/wsba_hockey/

{wsba_hockey-1.2.4.dist-info → wsba_hockey-1.2.5.dist-info}/RECORD

@@ -1,8 +1,8 @@
 wsba_hockey/__init__.py,sha256=DozjAXICrGwXBCF1ej2rgRZw6BtYtNimLqddD0gO4gM,595
-wsba_hockey/wsba_main.py,sha256=BvDcMv3SwlvCbuJgsAf2qo0naZ8v27ErPyDP09DIylU,76187
+wsba_hockey/wsba_main.py,sha256=ooD8qjZGJPGRi6uomCw0OMcH48htPuE34kzK8u_N35k,76963
 wsba_hockey/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wsba_hockey/tools/agg.py,sha256=OkIYd-ApvGVYe2JJLOI21jnDIN5LH8nkeH7eo0reWFI,23364
-wsba_hockey/tools/columns.py,sha256=S3956AVFGMyXdk86ssQqJrwQC-N7tWT3PQ_884W33w8,14020
+wsba_hockey/tools/columns.py,sha256=XR-yRlEmTWYuzRBk07YpKRCA-Co11R64S4yfYgIrrGM,14310
 wsba_hockey/tools/game_pred.py,sha256=OGh6o1vIcyLUixU80hOO0RPGNmDSY1cvvCNZFcP0wL4,1308
 wsba_hockey/tools/plotting.py,sha256=Ix-Kj-FEP4_uwSwyb5lIRj3tOFdO1z7XUjWManAkY24,6013
 wsba_hockey/tools/scraping.py,sha256=6_GyF8o56fuijTosm4x4OSrvpL61ZygluK2A26XajqU,52246
@@ -10,8 +10,8 @@ wsba_hockey/tools/xg_model.py,sha256=rY6D1YMuSZkORqHd8ZCp-2gBmH9KUG0DoyHEiXpzWlg
 wsba_hockey/tools/archive/old_scraping.py,sha256=hEjMI1RtfeZnf0RBiJFI38oXkLZ3WofeH5xqcF4pzgM,49585
 wsba_hockey/tools/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wsba_hockey/tools/utils/shared.py,sha256=KxeQVttGem73yncAlnuZvTclqpJpoerTKtLusRh5zsk,2472
-wsba_hockey-1.2.4.dist-info/licenses/LICENSE,sha256=Nr_Um1Pd5FQJTWWgm7maZArdtYMbDhzXYSwyJIZDGik,1114
-wsba_hockey-1.2.4.dist-info/METADATA,sha256=ObmBmiBvwQ3KIk17J0ydi5D2j-ovatxGbj51ksY47qw,3592
-wsba_hockey-1.2.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-wsba_hockey-1.2.4.dist-info/top_level.txt,sha256=acU7s3x-RZC1zGiqCOmO0g267iqCg34lzIfdmYxxGmQ,12
-wsba_hockey-1.2.4.dist-info/RECORD,,
+wsba_hockey-1.2.5.dist-info/licenses/LICENSE,sha256=Nr_Um1Pd5FQJTWWgm7maZArdtYMbDhzXYSwyJIZDGik,1114
+wsba_hockey-1.2.5.dist-info/METADATA,sha256=8ixTFSlyTLzdcXCVt_F4ZNrRzvgSgyc7O6_0FYSCOzo,3592
+wsba_hockey-1.2.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+wsba_hockey-1.2.5.dist-info/top_level.txt,sha256=acU7s3x-RZC1zGiqCOmO0g267iqCg34lzIfdmYxxGmQ,12
+wsba_hockey-1.2.5.dist-info/RECORD,,