akshare-proxy-patch 0.5.0__tar.gz → 0.5.1__tar.gz

{akshare_proxy_patch-0.5.0 → akshare_proxy_patch-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: akshare-proxy-patch
-Version: 0.5.0
+Version: 0.5.1
 Summary: 针对 AkShare 和 efinance 的底层补丁，自动为东财等接口注入代理认证头，从而避免 `stock_zh_a_spot_em` 等方法报错。
 Home-page: https://github.com/helloyie/akshare-proxy-patch
 Author: cheapproxy.net
@@ -21,14 +21,15 @@ Dynamic: requires-python
 
 # AkShare Proxy Patch
 
-> 针对 [akshare](https://github.com/akfamily/akshare)、[efinance](https://github.com/Micro-sheep/efinance)、[yfinance](https://github.com/ranaroussi/yfinance) 的🐒插件补丁，解决 `stock_zh_a_spot_em`、`stock_zh_a_hist`、`get_realtime_quotes` 等接口报错问题和 Yahoo `YFRateLimitError` 问题。并对 `akshare` 部分接口进行多线程加速处理。
+> 针对 [akshare](https://github.com/akfamily/akshare)、[efinance](https://github.com/Micro-sheep/efinance)、[yfinance](https://github.com/ranaroussi/yfinance)、[aktools](https://github.com/akfamily/aktools)、[daily_stock_analysis](https://github.com/ZhuLinsen/daily_stock_analysis/) 的🐒插件补丁，解决 `stock_zh_a_spot_em`、`stock_zh_a_hist`、`get_realtime_quotes` 等东财接口报错问题和 Yahoo `YFRateLimitError` 问题。并对 `akshare` 部分接口进行多线程加速处理。
 
 ## ✨ 特性
 
-- 解决 `akshare` 的 `stock_zh_a_spot_em`、`stock_zh_a_hist` 接口报错问题
+- 解决 `akshare` 或 `aktools` 的 `stock_zh_a_spot_em`、`stock_zh_a_hist` 接口报错问题
 - 解决 `efinance` 的 `get_realtime_quotes` 等接口报错问题
 - 解决 `yfinance` Yahoo接口国内无法使用问题
-- 加速 `stock_zh_a_spot_em`、`stock_individual_fund_flow_rank` 等 `akshare` 接口
+- 解决 `daily_stock_analysis` 的数据获取报错问题
+- 多线程加速 `stock_zh_a_spot_em` 等 `akshare` 接口
 
 ## 📦 安装
 
@@ -37,18 +38,19 @@ Dynamic: requires-python
 2. 安装 `akshare-proxy-patch` 插件
 
 ```
-pip install akshare-proxy-patch==0.5.0
+pip install akshare-proxy-patch==0.5.1
 ```
 
 ## 🚀 使用方法（akshare / efinance / yfinance）
 
 1. [点击前往插件官网](https://ak.cheapproxy.net/dashboard/akshare) 获取 `TOKEN`。
 
-2. `akshare` 或 `efinance` 用户：在 **Python 文件最顶部**添加如下代码，并替换 `你的TOKEN`。调用非常简单，不需要使用 AI 魔改。
+2. `akshare` 或 `efinance` 用户：在 **Python 文件最顶部** 添加如下代码，并替换 `你的TOKEN`。
 
 ```
 # python 文件顶部添加如下代码
-# 插件引入和调用一定要放到最顶部！不能在 akshare 或 efinance 之后引入！
+# 重要：插件引入和调用一定要放到最顶部！不能在 akshare 或 efinance 之后引入！
+
 import akshare_proxy_patch
 
 akshare_proxy_patch.install_patch(
@@ -61,12 +63,14 @@ akshare_proxy_patch.install_patch(
       "push2.eastmoney.com",
       "push2his.eastmoney.com",
       "emweb.securities.eastmoney.com",
+      "searchapi.eastmoney.com/api/suggest/get",
+      "push2delay.eastmoney.com/api/qt/clist/get"
     ],
     fast=True
 )
 
 # --------------------------
-# 后续你的正常业务代码保持不变
+# 后续你的业务代码保持不变
 # --------------------------
 
 # 假如你使用 akshare
@@ -78,11 +82,12 @@ import efinance as ef
 ef.stock.get_realtime_quotes()
 ```
 
-3. `yfinance` 用户：在 Python 文件顶部添加如下代码，并替换 `你的TOKEN`。调用非常简单，不需要使用 AI 魔改。
+3. `yfinance` 用户：在 Python 文件顶部添加如下代码，并替换 `你的TOKEN`，不需要使用 AI 魔改。
 
 ```
 # python 文件顶部添加如下代码
-# 插件引入和调用一定要放到最顶部！不能在 akshare 或 efinance 之后引入！
+# 重要：插件引入和调用一定要放到最顶部！不能在 akshare 或 efinance 之后引入！
+
 import akshare_proxy_patch
 
 akshare_proxy_patch.install_yfinance_patch(
@@ -92,7 +97,7 @@ akshare_proxy_patch.install_yfinance_patch(
 )
 
 # --------------------------
-# 后续你的正常业务代码保持不变
+# 后续你的业务代码保持不变
 # --------------------------
 
 import yfinance as yf
@@ -100,59 +105,44 @@ import yfinance as yf
 data = yf.download("AAPL", start="2017-01-01", end="2017-04-30")
 ```
 
-4. [尽量用 `efinance` 替代 `akshare` 的部分函数获取数据，效率更高，更省积分](https://github.com/Micro-sheep/efinance)
+4. 尽量用 [efinance](https://github.com/Micro-sheep/efinance) 替代 akshare 的部分函数配合插件获取数据，效率更高，更省积分。
 
 ## install_patch 参数说明
 
-- 参数1：网关
-  - 默认为 `101.201.173.125` 不可修改
-- 参数2：TOKEN
-  - 授权凭证
-- 参数3：重试次数
-  - 默认为30，建议保持不变
-- 参数4：封控的域名列表
-  - 接口 `URL` 包含数组中的其中一条，就会走插件。
-  - 可点击 `ak` 或 `ef` 函数查看接口源码对应的 `URL`，根据封控情况细化可以降低积分消耗。
-  - 如只封控 `stock_zh_a_spot_em` 这个接口，`hook_domains` 可设置为 `["https://82.push2.eastmoney.com/api/qt/clist/get"]`。
-- 参数5：是否启用快速模式，加速列表如下：
-  - 所有包含 `fetch_paginated_data` 函数的接口，如 `stock_zh_a_spot_em`
-  - `stock_individual_fund_flow_rank`
-  - `stock_sector_fund_flow_rank`
-  - `fund_money_fund_info_em`
-  - `fund_graded_fund_info_em`
-  - `fund_etf_fund_info_em`
-  - `fund_fh_em`
-  - `fund_cf_em`
-  - `fund_fh_rank_em`
+- `ip`
+  - 网关，不可修改
+- `auth_token`
+  - TOKEN 授权凭证
+- `retry`
+  - 重试次数，默认为30，建议保持不变
+- `hook_domains`
+  - 封控的域名列表，插件会 hook 住这些域名的请求，解除封控。已默认涵盖常见函数的域名。
+  - 可点击 `akshare` 或 `efinance` 函数查看接口源码对应的 `URL` 并添加到该列表。
+  - 如只用到 `stock_zh_a_spot_em` 这个接口，`hook_domains` 可设置为 `["https://82.push2.eastmoney.com/api/qt/clist/get"]`。
+- `fast`
+  - 是否启用 `akshare` 多线程加速，默认开启，加速函数列表如下
+  - 所有用到 `fetch_paginated_data` 分页函数的接口，如 `stock_zh_a_spot_em`、`stock_sh_a_spot_em`、`stock_board_industry_cons_em` 等
 
 ## 如何在 aktools 内集成插件？
 
 - `aktools` 想要集成插件，需下载 [akt.py](https://github.com/HelloYie/akshare-proxy-patch/blob/master/examples/aktools/akt.py) 文件，并填入您的 `TOKEN`。
 - 然后执行 `python akt.py` 即可启动一个 `http://127.0.0.1:8080/` 服务。只是启动方式不同而已，使用请参考 [aktools 官方文档](https://github.com/akfamily/aktools)。
 
-## 如何多进程使用，快速拉取数据？
-
-- 尽量使用 `efinance` 替代 `akshare` 来获取数据， `efinance` 内置多线程，效率更高，更省积分。
-- 在大规模快速获取数据时，仍需要多进程来提升效率。 参考 `examples/并发获取股票行情/` 目录下的 `main.py` 和 `worker.py` 示例，修改 `worker.py` 中的 `auth_token`，执行 `python main.py` 即可。
-- `main.py` 负责分割任务和启动多个 `worker.py` 进程，`worker.py` 中引入插件并执行数据获取逻辑，最后将结果合并保存。
-
-## 如何灵活禁用/启用插件？
+## 如何在 daily_stock_analysis 内集成插件？
 
-加入如下代码后插件就会被禁用：
+- [daily_stock_analysis](https://github.com/ZhuLinsen/daily_stock_analysis) 本身集成了 `akshare`、`efinance` 等数据源，如遇数据源报错，可以通过引入插件来解决。
+- 在 [daily_stock_analysis](https://github.com/ZhuLinsen/daily_stock_analysis) 项目中，找到入口 Python 文件 `main.py`，在文件顶部引入插件：
 
 ```
-akshare_proxy_patch.uninstall_patch()
-```
-
-下面是插件启用 --> 禁用 --> 再次启用的例子：
+# 找到 daily_stock_analysis main.py 的这行（第一行）代码，在其下方引入插件
+from __future__ import annotations
 
-```
-##### 1. 启用插件
+# 引入插件
 import akshare_proxy_patch
 
 akshare_proxy_patch.install_patch(
     "101.201.173.125",
-    auth_token='你的TOKEN',
+    auth_token="你的TOKEN",
     retry=30,
     # 封控的域名列表，可自行调整
     hook_domains=[
@@ -160,31 +150,36 @@ akshare_proxy_patch.install_patch(
       "push2.eastmoney.com",
       "push2his.eastmoney.com",
       "emweb.securities.eastmoney.com",
+      "searchapi.eastmoney.com/api/suggest/get",
+      "push2delay.eastmoney.com/api/qt/clist/get"
     ],
     fast=True
 )
-import akshare as ak
-df =  ak.fund_etf_hist_em()
+# daily_stock_analysis main.py 的其他后续代码
+# 插件引入成功后正常启动该项目即可
 
-###### 2. 禁用插件
-akshare_proxy_patch.uninstall_patch()
+```
+
+## 股票数量很多，如何快速拉取数据？
+
+- 尽量使用 `efinance` 替代 `akshare` 来获取数据， `efinance` 内置多线程，效率更高，更省积分。
+- 在大规模快速获取数据时，可使用多进程提升效率。 可以参考 [多进程获取所有股票的历史K线](https://github.com/HelloYie/akshare-proxy-patch/blob/master/examples/all_quote_history/main.py)，修改 `auth_token`，执行 `python main.py` 测试。
+
+## 如何灵活禁用/启用插件？
 
-# 禁用后测试
-try:
-  df = ak.fund_etf_hist_em()
-except Exception as e:
-   print('插件被禁用了，这里可能会报错')
+加入如下代码后插件就会被禁用，[点击查看禁用/启用插件示例](https://github.com/HelloYie/akshare-proxy-patch/blob/master/examples/disable/disable.py)：
 
-##### 3. 再次启用插件，重复上述 1、2 的代码即可
+```
+akshare_proxy_patch.uninstall_patch()
 ```
 
 ## 我没使用 akshare 或 efinance，能集成插件吗？
 
 - 如果使用 Python 语言的 `requests` 库请求接口，插件能自动 hook 住请求，正常工作。
-- 如果您使用其他语言或 python 的其他库，可 [手动提取代理](http://101.201.173.125:47001/api/akshare-auth?token=XXX&version=0.5.0) 自行实现封控解除。
+- 如果您使用其他语言或 python 的其他库，可 [手动提取代理](http://101.201.173.125:47001/api/akshare-auth?token=XXX&version=0.5.1) 自行实现封控解除。
 
 ## 使用问题交流群
 
 如使用时遇到问题，或对插件有什么意见或建议，可进群交流：
 
-![HExHHHovkcq1Kt9TEEzc8OKtFV2O1BBp.webp](https://cdn.nodeimage.com/i/HExHHHovkcq1Kt9TEEzc8OKtFV2O1BBp.webp)
+![eozuhg3eH48ntnAnZXjKpmiegT7Ws0cl.webp](https://cdn.nodeimage.com/i/eozuhg3eH48ntnAnZXjKpmiegT7Ws0cl.webp)

akshare_proxy_patch-0.5.1/README.md

@@ -0,0 +1,164 @@
+# AkShare Proxy Patch
+
+> 针对 [akshare](https://github.com/akfamily/akshare)、[efinance](https://github.com/Micro-sheep/efinance)、[yfinance](https://github.com/ranaroussi/yfinance)、[aktools](https://github.com/akfamily/aktools)、[daily_stock_analysis](https://github.com/ZhuLinsen/daily_stock_analysis/) 的🐒插件补丁，解决 `stock_zh_a_spot_em`、`stock_zh_a_hist`、`get_realtime_quotes` 等东财接口报错问题和 Yahoo `YFRateLimitError` 问题。并对 `akshare` 部分接口进行多线程加速处理。
+
+## ✨ 特性
+
+- 解决 `akshare` 或 `aktools` 的 `stock_zh_a_spot_em`、`stock_zh_a_hist` 接口报错问题
+- 解决 `efinance` 的 `get_realtime_quotes` 等接口报错问题
+- 解决 `yfinance` Yahoo接口国内无法使用问题
+- 解决 `daily_stock_analysis` 的数据获取报错问题
+- 多线程加速 `stock_zh_a_spot_em` 等 `akshare` 接口
+
+## 📦 安装
+
+1. 安装并升级官方 [akshare](https://github.com/akfamily/akshare) 或 [efinance](https://github.com/Micro-sheep/efinance) 包
+
+2. 安装 `akshare-proxy-patch` 插件
+
+```
+pip install akshare-proxy-patch==0.5.1
+```
+
+## 🚀 使用方法（akshare / efinance / yfinance）
+
+1. [点击前往插件官网](https://ak.cheapproxy.net/dashboard/akshare) 获取 `TOKEN`。
+
+2. `akshare` 或 `efinance` 用户：在 **Python 文件最顶部** 添加如下代码，并替换 `你的TOKEN`。
+
+```
+# python 文件顶部添加如下代码
+# 重要：插件引入和调用一定要放到最顶部！不能在 akshare 或 efinance 之后引入！
+
+import akshare_proxy_patch
+
+akshare_proxy_patch.install_patch(
+    "101.201.173.125",
+    auth_token="你的TOKEN",
+    retry=30,
+    # 封控的域名列表，可自行调整
+    hook_domains=[
+      "fund.eastmoney.com",
+      "push2.eastmoney.com",
+      "push2his.eastmoney.com",
+      "emweb.securities.eastmoney.com",
+      "searchapi.eastmoney.com/api/suggest/get",
+      "push2delay.eastmoney.com/api/qt/clist/get"
+    ],
+    fast=True
+)
+
+# --------------------------
+# 后续你的业务代码保持不变
+# --------------------------
+
+# 假如你使用 akshare
+import akshare as ak
+df = ak.stock_zh_a_spot_em()
+
+# 假如你使用 efinance
+import efinance as ef
+ef.stock.get_realtime_quotes()
+```
+
+3. `yfinance` 用户：在 Python 文件顶部添加如下代码，并替换 `你的TOKEN`，不需要使用 AI 魔改。
+
+```
+# python 文件顶部添加如下代码
+# 重要：插件引入和调用一定要放到最顶部！不能在 akshare 或 efinance 之后引入！
+
+import akshare_proxy_patch
+
+akshare_proxy_patch.install_yfinance_patch(
+    "101.201.173.125",
+    auth_token="你的TOKEN",
+    retry=30,
+)
+
+# --------------------------
+# 后续你的业务代码保持不变
+# --------------------------
+
+import yfinance as yf
+
+data = yf.download("AAPL", start="2017-01-01", end="2017-04-30")
+```
+
+4. 尽量用 [efinance](https://github.com/Micro-sheep/efinance) 替代 akshare 的部分函数配合插件获取数据，效率更高，更省积分。
+
+## install_patch 参数说明
+
+- `ip`
+  - 网关，不可修改
+- `auth_token`
+  - TOKEN 授权凭证
+- `retry`
+  - 重试次数，默认为30，建议保持不变
+- `hook_domains`
+  - 封控的域名列表，插件会 hook 住这些域名的请求，解除封控。已默认涵盖常见函数的域名。
+  - 可点击 `akshare` 或 `efinance` 函数查看接口源码对应的 `URL` 并添加到该列表。
+  - 如只用到 `stock_zh_a_spot_em` 这个接口，`hook_domains` 可设置为 `["https://82.push2.eastmoney.com/api/qt/clist/get"]`。
+- `fast`
+  - 是否启用 `akshare` 多线程加速，默认开启，加速函数列表如下
+  - 所有用到 `fetch_paginated_data` 分页函数的接口，如 `stock_zh_a_spot_em`、`stock_sh_a_spot_em`、`stock_board_industry_cons_em` 等
+
+## 如何在 aktools 内集成插件？
+
+- `aktools` 想要集成插件，需下载 [akt.py](https://github.com/HelloYie/akshare-proxy-patch/blob/master/examples/aktools/akt.py) 文件，并填入您的 `TOKEN`。
+- 然后执行 `python akt.py` 即可启动一个 `http://127.0.0.1:8080/` 服务。只是启动方式不同而已，使用请参考 [aktools 官方文档](https://github.com/akfamily/aktools)。
+
+## 如何在 daily_stock_analysis 内集成插件？
+
+- [daily_stock_analysis](https://github.com/ZhuLinsen/daily_stock_analysis) 本身集成了 `akshare`、`efinance` 等数据源，如遇数据源报错，可以通过引入插件来解决。
+- 在 [daily_stock_analysis](https://github.com/ZhuLinsen/daily_stock_analysis) 项目中，找到入口 Python 文件 `main.py`，在文件顶部引入插件：
+
+```
+# 找到 daily_stock_analysis main.py 的这行（第一行）代码，在其下方引入插件
+from __future__ import annotations
+
+# 引入插件
+import akshare_proxy_patch
+
+akshare_proxy_patch.install_patch(
+    "101.201.173.125",
+    auth_token="你的TOKEN",
+    retry=30,
+    # 封控的域名列表，可自行调整
+    hook_domains=[
+      "fund.eastmoney.com",
+      "push2.eastmoney.com",
+      "push2his.eastmoney.com",
+      "emweb.securities.eastmoney.com",
+      "searchapi.eastmoney.com/api/suggest/get",
+      "push2delay.eastmoney.com/api/qt/clist/get"
+    ],
+    fast=True
+)
+# daily_stock_analysis main.py 的其他后续代码
+# 插件引入成功后正常启动该项目即可
+
+```
+
+## 股票数量很多，如何快速拉取数据？
+
+- 尽量使用 `efinance` 替代 `akshare` 来获取数据， `efinance` 内置多线程，效率更高，更省积分。
+- 在大规模快速获取数据时，可使用多进程提升效率。 可以参考 [多进程获取所有股票的历史K线](https://github.com/HelloYie/akshare-proxy-patch/blob/master/examples/all_quote_history/main.py)，修改 `auth_token`，执行 `python main.py` 测试。
+
+## 如何灵活禁用/启用插件？
+
+加入如下代码后插件就会被禁用，[点击查看禁用/启用插件示例](https://github.com/HelloYie/akshare-proxy-patch/blob/master/examples/disable/disable.py)：
+
+```
+akshare_proxy_patch.uninstall_patch()
+```
+
+## 我没使用 akshare 或 efinance，能集成插件吗？
+
+- 如果使用 Python 语言的 `requests` 库请求接口，插件能自动 hook 住请求，正常工作。
+- 如果您使用其他语言或 python 的其他库，可 [手动提取代理](http://101.201.173.125:47001/api/akshare-auth?token=XXX&version=0.5.1) 自行实现封控解除。
+
+## 使用问题交流群
+
+如使用时遇到问题，或对插件有什么意见或建议，可进群交流：
+
+![eozuhg3eH48ntnAnZXjKpmiegT7Ws0cl.webp](https://cdn.nodeimage.com/i/eozuhg3eH48ntnAnZXjKpmiegT7Ws0cl.webp)

{akshare_proxy_patch-0.5.0 → akshare_proxy_patch-0.5.1}/akshare_proxy_patch/__init__.py

@@ -4,21 +4,6 @@ import time
 import random
 import threading
 
-from .akshare_src.func import fetch_paginated_data as _patched_fetch_paginated_data
-from .akshare_src.stock.stock_fund_em import (
-    stock_individual_fund_flow_rank as _patched_stock_individual_fund_flow_rank,
-    stock_sector_fund_flow_rank as _patched_stock_sector_fund_flow_rank,
-)
-from .akshare_src.fund.fund_em import (
-    fund_money_fund_info_em as _patched_fund_money_fund_info_em,
-    fund_graded_fund_info_em as _patched_fund_graded_fund_info_em,
-    fund_etf_fund_info_em as _patched_fund_etf_fund_info_em,
-)
-from .akshare_src.fund.fund_fhsp_em import (
-    fund_fh_em as _patched_fund_fh_em,
-    fund_cf_em as _patched_fund_cf_em,
-    fund_fh_rank_em as _patched_fund_fh_rank_em,
-)
 
 # 1. 导入必要的库
 try:
@@ -30,7 +15,7 @@ except ImportError:
 
 import requests as std_requests
 
-__version__ = "0.5.0"
+__version__ = "0.5.1"
 
 # 授权接口固定使用标准 requests，避免干扰
 _auth_session = std_requests.Session()
@@ -190,6 +175,11 @@ def install_fast_patch():
     """
     try:
         akshare = importlib.import_module("akshare")
+        try:
+            from .akshare_src.func_new import fetch_paginated_data as _patched_fetch_paginated_data
+        except:
+            from .akshare_src.func import fetch_paginated_data as _patched_fetch_paginated_data
+
         # 覆盖已加载的 akshare 模块的 fetch_paginated_data
         for name, mod in list(sys.modules.items()):
             if mod and name.startswith("akshare"):
@@ -198,74 +188,6 @@ def install_fast_patch():
                         mod, "fetch_paginated_data"
                     )
                     setattr(mod, "fetch_paginated_data", _patched_fetch_paginated_data)
-
-        origin_backup[("akshare", "stock_individual_fund_flow_rank")] = getattr(
-            akshare, "stock_individual_fund_flow_rank"
-        )
-        setattr(
-            akshare,
-            "stock_individual_fund_flow_rank",
-            _patched_stock_individual_fund_flow_rank,
-        )
-
-        origin_backup[("akshare", "stock_sector_fund_flow_rank")] = getattr(
-            akshare, "stock_sector_fund_flow_rank"
-        )
-        setattr(
-            akshare,
-            "stock_sector_fund_flow_rank",
-            _patched_stock_sector_fund_flow_rank,
-        )
-
-        origin_backup[("akshare", "fund_money_fund_info_em")] = getattr(
-            akshare, "fund_money_fund_info_em"
-        )
-        setattr(
-            akshare,
-            "fund_money_fund_info_em",
-            _patched_fund_money_fund_info_em,
-        )
-
-        origin_backup[("akshare", "fund_graded_fund_info_em")] = getattr(
-            akshare, "fund_graded_fund_info_em"
-        )
-        setattr(
-            akshare,
-            "fund_graded_fund_info_em",
-            _patched_fund_graded_fund_info_em,
-        )
-
-        origin_backup[("akshare", "fund_etf_fund_info_em")] = getattr(
-            akshare, "fund_etf_fund_info_em"
-        )
-        setattr(
-            akshare,
-            "fund_etf_fund_info_em",
-            _patched_fund_etf_fund_info_em,
-        )
-
-        origin_backup[("akshare", "fund_fh_em")] = getattr(akshare, "fund_fh_em")
-        setattr(
-            akshare,
-            "fund_fh_em",
-            _patched_fund_fh_em,
-        )
-
-        origin_backup[("akshare", "fund_cf_em")] = getattr(akshare, "fund_cf_em")
-        setattr(
-            akshare,
-            "fund_cf_em",
-            _patched_fund_cf_em,
-        )
-
-        origin_backup[("akshare", "fund_fh_rank_em")] = getattr(
-            akshare, "fund_fh_rank_em"
-        )
-        setattr(
-            akshare,
-            "fund_fh_rank_em",
-            _patched_fund_fh_rank_em,
-        )
     except Exception as e:
         print("启动快速模式失败：", e)
 

akshare_proxy_patch-0.5.1/akshare_proxy_patch/akshare_src/func_new.py

@@ -0,0 +1,97 @@
+# !/usr/bin/env python
+"""
+Date: 2025/3/10 18:00
+Desc: 通用帮助函数
+"""
+
+import math
+import random
+import time
+from concurrent.futures import ThreadPoolExecutor, as_completed
+from typing import List, Dict
+
+import pandas as pd
+
+from akshare.utils import http
+from akshare.utils.tqdm import get_tqdm
+
+
+def _get_page(url: str, params: Dict):
+    """
+    取一页东方财富分页数据；每次新建连接，失败时按 http.request_with_retry 的策略重试
+    :return: 响应对象
+    :rtype: requests.Response
+    """
+    r = http.request_with_retry("get", url, params=params, fresh_session=True)
+    r.raise_for_status()
+    return r
+
+
+def fetch_paginated_data(url: str, base_params: Dict):
+    """
+    东方财富-分页获取数据并合并结果
+    https://quote.eastmoney.com/f1.html?newcode=0.000001
+    :param url: 股票代码
+    :type url: str
+    :param base_params: 基础请求参数
+    :type base_params: dict
+    :return: 合并后的数据
+    :rtype: pandas.DataFrame
+    """
+    # 复制参数以避免修改原始参数
+    params = base_params.copy()
+    # 获取第一页数据，用于确定分页信息
+    r = _get_page(url, params)
+    data_json = r.json()
+    # 计算分页信息
+    per_page_num = len(data_json["data"]["diff"])
+    total_page = math.ceil(data_json["data"]["total"] / per_page_num)
+    # 存储所有页面数据
+    temp_list = []
+    # 添加第一页数据
+    temp_list.append(pd.DataFrame(data_json["data"]["diff"]))
+    # 获取进度条
+    tqdm = get_tqdm()
+    # 并发获取剩余页面；每个任务使用独立参数，避免线程间共享可变状态。
+    def fetch_page(page: int):
+        page_params = base_params.copy()
+        page_params.update({"pn": page})
+        # 保留原有随机延迟策略，降低请求过于频繁的概率。
+        time.sleep(random.uniform(0.5, 1.5))
+        r = _get_page(url, page_params)
+        return pd.DataFrame(r.json()["data"]["diff"])
+
+    pages = range(2, total_page + 1)
+    if pages:
+        # 固定并发上限，避免页数较多时创建过多线程。
+        with ThreadPoolExecutor(max_workers=min(8, len(pages))) as executor:
+            futures = {executor.submit(fetch_page, page): page for page in pages}
+            page_results = {}
+            for future in tqdm(as_completed(futures), total=len(futures), leave=False):
+                page_results[futures[future]] = future.result()
+        # 按页码恢复原顺序，保持合并及相同排序键下的数据顺序不受并发完成顺序影响。
+        temp_list.extend(page_results[page] for page in pages)
+    # 合并所有数据
+    temp_df = pd.concat(temp_list, ignore_index=True)
+    temp_df["f3"] = pd.to_numeric(temp_df["f3"], errors="coerce")
+    temp_df = temp_df.sort_values(by=["f3"], ascending=False, ignore_index=True)
+    temp_df = temp_df.reset_index()
+    temp_df["index"] = temp_df["index"].astype(int) + 1
+    return temp_df
+
+
+def set_df_columns(df: pd.DataFrame, cols: List[str]) -> pd.DataFrame:
+    """
+    设置 pandas.DataFrame 为空的情况
+    :param df: 需要设置命名的数据框
+    :type df: pandas.DataFrame
+    :param cols: 字段的列表
+    :type cols: list
+    :return: 重新设置后的数据
+    :rtype: pandas.DataFrame
+    """
+    if df.shape == (0, 0):
+        return pd.DataFrame(data=[], columns=cols)
+    else:
+        df.columns = cols
+        return df