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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rustat-python-api
-Version: 0.1.1
+Version: 0.2.0
 Summary: A Python wrapper for RuStat API
 Home-page: https://github.com/dailydaniel/rustat-python-api
 Author: Daniel Zholkovsky
@@ -41,4 +41,6 @@ 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.1 → rustat-python-api-0.2.0}/README.md

@@ -26,4 +26,6 @@ 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.1 → rustat-python-api-0.2.0}/rustat_python_api/parser.py

@@ -2,12 +2,19 @@ import requests
 import pandas as pd
 from collections import defaultdict
 from tqdm import tqdm
+import time
 
 from .urls import URLs
 
 
 class RuStatParser:
-    def __init__(self, user: str, password: str):
+    def __init__(
+        self,
+        user: str,
+        password: str,
+        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',
@@ -17,18 +24,23 @@ class RuStatParser:
 
         self.user = user
         self.password = password
+        self.urls = urls
+        self.sleep = sleep
 
         self.cached_info = {}
 
-    @staticmethod
-    def resp2data(query: str) -> dict:
+    def resp2data(self, query: str) -> dict:
+
+        if self.sleep > 0:
+            time.sleep(self.sleep)
+
         response = requests.get(query)
         return response.json()
 
     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 +50,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 +73,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 +97,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 +114,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.1 → rustat-python-api-0.2.0}/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.1 → rustat-python-api-0.2.0/rustat_python_api.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rustat-python-api
-Version: 0.1.1
+Version: 0.2.0
 Summary: A Python wrapper for RuStat API
 Home-page: https://github.com/dailydaniel/rustat-python-api
 Author: Daniel Zholkovsky
@@ -41,4 +41,6 @@ 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.1 → rustat-python-api-0.2.0}/setup.py

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