akshare-proxy-patch 0.1.5__tar.gz → 0.2.0__tar.gz

{akshare_proxy_patch-0.1.5 → akshare_proxy_patch-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: akshare-proxy-patch
-Version: 0.1.5
+Version: 0.2.0
 Summary: 针对 AkShare 的底层补丁，自动为东财等接口注入代理认证头，从而避免 `stock_zh_a_spot_em` 等方法报错。
 Home-page: https://github.com/helloyie/akshare-proxy-patch
 Author: cheapproxy.net
@@ -30,7 +30,7 @@ pip install --upgrade akshare-proxy-patch
 
 ## 使用
 
-1. 访问 https://cheapproxy.net 联系客服（备注: token），获取授权码 `AUTH_IP` 和 `AUTH_TOKEN`
+1. 登录 https://cheapproxy.net，获取授权码 `AUTH_IP` 和 `AUTH_TOKEN`
 
 2. 将 akshare 升级到最新
 
@@ -40,16 +40,21 @@ pip install --upgrade akshare
 
 3. 在使用 akshare 文件顶部添加下面的内容 
 ```
+# 文件顶部添加
 import akshare_proxy_patch
 
 akshare_proxy_patch.install_patch("AUTH_IP", "AUTH_TOKEN")
 
-
+# 你的正常业务代码......
 # import akshare as ak
 # df = ak.stock_zh_a_spot_em()
-# 你的其他代码......
 ```
 
-# 使用问题交流群
+## 目前Hook的接口域名清单 
+- fund.eastmoney.com
+- push2.eastmoney.com
+- push2his.eastmoney.com
+
+## 使用问题交流群
 
 ![8jmzvNOjKWFzwNUDW0Xiytcpfr50uMr5.webp](https://cdn.nodeimage.com/i/8jmzvNOjKWFzwNUDW0Xiytcpfr50uMr5.webp)

{akshare_proxy_patch-0.1.5 → akshare_proxy_patch-0.2.0}/README.md

@@ -9,7 +9,7 @@ pip install --upgrade akshare-proxy-patch
 
 ## 使用
 
-1. 访问 https://cheapproxy.net 联系客服（备注: token），获取授权码 `AUTH_IP` 和 `AUTH_TOKEN`
+1. 登录 https://cheapproxy.net，获取授权码 `AUTH_IP` 和 `AUTH_TOKEN`
 
 2. 将 akshare 升级到最新
 
@@ -19,16 +19,21 @@ pip install --upgrade akshare
 
 3. 在使用 akshare 文件顶部添加下面的内容 
 ```
+# 文件顶部添加
 import akshare_proxy_patch
 
 akshare_proxy_patch.install_patch("AUTH_IP", "AUTH_TOKEN")
 
-
+# 你的正常业务代码......
 # import akshare as ak
 # df = ak.stock_zh_a_spot_em()
-# 你的其他代码......
 ```
 
-# 使用问题交流群
+## 目前Hook的接口域名清单 
+- fund.eastmoney.com
+- push2.eastmoney.com
+- push2his.eastmoney.com
+
+## 使用问题交流群
 
 ![8jmzvNOjKWFzwNUDW0Xiytcpfr50uMr5.webp](https://cdn.nodeimage.com/i/8jmzvNOjKWFzwNUDW0Xiytcpfr50uMr5.webp)

akshare_proxy_patch-0.2.0/akshare_proxy_patch/__init__.py

@@ -0,0 +1,79 @@
+import time
+import threading
+import requests
+from requests.adapters import HTTPAdapter
+
+# 1. 备份原始发送方法
+original_send = HTTPAdapter.send
+
+# 2. 线程本地变量：用于标记当前线程是否正在请求“授权接口”，防止递归死循环
+_thread_status = threading.local()
+
+
+def get_auth_config_realtime(auth_url, auth_token):
+    """
+    实时获取授权信息，不使用缓存。
+    """
+    # 标记当前线程进入“授权请求状态”
+    _thread_status.is_authorizing = True
+    try:
+        # 这里的请求会进入 patched_send，但因为 is_authorizing 为 True，会走原逻辑
+        resp = requests.get(auth_url, params={"token": auth_token}, timeout=5)
+        data = resp.json()
+        if data.get("ua"):
+            return data
+    except Exception as e:
+        print(f"[Hook Error] 实时获取授权失败: {e}")
+    finally:
+        # 请求结束，重置标志位
+        _thread_status.is_authorizing = False
+    return None
+
+
+def install_patch(auth_ip, auth_token):
+    """安装全局 Hook"""
+
+    def patched_send(self, request, **kwargs):
+        # 防止递归：如果当前请求是由 get_auth_config 发出的，直接走原逻辑
+        if getattr(_thread_status, "is_authorizing", False):
+            return original_send(self, request, **kwargs)
+        url = request.url
+        # 需要hook的域名列表
+        is_target = any(
+            domain in url
+            for domain in [
+                "fund.eastmoney.com",
+                "push2.eastmoney.com",
+                "push2his.eastmoney.com",
+            ]
+        )
+
+        # 非目标域名，正常放行
+        if not is_target:
+            return original_send(self, request, **kwargs)
+
+        auth_url = f"http://{auth_ip}:47001/api/akshare-auth"
+
+        # 需要 patch 的接口，重试接口
+        for i in range(30):
+            try:
+                auth_res = get_auth_config_realtime(auth_url, auth_token)
+                if auth_res:
+                    request.headers["User-Agent"] = auth_res["ua"]
+                    request.headers["Cookie"] = (
+                        f"nid18={auth_res['nid18']}; nid18_create_time={auth_res['nid18_create_time']}"
+                    )
+                    proxy = auth_res["proxy"]
+                    kwargs["proxies"] = {"http": proxy, "https": proxy}
+                    kwargs["timeout"] = 8
+                else:
+                    print(f"[Hook Warning] 无法获取授权信息，将使用原始请求: {url}")
+                # 调用原始 send 方法完成请求
+                response = original_send(self, request, **kwargs)
+                if response.ok:
+                    return response
+            except:
+                time.sleep(0.01)
+
+    # 关键：全局替换 HTTPAdapter 的 send 方法
+    HTTPAdapter.send = patched_send

{akshare_proxy_patch-0.1.5 → akshare_proxy_patch-0.2.0}/akshare_proxy_patch.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: akshare-proxy-patch
-Version: 0.1.5
+Version: 0.2.0
 Summary: 针对 AkShare 的底层补丁，自动为东财等接口注入代理认证头，从而避免 `stock_zh_a_spot_em` 等方法报错。
 Home-page: https://github.com/helloyie/akshare-proxy-patch
 Author: cheapproxy.net
@@ -30,7 +30,7 @@ pip install --upgrade akshare-proxy-patch
 
 ## 使用
 
-1. 访问 https://cheapproxy.net 联系客服（备注: token），获取授权码 `AUTH_IP` 和 `AUTH_TOKEN`
+1. 登录 https://cheapproxy.net，获取授权码 `AUTH_IP` 和 `AUTH_TOKEN`
 
 2. 将 akshare 升级到最新
 
@@ -40,16 +40,21 @@ pip install --upgrade akshare
 
 3. 在使用 akshare 文件顶部添加下面的内容 
 ```
+# 文件顶部添加
 import akshare_proxy_patch
 
 akshare_proxy_patch.install_patch("AUTH_IP", "AUTH_TOKEN")
 
-
+# 你的正常业务代码......
 # import akshare as ak
 # df = ak.stock_zh_a_spot_em()
-# 你的其他代码......
 ```
 
-# 使用问题交流群
+## 目前Hook的接口域名清单 
+- fund.eastmoney.com
+- push2.eastmoney.com
+- push2his.eastmoney.com
+
+## 使用问题交流群
 
 ![8jmzvNOjKWFzwNUDW0Xiytcpfr50uMr5.webp](https://cdn.nodeimage.com/i/8jmzvNOjKWFzwNUDW0Xiytcpfr50uMr5.webp)

{akshare_proxy_patch-0.1.5 → akshare_proxy_patch-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "akshare-proxy-patch"  # 包名，需在 PyPI 唯一
-version = "0.1.5"
+version = "0.2.0"
 authors = [
   { name="cheapproxy", email="emezxyj1@gmail.com" },
 ]

{akshare_proxy_patch-0.1.5 → akshare_proxy_patch-0.2.0}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="akshare-proxy-patch",  # 包名，用户 pip install 时的名称
-    version="0.1.5",  # 版本号
+    version="0.2.0",  # 版本号
     author="cheapproxy.net",
     author_email="emezxyj1@gmail.com",
     description="AkShare 报错全局Hook",

akshare_proxy_patch-0.1.5/akshare_proxy_patch/__init__.py

@@ -1,49 +0,0 @@
-import time
-import requests
-from requests.adapters import HTTPAdapter
-
-original_send = HTTPAdapter.send
-
-
-def install_patch(auth_ip, auth_token):
-    def patched_send(self, request, **kwargs):
-        auth_url = f"http://{auth_ip}:47001/api/akshare-auth"
-        url = request.url
-        is_hook = (
-            ("fund.eastmoney.com" in url)
-            or ("push2.eastmoney.com" in url)
-            or ("push2his.eastmoney.com" in url)
-        )
-        for i in range(60):
-            try:
-                if not is_hook:
-                    return original_send(self, request, **kwargs)
-                with requests.Session() as s:
-                    HTTPAdapter.send = original_send
-                    auth_res = s.get(
-                        auth_url, params={"token": auth_token}, timeout=5
-                    ).json()
-                    if not auth_res["ua"]:
-                        error_msg = (
-                            auth_res.get("error_msg", "")
-                            or "未知异常，请联系 cheapproxy.net 客服"
-                        )
-                        print(error_msg)
-                        return None
-
-                    HTTPAdapter.send = patched_send
-                request.headers["User-Agent"] = auth_res["ua"]
-                cookie_str = f"nid18={auth_res['nid18']}; nid18_create_time={auth_res['nid18_create_time']}"
-                request.headers["Cookie"] = cookie_str
-                proxy = auth_res["proxy"]
-                kwargs["proxies"] = {"http": proxy, "https": proxy}
-                kwargs["timeout"] = 8
-                return original_send(self, request, **kwargs)
-            except Exception as e:
-                if not is_hook:
-                    time.sleep(0.5)
-                HTTPAdapter.send = patched_send
-
-        return original_send(self, request, **kwargs)
-
-    HTTPAdapter.send = patched_send