rustat-python-api 0.5.3__py3-none-any.whl → 0.5.4__py3-none-any.whl

rustat_python_api/parser.py

@@ -173,6 +173,80 @@ class RuStatParser:
 
         return df
 
+    def get_tracking_30fps(
+        self,
+        match_id: int, half: int, lang_id: int = 0,
+        referee_data: int = 0, ball_data: int = 1
+    ) -> pd.DataFrame | tuple | None:
+        data = self.resp2data(
+            self.urls["tracking_30fps"].format(
+                user=self.user,
+                password=self.password,
+                match_id=match_id,
+                half=half,
+                lang_id=lang_id,
+                referee_data=referee_data,
+                ball_data=ball_data
+            )
+        )
+
+        if not data:
+            return None
+
+        teams_data = data['data']['teams']
+        if 'ball' in data['data']:
+            ball_data = data['data']['ball']
+        if 'referee' in data['data']:
+            referee_data = data['data']['referee']
+        del data
+
+        tracking = pd.DataFrame(columns=[
+            "half", "second", "pos_x", "pos_y", "team_id", "player_id", "player_name", "distance", "speed"
+        ])
+
+        for team_data in tqdm(teams_data):
+            team_id = team_data["id"]
+            side_1h = team_data["gate_position_half_1"]
+
+            for player_data in team_data["players"]:
+                player_id = player_data["id"]
+                player_name = player_data["name"]
+
+                cur_df = pd.json_normalize(player_data["rows"])
+                cur_df = cur_df.apply(pd.to_numeric, errors='coerce')
+                cur_df["team_id"] = team_id
+                cur_df["player_id"] = player_id
+                cur_df["player_name"] = player_name
+                cur_df["side_1h"] = side_1h
+
+                tracking = pd.concat([tracking, cur_df], ignore_index=True)
+
+        tracking['second'] = tracking['second'].astype(float)
+        tracking = tracking.sort_values(by=['half', 'second', 'team_id', 'player_id']).reset_index(drop=True)
+        tracking['pos_x'] = tracking['pos_x'] + 105/2
+        tracking['team_id'] = tracking['team_id'].astype(str)
+
+        if referee_data:
+            referee = pd.json_normalize(referee_data['rows'])
+            referee['second'] = referee['second'].astype(float)
+            referee = referee.sort_values(by=['half', 'second']).reset_index(drop=True)
+            referee['pos_x'] = referee['pos_x'] + 105/2
+
+        if ball_data:
+            ball = pd.json_normalize(ball_data['rows'])
+            ball['second'] = ball['second'].astype(float)
+            ball = ball.sort_values(by=['half', 'second']).reset_index(drop=True)
+            ball['pos_x'] = ball['pos_x'] + 105/2
+
+        if referee_data and ball_data:
+            return tracking, referee, ball
+        elif referee_data:
+            return tracking, referee
+        elif ball_data:
+            return tracking, ball
+        else:
+            return tracking
+
     def get_match_stats(self, match_id: int) -> dict:
         data = self.resp2data(
             self.urls["match_stats"].format(

rustat_python_api/pitch_control.py

@@ -7,7 +7,7 @@ import matplotsoccer as mpl
 
 
 class PitchControl:
-    def __init__(self, tracking: pd.DataFrame, events: pd.DataFrame):
+    def __init__(self, tracking: pd.DataFrame, events: pd.DataFrame, ball_data: pd.DataFrame = None):
         self.team_ids = tracking['team_id'].unique()
         sides = tracking.groupby('team_id')['side_1h'].unique()
         side_by_team = dict(zip(self.team_ids, sides[self.team_ids].apply(lambda x: x[0])))
@@ -22,10 +22,11 @@ class PitchControl:
 
         self.locs_home, self.locs_away, self.locs_ball, self.t = self.get_locs(
             tracking,
-            events
+            events,
+            ball_data
         )
 
-    def get_locs(self, tracking: pd.DataFrame, events: pd.DataFrame) -> tuple:
+    def get_locs(self, tracking: pd.DataFrame, events: pd.DataFrame, ball_data: pd.DataFrame | None) -> tuple:
         events = events[[
             'possession_number', 'team_id', 'possession_team_id',
             'half', 'second', 'pos_x', 'pos_y'
@@ -38,10 +39,11 @@ class PitchControl:
             lambda x: self.swap_coords(x, 'y'), axis=1
         )
 
-        ball_data = self.interpolate_ball_data(
-            events[['half', 'second', 'pos_x', 'pos_y']],
-            tracking
-        )
+        if ball_data is None:
+            ball_data = self.interpolate_ball_data(
+                events[['half', 'second', 'pos_x', 'pos_y']],
+                tracking
+            )
 
         locs_home = {
             half: {

rustat_python_api/urls.py

@@ -4,5 +4,6 @@ URLs = {
     "match_stats": "http://feeds.rustatsport.ru/?tpl=207&user={user}&key={password}&match_id={match_id}&lang_id=1&format=json",
     "tournament_teams": "http://feeds.rustatsport.ru/?tpl=32&user={user}&key={password}&tournament_id=2&season_id={season_id}&date_start=&date_end=&lang_id=1&format=json",
     "tracking": "https://feeds.rustatsport.ru/?tpl=274&user={user}&key={password}&match_id={match_id}&lang_id=0&format=json",
+    "tracking_30fps": "https://feeds.rustatsport.ru/?tpl=1013&user={user}&key={password}&match_id={match_id}&referee_data={referee_data}&ball_data={ball_data}&half={half}&lang_id={lang_id}&format=json",
     "player_match_stats": "http://feeds.rustatsport.ru/?tpl=227&user={user}&key={password}&match_id={match_id}&lang_id=1&format=json"
 }

{rustat_python_api-0.5.3.dist-info → rustat_python_api-0.5.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rustat-python-api
-Version: 0.5.3
+Version: 0.5.4
 Summary: A Python wrapper for RuStat API
 Home-page: https://github.com/dailydaniel/rustat-python-api
 Author: Daniel Zholkovsky

rustat_python_api-0.5.4.dist-info/RECORD

@@ -0,0 +1,12 @@
+rustat_python_api/__init__.py,sha256=Ij-PAm2y5ss_XAZhKTZus35cRPLzvXFyIswDa_Iq3rs,164
+rustat_python_api/config.py,sha256=eMvi1p8Cfvnbp6Cd4bBOwgehVN7thKnaQV5uzWyGZXM,1844
+rustat_python_api/models_api.py,sha256=oHXEqeCupvZwjVEdoxf7W9LP7ELFKA8-9DuRXpQHLno,1701
+rustat_python_api/parser.py,sha256=YsA4apnm0xolfUC8E2YUSfhDkUugCR2dUhG9iZKsn2E,9861
+rustat_python_api/pitch_control.py,sha256=wBL9X310OBeFvheJssLR4e7wX5suUPjHegZKbEun1Lo,11540
+rustat_python_api/processing.py,sha256=46O7wUGv5boWf4A38zYinbtqEC7ke-BwTdEg589XQaw,2816
+rustat_python_api/urls.py,sha256=iJTD31T6OyXPAhmhViwFXVehrzwsOjBDONA1SIVc_40,1068
+rustat_python_api-0.5.4.dist-info/LICENSE,sha256=4Cohqg5p6Mq1xyrzdEX8AvFSA62GSVvapEOr2xK_tgY,57
+rustat_python_api-0.5.4.dist-info/METADATA,sha256=KU_FqqVPi4RiD5BPq-MvDpRqTs6y0MPYiMWajcxGL08,1730
+rustat_python_api-0.5.4.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+rustat_python_api-0.5.4.dist-info/top_level.txt,sha256=VK0hmkKZE9YThxolUcoE6JtGI67NFeKJMBLuet8kI4w,18
+rustat_python_api-0.5.4.dist-info/RECORD,,

rustat_python_api-0.5.3.dist-info/RECORD

@@ -1,12 +0,0 @@
-rustat_python_api/__init__.py,sha256=Ij-PAm2y5ss_XAZhKTZus35cRPLzvXFyIswDa_Iq3rs,164
-rustat_python_api/config.py,sha256=eMvi1p8Cfvnbp6Cd4bBOwgehVN7thKnaQV5uzWyGZXM,1844
-rustat_python_api/models_api.py,sha256=oHXEqeCupvZwjVEdoxf7W9LP7ELFKA8-9DuRXpQHLno,1701
-rustat_python_api/parser.py,sha256=ATZzBctIiTs7rcPeszmiMExbQ_Nkra-7xWtw8JBTNI4,7103
-rustat_python_api/pitch_control.py,sha256=2tIiLtGBgANRet9csfZPZo2b2enxRDLREf5g1a6YpcM,11407
-rustat_python_api/processing.py,sha256=46O7wUGv5boWf4A38zYinbtqEC7ke-BwTdEg589XQaw,2816
-rustat_python_api/urls.py,sha256=MIUDXq5QFacyVWbrIIuMbwf-WRARsxb32kFY97zw4OQ,865
-rustat_python_api-0.5.3.dist-info/LICENSE,sha256=4Cohqg5p6Mq1xyrzdEX8AvFSA62GSVvapEOr2xK_tgY,57
-rustat_python_api-0.5.3.dist-info/METADATA,sha256=4697XDS-Yqmv79JQ6iRQC5wirzkkDZ29lkfmaIVEBO8,1730
-rustat_python_api-0.5.3.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-rustat_python_api-0.5.3.dist-info/top_level.txt,sha256=VK0hmkKZE9YThxolUcoE6JtGI67NFeKJMBLuet8kI4w,18
-rustat_python_api-0.5.3.dist-info/RECORD,,