rustat-python-api 0.5.0__tar.gz → 0.5.1__tar.gz

{rustat-python-api-0.5.0/rustat_python_api.egg-info → rustat-python-api-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rustat-python-api
-Version: 0.5.0
+Version: 0.5.1
 Summary: A Python wrapper for RuStat API
 Home-page: https://github.com/dailydaniel/rustat-python-api
 Author: Daniel Zholkovsky
@@ -23,7 +23,7 @@ pip install rustat-python-api
 ```
 1. Usage:
 ```python
-from rustat_python_api import RuStatParser
+from rustat_python_api import RuStatParser, DynamoLab, PitchControl
 
 user = "your_login"
 password = "your_password"
@@ -39,8 +39,20 @@ keys = list(schedule.keys())
 match_id = keys[-1]
 
 events, subs = parser.get_events(match_id, process=True, return_subs=True)
-
 stats = parser.get_match_stats(match_id)
-
 tracking = parser.get_tracking(match_id)
+
+host = "http://localhost:8001/"
+client = DynamoLab(host)
+client.run_model(
+    model="xT",
+    data=events,
+    inplace=True,
+    inplace_column=model
+)
+
+pc = PitchControl(tracking, events)
+pc.draw_pitch_control(half=1, tp=100, save=True, filename="pitch_control")
+# ffmpeg required for animation
+pc.animate_pitch_control(half=1, tp=100, frames=30, filename="pitch_control")
 ```

{rustat-python-api-0.5.0 → rustat-python-api-0.5.1}/README.md

@@ -8,7 +8,7 @@ pip install rustat-python-api
 ```
 1. Usage:
 ```python
-from rustat_python_api import RuStatParser
+from rustat_python_api import RuStatParser, DynamoLab, PitchControl
 
 user = "your_login"
 password = "your_password"
@@ -24,8 +24,20 @@ keys = list(schedule.keys())
 match_id = keys[-1]
 
 events, subs = parser.get_events(match_id, process=True, return_subs=True)
-
 stats = parser.get_match_stats(match_id)
-
 tracking = parser.get_tracking(match_id)
+
+host = "http://localhost:8001/"
+client = DynamoLab(host)
+client.run_model(
+    model="xT",
+    data=events,
+    inplace=True,
+    inplace_column=model
+)
+
+pc = PitchControl(tracking, events)
+pc.draw_pitch_control(half=1, tp=100, save=True, filename="pitch_control")
+# ffmpeg required for animation
+pc.animate_pitch_control(half=1, tp=100, frames=30, filename="pitch_control")
 ```

{rustat-python-api-0.5.0 → rustat-python-api-0.5.1}/rustat_python_api/pitch_control.py

@@ -45,7 +45,7 @@ class PitchControl:
 
         locs_home = {
             half: {
-                player_id: self.get_player_data(player_id, half=half)
+                player_id: self.get_player_data(player_id, half, tracking)
                 for player_id in tracking[tracking['side_1h'] == 'left']['player_id'].unique()
             }
             for half in tracking['half'].unique()
@@ -53,7 +53,7 @@ class PitchControl:
 
         locs_away = {
             half: {
-                player_id: self.get_player_data(player_id, half=half)
+                player_id: self.get_player_data(player_id, half, tracking)
                 for player_id in tracking[tracking['side_1h'] == 'right']['player_id'].unique()
             }
             for half in tracking['half'].unique()
@@ -213,7 +213,8 @@ class PitchControl:
         tp: int,
         pitch_control: tuple = None,
         save: bool = False,
-        dt: int = 200
+        dt: int = 200,
+        filename: str = 'pitch_control'
     ):
         if pitch_control is None:
             pitch_control, xx, yy = self.fit(half, tp, dt)
@@ -245,7 +246,7 @@ class PitchControl:
         )
 
         if save:
-            plt.savefig('pitch_control.png', dpi=300)
+            plt.savefig(f'{filename}.png', dpi=300)
         else:
             plt.show()
 
@@ -253,7 +254,7 @@ class PitchControl:
         self,
         half: int,
         tp: int,
-        filename: str,
+        filename: str = "pitch_control_animation",
         dt: int = 200,
         frames: int = 30,
         interval: int = 1000

{rustat-python-api-0.5.0 → rustat-python-api-0.5.1/rustat_python_api.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rustat-python-api
-Version: 0.5.0
+Version: 0.5.1
 Summary: A Python wrapper for RuStat API
 Home-page: https://github.com/dailydaniel/rustat-python-api
 Author: Daniel Zholkovsky
@@ -23,7 +23,7 @@ pip install rustat-python-api
 ```
 1. Usage:
 ```python
-from rustat_python_api import RuStatParser
+from rustat_python_api import RuStatParser, DynamoLab, PitchControl
 
 user = "your_login"
 password = "your_password"
@@ -39,8 +39,20 @@ keys = list(schedule.keys())
 match_id = keys[-1]
 
 events, subs = parser.get_events(match_id, process=True, return_subs=True)
-
 stats = parser.get_match_stats(match_id)
-
 tracking = parser.get_tracking(match_id)
+
+host = "http://localhost:8001/"
+client = DynamoLab(host)
+client.run_model(
+    model="xT",
+    data=events,
+    inplace=True,
+    inplace_column=model
+)
+
+pc = PitchControl(tracking, events)
+pc.draw_pitch_control(half=1, tp=100, save=True, filename="pitch_control")
+# ffmpeg required for animation
+pc.animate_pitch_control(half=1, tp=100, frames=30, filename="pitch_control")
 ```

{rustat-python-api-0.5.0 → rustat-python-api-0.5.1}/setup.py

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