rustat-python-api 0.1.0__py3-none-any.whl

rustat_python_api/__init__.py

@@ -0,0 +1,3 @@
+from .parser import RuStatParser
+
+__all__ = ['RuStatParser']

rustat_python_api/parser.py

@@ -0,0 +1,127 @@
+import requests
+import pandas as pd
+from collections import defaultdict
+from tqdm import tqdm
+
+from urls import URLs
+
+
+class RuStatParser:
+    def __init__(self, user: str, password: str):
+        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.cached_info = {}
+
+    @staticmethod
+    def resp2data(query: str) -> dict:
+        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(
+                    user=self.user,
+                    password=self.password,
+                    season_id=season_id
+                )
+            )
+
+            if data:
+                first_team_id = data["data"]["row"][0]["id"]
+                first_team_schedule = self.resp2data(
+                    URLs["schedule"].format(
+                        user=self.user,
+                        password=self.password,
+                        team_id=first_team_id,
+                        season_id=season_id
+                    )
+                )
+
+                if first_team_schedule:
+                    last_match = first_team_schedule["data"]["row"][0]
+                    season_name = f'{last_match["tournament_name"]} {last_match["season_name"]}'
+                else:
+                    season_name = ""
+
+                self.cached_info[season_id] = {
+                    "season_name": season_name,
+                    "season_teams": data["data"]["row"]
+                }
+
+        return self.cached_info
+
+    def get_schedule(self, team_id: str, season_id: str) -> dict:
+        data = self.resp2data(
+            URLs["schedule"].format(
+                user=self.user,
+                password=self.password,
+                team_id=team_id,
+                season_id=season_id
+            )
+        )
+
+        if not data:
+            return {}
+
+        return {
+            int(row["id"]): {
+                "match_date": row["match_date"],
+                "team1_id": int(row["team1_id"]),
+                "team2_id": int(row["team2_id"]),
+                "team1_name": row["team1_name"],
+                "team2_name": row["team2_name"]
+            }
+            for row in data["data"]["row"]
+        }
+
+    def get_events(self, match_id: int) -> pd.DataFrame | None:
+        data = self.resp2data(
+            URLs["events"].format(
+                user=self.user,
+                password=self.password,
+                match_id=match_id
+            )
+        )
+
+        if not data:
+            return None
+
+        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')
+
+        return df
+
+    def get_match_stats(self, match_id: int) -> dict:
+        data = self.resp2data(
+            URLs["match_stats"].format(
+                user=self.user,
+                password=self.password,
+                match_id=match_id
+            )
+        )
+
+        if not data:
+            return {}
+
+        stats = defaultdict(dict)
+
+        for row in data['data']['row']:
+            team_id = int(row['team_id'])
+            param_name = row['param_name']
+
+            param_value = float(row['value'])
+
+            stats[param_name][team_id] = param_value
+
+        return stats

rustat_python_api/urls.py

@@ -0,0 +1,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"
+}

rustat_python_api-0.1.0.dist-info/LICENSE

@@ -0,0 +1 @@
+Делайте че хотите с этим кодом.

rustat_python_api-0.1.0.dist-info/METADATA

@@ -0,0 +1,18 @@
+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
+Requires-Dist: requests==2.32.3
+Requires-Dist: pandas==2.2.3
+
+# rustat-python-api

rustat_python_api-0.1.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+rustat_python_api/__init__.py,sha256=wJVi55HFoMQltJCu09lzrnTUJ77zBCs-SKORDoLR3jw,61
+rustat_python_api/parser.py,sha256=1zgY3WVEkbKmUOF82GmuFT3gu7ym7IxOQAG9YVmkw9Q,3835
+rustat_python_api/urls.py,sha256=YZyfFd6UdvDwBF_DC4-xm3H_F5fV4bppeUA2Iur7Acw,604
+rustat_python_api-0.1.0.dist-info/LICENSE,sha256=4Cohqg5p6Mq1xyrzdEX8AvFSA62GSVvapEOr2xK_tgY,57
+rustat_python_api-0.1.0.dist-info/METADATA,sha256=3BbI1xyGY820_j9-7-B1HpwMEBJIFH5yo6y4gDxOhAA,551
+rustat_python_api-0.1.0.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+rustat_python_api-0.1.0.dist-info/top_level.txt,sha256=VK0hmkKZE9YThxolUcoE6JtGI67NFeKJMBLuet8kI4w,18
+rustat_python_api-0.1.0.dist-info/RECORD,,

rustat_python_api-0.1.0.dist-info/WHEEL

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

rustat_python_api-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+rustat_python_api