rustat-python-api 0.1.0__tar.gz → 0.1.2__tar.gz

rustat-python-api-0.1.2/PKG-INFO

@@ -0,0 +1,46 @@
+Metadata-Version: 2.1
+Name: rustat-python-api
+Version: 0.1.2
+Summary: A Python wrapper for RuStat API
+Home-page: https://github.com/dailydaniel/rustat-python-api
+Author: Daniel Zholkovsky
+Author-email: daniel@zholkovsky.com
+License: MIT
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+
+# rustat-python-api
+
+### Python wrapper for the Rustat API
+### Example of usage:
+0. Install the package:
+```bash
+pip install rustat-python-api
+```
+1. Usage:
+```python
+from rustat_python_api import RuStatParser
+
+user = "your_login"
+password = "your_password"
+
+parser = RuStatParser(user, password)
+
+info = parser.get_rpl_info()
+keys = list(info.keys())
+season_id, team_id = keys[-1], info[keys[-1]]["season_teams"][0]["id"]
+
+schedule = parser.get_schedule(team_id, season_id)
+keys = list(schedule.keys())
+match_id = keys[-1]
+
+events = parser.get_events(match_id)
+
+stats = parser.get_match_stats(match_id)
+
+tracking = parser.get_tracking(match_id)
+```

rustat-python-api-0.1.2/README.md

@@ -0,0 +1,31 @@
+# rustat-python-api
+
+### Python wrapper for the Rustat API
+### Example of usage:
+0. Install the package:
+```bash
+pip install rustat-python-api
+```
+1. Usage:
+```python
+from rustat_python_api import RuStatParser
+
+user = "your_login"
+password = "your_password"
+
+parser = RuStatParser(user, password)
+
+info = parser.get_rpl_info()
+keys = list(info.keys())
+season_id, team_id = keys[-1], info[keys[-1]]["season_teams"][0]["id"]
+
+schedule = parser.get_schedule(team_id, season_id)
+keys = list(schedule.keys())
+match_id = keys[-1]
+
+events = parser.get_events(match_id)
+
+stats = parser.get_match_stats(match_id)
+
+tracking = parser.get_tracking(match_id)
+```

{rustat-python-api-0.1.0 → rustat-python-api-0.1.2}/rustat_python_api/parser.py

@@ -3,11 +3,11 @@ import pandas as pd
 from collections import defaultdict
 from tqdm import tqdm
 
-from urls import URLs
+from .urls import URLs
 
 
 class RuStatParser:
-    def __init__(self, user: str, password: str):
+    def __init__(self, user: str, password: str, urls: dict = URLs):
         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',
@@ -17,6 +17,7 @@ class RuStatParser:
 
         self.user = user
         self.password = password
+        self.urls = urls
 
         self.cached_info = {}
 
