hkjc 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

hkjc/speedpro.py

@@ -0,0 +1,55 @@
+"""Functions to scrap SpeedPro data.
+"""
+from __future__ import annotations
+
+from datetime import datetime as dt
+
+import polars as pl
+import requests
+import json
+
+ENERGY_XLS_TEMPLATE = "https://racing.hkjc.com/racing/speedpro/assets/excel/{date}_energygrid_en.xls"
+SPEEDMAP_URL_TEMPLATE = "https://racing.hkjc.com/racing/speedpro/assets/json/speedguide/race_{race_num}.json"
+
+
+def speedpro_df(race_date: str) -> pl.DataFrame:
+    """Fetch and process SpeedPro scores for a given race date.
+
+    Args:
+        race_date (str): Date in 'YYYY-MM-DD' format.
+
+    Returns:
+        pl.DataFrame: Processed DataFrame with SpeedPro scores.
+    """
+    # validate date format
+    try:
+        dt.strptime(race_date, "%Y-%m-%d")
+    except Exception:
+        raise ValueError("Date must be in 'YYYY-MM-DD' format")
+
+    df = pl.read_excel(ENERGY_XLS_TEMPLATE.format(
+        date=dt.strptime(race_date, "%Y-%m-%d").strftime("%Y%m%d")))
+
+    # Clean column names
+    df.columns = [col.strip().replace(" ", "").replace(
+        "\n", "_").replace('.', '') for col in df.columns]
+
+    return df
+
+
+def speedmap(race_num: int) -> str:
+    """Fetch SpeedMap as base64 encoded string.
+
+    Args:
+        race_date (str): Date in 'YYYY-MM-DD' format.
+        race_num (int): Race number.
+    
+    Returns:
+        str: Base64 encoded string of the SpeedMap image.
+    """
+    r = requests.get(SPEEDMAP_URL_TEMPLATE.format(race_num=race_num))
+    if r.status_code != 200:
+        raise RuntimeError(f"Request failed: {r.status_code} - {r.text}")
+    # Decode with 'utf-8-sig' to strip a possible UTF-8 BOM before JSON parsing
+    content = r.content.decode("utf-8-sig")
+    return json.loads(content)['en-us']['RaceMap']

{hkjc-0.1.0.dist-info → hkjc-0.2.0.dist-info}/METADATA

@@ -1,11 +1,13 @@
 Metadata-Version: 2.4
 Name: hkjc
-Version: 0.1.0
+Version: 0.2.0
 Summary: Library for scrapping HKJC data and perform basic analysis
 Requires-Python: >=3.11
 Requires-Dist: cachetools>=6.2.0
+Requires-Dist: fastexcel>=0.16.0
 Requires-Dist: numpy>=2.3.3
 Requires-Dist: polars>=1.33.1
+Requires-Dist: pyarrow>=21.0.0
 Requires-Dist: requests>=2.32.5
 Requires-Dist: scipy>=1.16.2
 Requires-Dist: tqdm>=4.67.1

{hkjc-0.1.0.dist-info → hkjc-0.2.0.dist-info}/RECORD

@@ -6,7 +6,8 @@ hkjc/optimization.py,sha256=OArQ3w9bwcIV_lTNuE5za6AROoa90xk_gwAoGwQ-8RE,3784
 hkjc/processing.py,sha256=9AiTkjsx51sZtyA4XcfK-werwFWxdea0BeIEuNvGQYQ,2983
 hkjc/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hkjc/qpbanker.py,sha256=vhvYb5_nGrKgYgre9gGF6tgswovca5C9pZVOPGxEP1Q,4804
+hkjc/speedpro.py,sha256=vKnSz9yY1rfVmRo7GVxXLjsiQN-YgwxSbV0B7yuszS4,1702
 hkjc/visualization.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hkjc-0.1.0.dist-info/METADATA,sha256=fsrs2M07EvjjNyLqXMAQnyRg8Ebk-IcVq415v2QZEBQ,320
-hkjc-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hkjc-0.1.0.dist-info/RECORD,,
+hkjc-0.2.0.dist-info/METADATA,sha256=6rH6BxIZ1KMMh_TO_Br3o3IM9zjsrZMrr_zbBbL8Em0,384
+hkjc-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hkjc-0.2.0.dist-info/RECORD,,