rustat-python-api 0.2.0__tar.gz → 0.3.1__tar.gz

{rustat-python-api-0.2.0/rustat_python_api.egg-info → rustat-python-api-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rustat-python-api
-Version: 0.2.0
+Version: 0.3.1
 Summary: A Python wrapper for RuStat API
 Home-page: https://github.com/dailydaniel/rustat-python-api
 Author: Daniel Zholkovsky
@@ -38,7 +38,7 @@ schedule = parser.get_schedule(team_id, season_id)
 keys = list(schedule.keys())
 match_id = keys[-1]
 
-events = parser.get_events(match_id)
+events = parser.get_events(match_id, process=True)
 
 stats = parser.get_match_stats(match_id)
 

{rustat-python-api-0.2.0 → rustat-python-api-0.3.1}/README.md

@@ -23,7 +23,7 @@ schedule = parser.get_schedule(team_id, season_id)
 keys = list(schedule.keys())
 match_id = keys[-1]
 
-events = parser.get_events(match_id)
+events = parser.get_events(match_id, process=True)
 
 stats = parser.get_match_stats(match_id)
 

rustat-python-api-0.3.1/rustat_python_api/config.py

@@ -0,0 +1,37 @@
+columns = [
+    'player_name', 'team_name', 'half', 'second', 'action_id', 'action_name',
+    'position_name', 'possession_number', 'pos_x', 'pos_y', 'pos_dest_x', 'pos_dest_y',
+    'player_id', 'number', 'team_id', 'standart_name', 'possession_time',
+    'opponent_id', 'opponent_name', 'opponent_team_id', 'opponent_team_name',
+    'opponent_position_name', 'zone_name', 'zone_dest_name', 'len',
+    'possession_team_id', 'possession_team_name', 'possession_name',
+    'attack_status_name', 'attack_type_name', 'attack_flang_name',
+    'attack_team_id', 'attack_team_name', 'attack_number',
+    'body_name', 'gate_x', 'gate_y', 'assistant_id',
+    'assistant_name', 'shot_type', 'touches', 'xg',
+    'shot_handling', 'match_id', 'receiver_id', 'receiver_name'
+]
+
+numeric_columns = [
+    'id', 'number', 'player_id', 'team_id', 'half', 'second',
+    'pos_x', 'pos_y', 'pos_dest_x', 'pos_dest_y', 'len', 'possession_id', 'possession_team_id',
+    'opponent_id', 'opponent_team_id', 'zone_id', 'zone_dest_id',
+    'possession_number', 'attack_status_id', 'attack_team_id', 'assistant_id', 'touches', 'xg'
+]
+
+id2type = {
+    1: 'pass', 2: 'duel', 3: 'foul',
+    4: 'shot', 5: 'free kick', 6: 'interception',
+    7: 'rebound', 8: 'goal', 9: 'clearance',
+    10: 'bad ball control', 11: 'control', 12: 'attack',
+    13: 'keeper', 14: 'substitution', 15: 'formation',
+    16: 'player position', 17: 'ball off', 18: 'match status',
+    19: 'mistake', 20: 'translation problem', 21: 'carry',
+    22: 'receive', 23: 'goal attack involvement', 24: 'rating',
+    25: 'average position', 26: 'cross', 27: 'ball out',
+    28: 'other', 29: 'video', 30: 'bad mistake',
+    31: 'bad keeper mistake', 32: 'goal moment', 33: 'team pressing',
+    34: 'line up', 35: 'sync', 36: 'referee',
+    37: 'insurance', 38: 'injury',
+    128: 'staff', 161: 'sub player'
+}

{rustat-python-api-0.2.0 → rustat-python-api-0.3.1}/rustat_python_api/parser.py

@@ -5,6 +5,8 @@ from tqdm import tqdm
 import time
 
 from .urls import URLs
+from .config import numeric_columns
+from .processing import processing
 
 
 class RuStatParser:
@@ -15,13 +17,6 @@ class RuStatParser:
         urls: dict = URLs,
         sleep: int = -1
     ):
-        self.numeric_columns = [
-            'id', 'number', 'player_id', 'team_id', 'half', 'second',
-            'pos_x', 'pos_y', 'pos_dest_x', 'pos_dest_y', 'len', 'possession_id', 'possession_team_id',
-            'opponent_id', 'opponent_team_id', 'zone_id', 'zone_dest_id',
-            'possession_number', 'attack_status_id', 'attack_team_id', 'assistant_id', 'touches', 'xg'
-        ]
-
         self.user = user
         self.password = password
         self.urls = urls
@@ -95,7 +90,7 @@ class RuStatParser:
             for row in data["data"]["row"]
         }
 
