nfelodcm 0.1.4__py3-none-any.whl → 0.1.7__py3-none-any.whl

nfelodcm/nfelodcm/Engine/Assignments/assignment.py

@@ -1,10 +1,24 @@
-from .assignments import game_id_repl
+from .assignments import (
+    game_id_repl, penalty_formatting, desc_based_plays
+)
 
 ## a dictionary that holds assignments and the columns they add ##
 assignments = {
     'game_id_repl' : {
         'func' : game_id_repl,
         'new_columns' : []
+    },
+    'penalty_formatting' : {
+        'func' : penalty_formatting,
+        'new_columns' : []
+    },
+    'desc_based_plays' : {
+        'func' : desc_based_plays,
+        'new_columns' : [
+            ('qb_dropback_all', 'float32'),
+            ('rush_attempt_all', 'float32'),
+            ('play_call', 'str'),
+        ]
     }
 }
 

nfelodcm/nfelodcm/Engine/Assignments/assignments/__init__.py

@@ -1 +1,3 @@
 from .fastr_team_repl import game_id_repl
+from .penalty_formatting import penalty_formatting
+from .desc_based_plays import desc_based_plays

nfelodcm/nfelodcm/Engine/Assignments/assignments/desc_based_plays.py

@@ -0,0 +1,57 @@
+import pandas as pd
+import numpy
+
+def desc_based_plays(df):
+    '''
+    Infers play types from description that may show up
+    differently in the standard columns, especially when 
+    there is a penalty
+    '''
+    ## infer pass play ##
+    df['desc_based_dropback'] = numpy.where(
+        (
+            (df['desc'].str.contains(' pass ', regex=False)) |
+            (df['desc'].str.contains(' sacked', regex=False)) |
+            (df['desc'].str.contains(' scramble', regex=False))
+        ),
+        1,
+        0
+    )
+    ## infer run ##
+    df['desc_based_run'] = numpy.where(
+        (
+            (~df['desc'].str.contains(' pass ', regex=False, na=False)) &
+            (~df['desc'].str.contains(' sacked', regex=False, na=False)) &
+            (~df['desc'].str.contains(' scramble', regex=False, na=False)) &
+            (~df['desc'].str.contains(' kicks ', regex=False, na=False)) &
+            (~df['desc'].str.contains(' punts ', regex=False, na=False)) &
+            (~df['desc'].str.contains(' field goal ', regex=False, na=False)) &
+            (df['desc'].str.contains(' to ', regex=False)) &
+            (df['desc'].str.contains(' for ', regex=False))
+        ),
+        1,
+        0
+    )
+    ## add new cols ##
+    df['qb_dropback_all'] = df[[
+        'qb_dropback', 'desc_based_dropback'
+    ]].max(axis=1)
+    df['rush_attempt_all'] = df[[
+        'rush_attempt', 'desc_based_run'
+    ]].max(axis=1)
+    ## create a specific play call field ##
+    df['play_call'] = numpy.where(
+        df['qb_dropback_all'] == 1,
+        'Pass',
+        numpy.where(
+            df['rush_attempt_all'] == 1,
+            'Run',
+            numpy.nan
+        )
+    )
+    ## clean up ##
+    df = df.drop(columns=[
+        'desc_based_dropback', 'desc_based_run'
+    ]).copy()
+    ## return ##
+    return df

nfelodcm/nfelodcm/Engine/Assignments/assignments/fastr_team_repl.py

