Update project

backend/app/clients/ths_etf_client.py

@@ -0,0 +1,56 @@
+from __future__ import annotations
+
+import json
+import re
+from typing import Any
+from urllib.request import Request, urlopen
+
+
+class ThsEtfClient:
+    TODAY_ENDPOINT = "https://d.10jqka.com.cn/v6/line/{market}_{code}/01/today.js"
+    TIME_ENDPOINT = "https://d.10jqka.com.cn/v6/time/{market}_{code}/last.js"
+    HISTORY_ENDPOINT = "https://d.10jqka.com.cn/v6/line/{market}_{code}/01/last.js"
+    PROFILE_ENDPOINT = "https://fund.10jqka.com.cn/interface/net/index/0_{code}"
+
+    def __init__(self) -> None:
+        self.default_headers = {
+            "User-Agent": "Mozilla/5.0",
+            "Accept-Language": "zh-CN,zh;q=0.9",
+        }
+
+    def _get_text(self, url: str, *, referer: str | None = None) -> str:
+        headers = dict(self.default_headers)
+        if referer:
+            headers["Referer"] = referer
+        request = Request(url, headers=headers)
+        with urlopen(request, timeout=20) as response:
+            return response.read().decode("utf-8", "ignore")
+
+    @staticmethod
+    def _extract_wrapped_json(payload: str) -> dict[str, Any]:
+        match = re.search(r"\((\{.*\})\)\s*$", payload.strip(), re.S)
+        if not match:
+            raise ValueError("unexpected ths payload")
+        return json.loads(match.group(1))
+
+    @staticmethod
+    def _fund_detail_url(code: str) -> str:
+        return f"https://fund.10jqka.com.cn/{code}/"
+
+    def fetch_profile(self, code: str) -> dict[str, Any]:
+        url = self.PROFILE_ENDPOINT.format(code=code)
+        return json.loads(self._get_text(url, referer=self._fund_detail_url(code)))
+
+    def fetch_today_quote(self, market: str, code: str) -> dict[str, Any]:
+        url = self.TODAY_ENDPOINT.format(market=market, code=code)
+        payload = self._extract_wrapped_json(self._get_text(url, referer=self._fund_detail_url(code)))
+        return payload.get(f"{market}_{code}", {})
+
+    def fetch_intraday_time(self, market: str, code: str) -> dict[str, Any]:
+        url = self.TIME_ENDPOINT.format(market=market, code=code)
+        payload = self._extract_wrapped_json(self._get_text(url, referer=self._fund_detail_url(code)))
+        return payload.get(f"{market}_{code}", {})
+
+    def fetch_history(self, market: str, code: str) -> dict[str, Any]:
+        url = self.HISTORY_ENDPOINT.format(market=market, code=code)
+        return self._extract_wrapped_json(self._get_text(url, referer=self._fund_detail_url(code)))