-    def get_events(self, match_id: int) -> pd.DataFrame | None:
+    def get_events(self, match_id: int, process: bool = True) -> pd.DataFrame | None:
         data = self.resp2data(
             self.urls["events"].format(
                 user=self.user,
@@ -109,8 +104,12 @@ class RuStatParser:
 
         df = pd.json_normalize(data["data"]["row"])
 
-        numeric_columns = [column for column in self.numeric_columns if column in df.columns]
-        df[numeric_columns] = df[numeric_columns].apply(pd.to_numeric, errors='coerce')
+        current_numeric_columns = [column for column in numeric_columns if column in df.columns]
+        df[current_numeric_columns] = df[current_numeric_columns].apply(pd.to_numeric, errors='coerce')
+
+        if process:
+            df['match_id'] = match_id
+            df = processing(df)
 
         return df
 

rustat-python-api-0.3.1/rustat_python_api/processing.py

@@ -0,0 +1,82 @@
+import pandas as pd
+import numpy as np
+
+from .config import columns, id2type
+
+
+def process_list(x: pd.Series):
+    lst = x.dropna().unique().tolist()
+    if len(lst) == 1:
+        return lst[0]
+    elif len(lst) == 0:
+        return np.nan
+    else:
+        return lst
+
+
+def gluing(df: pd.DataFrame) -> pd.DataFrame:
+    cols = ['player_id', 'half', 'second', 'pos_x', 'pos_y']
+
+    df_gb = df.groupby(cols).agg(process_list).reset_index()
+    df_gb['possession_number'] = df_gb['possession_number'].apply(
+        lambda x: max(x) if isinstance(x, list) else x
+    )
+    df_gb = df_gb.sort_values(by=['half', 'second', 'possession_number']).reset_index(drop=True)
+    return df_gb
+
+
+def add_reciever(glued_df: pd.DataFrame) -> pd.DataFrame:
+    df = glued_df.copy()
+    df['receiver_id'] = df['player_id'].shift(1)
+    df['receiver_name'] = df['player_name'].shift(1)
+
+    mask = (
+            (df['action_name'] == 'Ball receiving')
+            & (df['pos_x'] == df['pos_dest_x'].shift(1))
+            & (df['pos_y'] == df['pos_dest_y'].shift(1))
+            & (df['team_id'] == df['team_id'].shift(1))
+            & (df['player_id'] != df['player_id'].shift(1))
+            & (df['possession_number'] == df['possession_number'].shift(1))
+    )
+
+    idx = df[mask].index
+    remaining_idx = df.drop(idx-1).index
+
+    df.loc[remaining_idx, 'receiver_id'] = np.nan
+    df.loc[remaining_idx, 'receiver_name'] = np.nan
+
+    df = df[df['action_name'] != 'Ball receiving'].reset_index(drop=True)
+
+    return df
+
+
+def filter_data(df: pd.DataFrame) -> pd.DataFrame:
+    for column in columns:
+        if column not in df.columns:
+            df[column] = np.nan
+
+    return df[(~df['possession_number'].isna()) | (df['second'] != 0)][columns].reset_index(drop=True)
+
+
+def tagging(df: pd.DataFrame) -> pd.DataFrame:
+    df = df.rename(columns={'action_name': 'sub_tags', 'action_id': 'sub_tags_ids'})
+    df['sub_tags'] = df['sub_tags'].apply(lambda x: x if isinstance(x, list) else [x])
+    df['sub_tags_ids'] = df['sub_tags_ids'].apply(
+        lambda x:
+        list(set([int(t) // 1000 for t in x]))
+        if isinstance(x, list)
+        else [int(x) // 1000]
+    )
+    df['sub_tags_ids'] = df['sub_tags_ids'].apply(lambda x: [id2type[t] for t in x])
+    df = df.rename(columns={'sub_tags_ids': 'tags'})
+
+    return df
+
+
+def processing(df: pd.DataFrame) -> pd.DataFrame:
+    df = gluing(df)
+    df = add_reciever(df)
+    df = filter_data(df)
+    df = tagging(df)
+
+    return df

{rustat-python-api-0.2.0 → rustat-python-api-0.3.1/rustat_python_api.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rustat-python-api
-Version: 0.2.0
+Version: 0.3.1
 Summary: A Python wrapper for RuStat API
 Home-page: https://github.com/dailydaniel/rustat-python-api
 Author: Daniel Zholkovsky
@@ -38,7 +38,7 @@ schedule = parser.get_schedule(team_id, season_id)
 keys = list(schedule.keys())
 match_id = keys[-1]
 
-events = parser.get_events(match_id)
+events = parser.get_events(match_id, process=True)
 
 stats = parser.get_match_stats(match_id)
 

{rustat-python-api-0.2.0 → rustat-python-api-0.3.1}/rustat_python_api.egg-info/SOURCES.txt

@@ -3,7 +3,9 @@ README.md
 pyproject.toml
 setup.py
 rustat_python_api/__init__.py
+rustat_python_api/config.py
 rustat_python_api/parser.py
+rustat_python_api/processing.py
 rustat_python_api/urls.py
 rustat_python_api.egg-info/PKG-INFO
 rustat_python_api.egg-info/SOURCES.txt

{rustat-python-api-0.2.0 → rustat-python-api-0.3.1}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='rustat-python-api',
-    version='0.2.0',
+    version='0.3.1',
     description='A Python wrapper for RuStat API',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',