@@ -28,7 +29,7 @@ class RuStatParser:
     def get_rpl_info(self):
         for season_id in tqdm(range(1, 36)):
             data = self.resp2data(
-                URLs["tournament_teams"].format(
+                self.urls["tournament_teams"].format(
                     user=self.user,
                     password=self.password,
                     season_id=season_id
@@ -38,7 +39,7 @@ class RuStatParser:
             if data:
                 first_team_id = data["data"]["row"][0]["id"]
                 first_team_schedule = self.resp2data(
-                    URLs["schedule"].format(
+                    self.urls["schedule"].format(
                         user=self.user,
                         password=self.password,
                         team_id=first_team_id,
@@ -61,7 +62,7 @@ class RuStatParser:
 
     def get_schedule(self, team_id: str, season_id: str) -> dict:
         data = self.resp2data(
-            URLs["schedule"].format(
+            self.urls["schedule"].format(
                 user=self.user,
                 password=self.password,
                 team_id=team_id,
@@ -85,7 +86,7 @@ class RuStatParser:
 
     def get_events(self, match_id: int) -> pd.DataFrame | None:
         data = self.resp2data(
-            URLs["events"].format(
+            self.urls["events"].format(
                 user=self.user,
                 password=self.password,
                 match_id=match_id
@@ -102,9 +103,43 @@ class RuStatParser:
 
         return df
 
+    def get_tracking(self, match_id: int) -> pd.DataFrame | None:
+        data = self.resp2data(
+            self.urls["tracking"].format(
+                user=self.user,
+                password=self.password,
+                match_id=match_id
+            )
+        )
+
+        if not data:
+            return None
+
+        data = data["data"]["team"]
+        df = pd.DataFrame(columns=["half", "second", "pos_x", "pos_y", "team_id", "player_id", "player_name", "side_1h"])
+
+        for team_data in tqdm(data):
+            team_id = team_data["id"]
+            side_1h = team_data["gate_position_half_1"]
+
+            for player_data in team_data["player"]:
+                player_id = player_data["id"]
+                player_name = player_data["name"]
+
+                cur_df = pd.json_normalize(player_data["row"])
+                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
+
+                df = pd.concat([df, cur_df], ignore_index=True)
+
+        return df.sort_values(by=["second", "team_id", "player_id"])
+
     def get_match_stats(self, match_id: int) -> dict:
         data = self.resp2data(
-            URLs["match_stats"].format(
+            self.urls["match_stats"].format(
                 user=self.user,
                 password=self.password,
                 match_id=match_id

{rustat-python-api-0.1.0 → rustat-python-api-0.1.2}/rustat_python_api/urls.py

@@ -2,5 +2,6 @@ URLs = {
     "schedule": "http://feeds.rustatsport.ru/?tpl=35&user={user}&key={password}&team_id={team_id}&season_id={season_id}&date_start=&date_end=&format=json",
     "events": "http://feeds.rustatsport.ru/?tpl=36&user={user}&key={password}&match_id={match_id}&start_ms=0&dl=0&lang_id=1&format=json",
     "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"
+    "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"
 }

rustat-python-api-0.1.2/rustat_python_api.egg-info/PKG-INFO

@@ -0,0 +1,46 @@
+Metadata-Version: 2.1
+Name: rustat-python-api
+Version: 0.1.2
+Summary: A Python wrapper for RuStat API
+Home-page: https://github.com/dailydaniel/rustat-python-api
+Author: Daniel Zholkovsky
+Author-email: daniel@zholkovsky.com
+License: MIT
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+
+# rustat-python-api
+
+### Python wrapper for the Rustat API
+### Example of usage:
+0. Install the package:
+```bash
+pip install rustat-python-api
+```
+1. Usage:
+```python
+from rustat_python_api import RuStatParser
+
+user = "your_login"
+password = "your_password"
+
+parser = RuStatParser(user, password)
+
+info = parser.get_rpl_info()
+keys = list(info.keys())
+season_id, team_id = keys[-1], info[keys[-1]]["season_teams"][0]["id"]
+
+schedule = parser.get_schedule(team_id, season_id)
+keys = list(schedule.keys())
+match_id = keys[-1]
+
+events = parser.get_events(match_id)
+
+stats = parser.get_match_stats(match_id)
+
+tracking = parser.get_tracking(match_id)
+```

{rustat-python-api-0.1.0 → rustat-python-api-0.1.2}/rustat_python_api.egg-info/requires.txt

@@ -1,2 +1,3 @@
 requests==2.32.3
 pandas==2.2.3
+tqdm==4.66.5

{rustat-python-api-0.1.0 → rustat-python-api-0.1.2}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='rustat-python-api',
-    version='0.1.0',
+    version='0.1.2',
     description='A Python wrapper for RuStat API',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',
@@ -14,6 +14,7 @@ setup(
     install_requires=[
         'requests==2.32.3',
         'pandas==2.2.3',
+        'tqdm==4.66.5'
     ],
     classifiers=[
         'Programming Language :: Python :: 3',

rustat-python-api-0.1.0/PKG-INFO

@@ -1,16 +0,0 @@
-Metadata-Version: 2.1
-Name: rustat-python-api
-Version: 0.1.0
-Summary: A Python wrapper for RuStat API
-Home-page: https://github.com/dailydaniel/rustat-python-api
-Author: Daniel Zholkovsky
-Author-email: daniel@zholkovsky.com
-License: MIT
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-License-File: LICENSE
-
-# rustat-python-api

rustat-python-api-0.1.0/README.md

@@ -1 +0,0 @@
-# rustat-python-api

rustat-python-api-0.1.0/rustat_python_api.egg-info/PKG-INFO

@@ -1,16 +0,0 @@
-Metadata-Version: 2.1
-Name: rustat-python-api
-Version: 0.1.0
-Summary: A Python wrapper for RuStat API
-Home-page: https://github.com/dailydaniel/rustat-python-api
-Author: Daniel Zholkovsky
-Author-email: daniel@zholkovsky.com
-License: MIT
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-License-File: LICENSE
-
-# rustat-python-api