@@ -14,6 +14,15 @@ def game_id_repl(df):
     """
     df['away_team'] = df['away_team'].replace(repl)
     df['home_team'] = df['home_team'].replace(repl)
+    ## add extra checks to make comparable with pbp ##
+    for col in [
+        'posteam', 'defteam', 'penalty_team',
+        'side_of_field', 'timeout_team', 'td_team',
+        'return_team'
+    ]:
+        if col in df.columns:
+            df[col] = df[col].replace(repl)
+    ## id col ##
     df['game_id'] = (
         df['season'].astype('str') + '_' +
         df['week'].astype('str').str.zfill(2) + '_' +

nfelodcm/nfelodcm/Engine/Assignments/assignments/penalty_formatting.py

@@ -0,0 +1,21 @@
+import pandas as pd
+import numpy
+
+def penalty_formatting(df):
+    """
+    Adds additional context to penalties
+    """
+    ## offensive and defensive penalties ##
+    df['off_penalty'] = numpy.where(
+        df['penalty_team'] == df['posteam'],
+        1,
+        0
+    )
+    df['def_penalty'] = numpy.where(
+        df['penalty_team'] == df['defteam'],
+        1,
+        0
+    )
+    ## remove nans from penalties to enable groupings ##
+    df['penalty_type'] = df['penalty_type'].fillna('No Penalty')
+    return df

nfelodcm/nfelodcm/Maps/pbp.json

@@ -16,7 +16,9 @@
     "type": "season",
     "start": 1999
   },
-  "assignments": [],
+  "assignments": [
+    "game_id_repl", "penalty_formatting", "desc_based_plays"
+  ],
   "map": {
     "play_id": "int32",
     "game_id": "object",

nfelodcm/nfelodcm/nfelodcm.py

@@ -1,5 +1,7 @@
 import pandas as pd
 import numpy
+import pathlib
+import json
 
 from .Engine import DCMTable
 import nfelodcm.nfelodcm.Utilities as utils
@@ -46,4 +48,23 @@ def get_map(url):
         print('"{0}": "{1}",'.format(
             col, dtype
         ))
-        
+
+def get_season_state(state_type='last_full_week'):
+    '''
+    Returns the season and week of the specified state type:
+       'last_full_week' (default): the last week in which no games are still unplayed
+       'last_partual_week': the last week where any game has been played
+       'next_week': the first week with no games played
+    '''
+    ## open global variables json ##
+    global_variables = None
+    with open('{0}/global_variables.json'.format(pathlib.Path(__file__).parent.resolve()), 'r') as fp:
+        ## load config ##
+        global_variables = json.load(fp)
+    ## check input ##
+    if state_type not in list(global_variables['season_states'].keys()):
+        raise Exception('UTILITY ERROR: {0} is not an available state_type. Available types: {1}'.format(
+            state_type, '\n     '.join(list(global_variables['season_states'].keys()))
+        ))
+    ## if state valid, return ##
+    return global_variables[state_type]['season'], global_variables[state_type]['week']

{nfelodcm-0.1.4.dist-info → nfelodcm-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nfelodcm
-Version: 0.1.4
+Version: 0.1.7
 Summary: Python package for loading and caching CSVs hosted on github into pandas dataframes
 Author-email: Robert Greer <nfl@robbygreer.com>
 Maintainer-email: Robert Greer <nfl@robbygreer.com>

{nfelodcm-0.1.4.dist-info → nfelodcm-0.1.7.dist-info}/RECORD

@@ -1,11 +1,13 @@
 nfelodcm/__init__.py,sha256=T4OUcbJYVwMLzKSrm7CRT84jeqlvQeVWhfV2RCKlYM0,112
-nfelodcm/nfelodcm/nfelodcm.py,sha256=Q96CBvdmyx-c-AQ67szBHoG38yxXlsjZq-liomu6oro,1128
+nfelodcm/nfelodcm/nfelodcm.py,sha256=97BA_U3RikyzxS5Mbewrgb4HYYTa1YQ4S47l5wN_tA0,2145
 nfelodcm/nfelodcm/Engine/DCMTable.py,sha256=9_x82HKPWOwkIqQF4qZvUCLg8n6IUQfrtLlaU4bPIxM,2646
 nfelodcm/nfelodcm/Engine/__init__.py,sha256=vEEWcIBE4KLHoBCT1s2AIeSZxMfv1bcAqUNAJzSMSMo,31
 nfelodcm/nfelodcm/Engine/Assignments/__init__.py,sha256=IPKxTBbO3b0xaO922zkwif0FFd8Xqe3ZvCIFjhYg1oE,80
-nfelodcm/nfelodcm/Engine/Assignments/assignment.py,sha256=lTomhBeW2rqgYOboh4RKVbhX1FqVdG0EElZjSHGJ8mA,817
-nfelodcm/nfelodcm/Engine/Assignments/assignments/__init__.py,sha256=qkbpUcNOvl9pDjC1kEaIFwTJQBEE9LJf3Hnc6LP_qI8,42
-nfelodcm/nfelodcm/Engine/Assignments/assignments/fastr_team_repl.py,sha256=mDIEn5XjAVqOMAbtqKbnhtzwGKkjXDeefHJ5ijDRuDU,575
+nfelodcm/nfelodcm/Engine/Assignments/assignment.py,sha256=Mol-Sc5buLfJ-h9ItzeUxXxPAzvj0rvQbUJhl3_lgL4,1191
+nfelodcm/nfelodcm/Engine/Assignments/assignments/__init__.py,sha256=9dMIp6FZu3ZrvTHomWqkzF77em8P9hxKhuXl-lIyRQ4,139
+nfelodcm/nfelodcm/Engine/Assignments/assignments/desc_based_plays.py,sha256=LAXOlE-IW8A49rRagxcAGRao_4c2RdC7AK6_iV3M2Vc,1810
+nfelodcm/nfelodcm/Engine/Assignments/assignments/fastr_team_repl.py,sha256=YD3ci5BsuEelFEuyRyUbAqSuAiBeIFsMuhTvXbvg6Ik,865
+nfelodcm/nfelodcm/Engine/Assignments/assignments/penalty_formatting.py,sha256=Ia3x_Z8EnzI4Bz_6LEfY0ICWvdmyzJsckrOUXGYgfRY,517
 nfelodcm/nfelodcm/Engine/Primatives/DataPull.py,sha256=x1SUfCpxbXr923o52ExLMJItVKujSfcBeJDrDkFop70,2909
 nfelodcm/nfelodcm/Engine/Primatives/Freshness.py,sha256=5dGPoFCeCBdhvr5g_Ey-xAK742F596uQfwBEfDouj7U,9679
 nfelodcm/nfelodcm/Engine/Primatives/LocalIO.py,sha256=aVdPiz-gkyrJLPOHBg87PiNWZAI6EZGfyJVej9Z2uk0,1527
@@ -14,7 +16,7 @@ nfelodcm/nfelodcm/Engine/Primatives/__init__.py,sha256=JUHF_8_1vqbp7M5rJfDwzqkgi
 nfelodcm/nfelodcm/Maps/games.json,sha256=TPEsWetmMBhmSWkCkSfgWqv1R1X9FkNIdDnWs_nqzGw,1829
 nfelodcm/nfelodcm/Maps/logos.json,sha256=7RLhXIp9bvp6EwALrwccCiY1_iNizYVV7M2XEwDRb5E,1776
 nfelodcm/nfelodcm/Maps/participants.json,sha256=Q0eeE0qbp91pIwjlH9WM7wI8n2gbwgywm-vRAkw1ofA,1211
-nfelodcm/nfelodcm/Maps/pbp.json,sha256=-91dzwDOG5aQP432MeRxlm2ybkCZksKnu6AgIPjjTSg,10361
+nfelodcm/nfelodcm/Maps/pbp.json,sha256=tU_Zhwju3iD7bju_4DnCJuUxXofkI-9jIe-zDQgylLU,10425
 nfelodcm/nfelodcm/Maps/qbelo.json,sha256=owSpd3hlC58HLCAeeTAhpIzhx9lA6EuvPfTlETklOpU,1832
 nfelodcm/nfelodcm/Maps/qbr.json,sha256=8zewIU0ted9FNFkjwyM308sR_pGvoB0PzZDeZhQUKXg,1153
 nfelodcm/nfelodcm/Maps/rosters.json,sha256=RvinVFIOSuBGeDwp3XqhKFeaWH8-A_nQRUO4KEM7JZ8,1435
@@ -24,7 +26,7 @@ nfelodcm/nfelodcm/Utilities/extract_map.py,sha256=DVITMr0mlP6p2V1ZdaIcbEavs-FZ28
 nfelodcm/nfelodcm/Utilities/global_variables.json,sha256=FrKTTJ3VJRTV8RdQmjpXZzIPLLC594SpiMBR6xYs7Nw,299
 nfelodcm/nfelodcm/Utilities/retrieve_season_state.py,sha256=ZrBaMWmo2DA9OZqZ6MBWmrY9Llkc5OUfxmDk7EB1tRU,579
 nfelodcm/nfelodcm/Utilities/set_season_state.py,sha256=Y5fvop3EZfWcx2F85MtSNEGnwjofWyEMg4_HLTenAUA,3737
-nfelodcm-0.1.4.dist-info/METADATA,sha256=GYZZ4QzCUnOVoPKsADzhIrYUP7AneGFDvUYpgFGAauc,4409
-nfelodcm-0.1.4.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-nfelodcm-0.1.4.dist-info/top_level.txt,sha256=SlNA7_yZSwWvZIl9Md6brBH_C7nAYnATe4CrvuUXOGo,9
-nfelodcm-0.1.4.dist-info/RECORD,,
+nfelodcm-0.1.7.dist-info/METADATA,sha256=i33HF_CTQxLPayZsoNav3yyeHWywSuKmQHioMSBgIus,4409
+nfelodcm-0.1.7.dist-info/WHEEL,sha256=mguMlWGMX-VHnMpKOjjQidIo1ssRlCFu4a4mBpz1s2M,91
+nfelodcm-0.1.7.dist-info/top_level.txt,sha256=SlNA7_yZSwWvZIl9Md6brBH_C7nAYnATe4CrvuUXOGo,9
+nfelodcm-0.1.7.dist-info/RECORD,,

{nfelodcm-0.1.4.dist-info → nfelodcm-0.1.7.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.42.0)
+Generator: setuptools (70.1.1)
 Root-Is-Purelib: true
 Tag: py3-none-any