ezKit 1.9.12__py3-none-any.whl → 1.10.1__py3-none-any.whl

ezKit/utils.py

@@ -9,12 +9,11 @@ import time
 import tomllib
 from copy import deepcopy
 from itertools import islice
-from multiprocessing import Pool, Process
+from multiprocessing import Pool
 from multiprocessing.pool import ThreadPool
 from pathlib import Path
 from shutil import rmtree
-from threading import Thread
-from typing import Any, Callable, List, Optional, Union
+from typing import Any, Callable, List, Union
 from urllib.parse import ParseResult, urlparse
 from uuid import uuid4
 
@@ -742,31 +741,23 @@ def parent_dir(
 # --------------------------------------------------------------------------------------------------
 
 
-def retry(
-    times: int,
-    func: Callable,
-    **kwargs
-):
+def retry(func: Callable, times: int = 3, **kwargs):
     """重试"""
+
     # 函数传递参数: https://stackoverflow.com/a/803632
     # callable() 判断类型是非为函数: https://stackoverflow.com/a/624939
-    try:
-        _num = 0
-        while True:
-            # 重试次数判断 (0 表示无限次数, 这里条件使用 > 0, 表示有限次数)
-            if times > 0:
-                _num += 1
-                if _num > times:
-                    return
-            # 执行函数
-            try:
-                return func(**kwargs)
-            except Exception as e:
-                logger.exception(e)
+
+    for attempt in range(times):
+        try:
+            # 执行函数并结果
+            return func(**kwargs)
+        except Exception as e:
+            logger.exception(e)
+            if attempt < (times - 1):
                 logger.info('retrying ...')
-                continue
-    except Exception as e:
-        logger.exception(e)
+            else:
+                logger.error("all retries failed")
+                return False
 
 
 # --------------------------------------------------------------------------------------------------
@@ -1199,65 +1190,61 @@ def delete_directory(
 # --------------------------------------------------------------------------------------------------
 
 
-def process_pool(
+def processor(
     process_func: Callable,
-    process_data: Any = None,
+    process_data: List[Any],
     process_num: int = 2,
     thread: bool = False,
     **kwargs
-) -> list | bool:
-    """
-    多线程(MultiThread) | 多进程(MultiProcess)
-    """
-    # ThreadPool 线程池
+) -> Union[List[Any], bool]:
+    """使用多线程或多进程对数据进行并行处理"""
+
+    # :param process_func: 处理函数
+    # :param process_data: 待处理数据列表
+    # :param process_num: 并行数量
+    # :param thread: 是否使用多线程
+    # :param kwargs: 其他可选参数传递给线程池或进程池
+    # :return: 处理后的结果列表或 False（异常情况）
+    #
+    # MultiThread   多线程
+    # MultiProcess  多进程
+    #
+    # ThreadPool    线程池
+    # Pool          进程池
+    #
     # ThreadPool 共享内存, Pool 不共享内存
     # ThreadPool 可以解决 Pool 在某些情况下产生的 Can't pickle local object 的错误
-    # https://stackoverflow.com/a/58897266
-    try:
+    #   https://stackoverflow.com/a/58897266
+    #
+    # 如果要启动一个新的进程或者线程, 将 process_num 设置为 1 即可
 
-        # 处理数据
-        if len(process_data) <= process_num:
-            process_num = len(process_data)
-            _data = process_data
-        else:
-            _data = list_split(process_data, process_num, equally=True)
+    try:
 
-        if _data is None:
+        # 检查参数
+        if not check_arguments([(process_data, list, "process_data")]):
             return False
 
-        # 执行函数
-        if isTrue(thread, bool):
-            # 多线程
-            logger.info("execute multi thread ......")
-            with ThreadPool(process_num, **kwargs) as p:
-                return p.map(process_func, _data)
-        else:
-            # 多进程
-            logger.info("execute multi process ......")
-            with Pool(process_num, **kwargs) as p:
-                return p.map(process_func, _data)
+        # 确保并行数不超过数据量
+        process_num = min(len(process_data), process_num)
+        _data_chunks = (
+            list_split(process_data, process_num, equally=True)
+            if process_num > 1
+            else [process_data]
+        )
 
-    except Exception as e:
-        logger.exception(e)
-        return False
+        if not _data_chunks:
+            logger.error("data chunks error")
+            return False
 
+        logger.info(
+            f"Starting {'multi-threading' if thread else 'multi-processing'} with {process_num} workers..."
+        )
+
+        # 执行多线程或多进程任务
+        pool_cls = ThreadPool if thread else Pool
+        with pool_cls(process_num, **kwargs) as pool:
+            return pool.map(process_func, _data_chunks)
 
-def new_process(
-    process_func: Callable,
-    process_data: Any = None,
-    thread: bool = False,
-    daemon: bool = True,
-    **kwargs
-) -> Thread | Process | bool:
-    """New Process"""
-    try:
-        if isTrue(thread, bool):
-            process = Thread(target=process_func, args=process_data, **kwargs)
-        else:
-            process = Process(target=process_func, args=process_data, **kwargs)
-        process.daemon = daemon
-        process.start()
-        return process
     except Exception as e:
         logger.exception(e)
         return False

{ezKit-1.9.12.dist-info → ezKit-1.10.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ezKit
-Version: 1.9.12
+Version: 1.10.1
 Summary: Easy Kit
 Author: septvean
 Author-email: septvean@gmail.com

{ezKit-1.9.12.dist-info → ezKit-1.10.1.dist-info}/RECORD

@@ -2,19 +2,17 @@ ezKit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ezKit/bottle.py,sha256=usKK1wVaZw4_D-4VwMYmOIc8jtz4TrpM30nck59HMFw,180178
 ezKit/bottle_extensions.py,sha256=3reEQVZuHklXTl6r7F8kiBFFPb0RaAGc3mYJJnrMDjQ,1129
 ezKit/cipher.py,sha256=0T_StbjiNI4zgrjVgcfU-ffKgu1waBA9UDudAnqFcNM,2896
-ezKit/cls.py,sha256=e7_72kv0Q_o023xcjKNtrkfKg7frABQvCF_JjoHV94U,10800
 ezKit/database.py,sha256=Rc4RgjHOOtf5dMLvMkK1beRfbIai5E1x4HTsDwKsA-Q,6822
 ezKit/http.py,sha256=i3Kn5AMAMicDMcDjxKKZU7zqEKTU88Ec9_LwCuBJy-0,1801
 ezKit/mongo.py,sha256=dOm_1wXEPp_e8Ml5Qq78M7FDNrQUAZaThzVIiiLJJwk,2393
 ezKit/qywx.py,sha256=X_H4fzP-iEqeDEbumr7D1bXi6dxczaxfO8iyutzy02s,7171
 ezKit/redis.py,sha256=g2_V4jvq0djRc20jLZkgeAeF_bYrq-Rbl_kHcCUPZcA,1965
 ezKit/sendemail.py,sha256=tRXCsJm_RfTJ9xEWe_lTQ5kOs2JxHGPXvq0oWA7prq0,7263
-ezKit/stock.py,sha256=4wphZahpiDs0MuPVCUcD22joOQldJhmXjogdroxyR00,12346
 ezKit/token.py,sha256=HKREyZj_T2S8-aFoFIrBXTaCKExQq4zE66OHXhGHqQg,1750
-ezKit/utils.py,sha256=TDsL3PRkQy6NdZgphkgwacbWvHqEmq4LOkfNzmxV4DY,42682
+ezKit/utils.py,sha256=k3hSnOwNSyyRDVwfEzQUXQh_oJJ51KOT-PvLFPYOtOE,42517
 ezKit/xftp.py,sha256=XyIdr_2rxRVLqPofG6fIYWhAMVsFwTyp46dg5P9FLW4,7774
-ezKit-1.9.12.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-ezKit-1.9.12.dist-info/METADATA,sha256=cCgqBuFpmH0zpv8j267E9tTkm6q1WzdfOhUCwS3MTjI,191
-ezKit-1.9.12.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-ezKit-1.9.12.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
-ezKit-1.9.12.dist-info/RECORD,,
+ezKit-1.10.1.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+ezKit-1.10.1.dist-info/METADATA,sha256=7-BrvzaxysoECMgvFVFeBbVHX2RgQGbdYUrZS3xumUo,191
+ezKit-1.10.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+ezKit-1.10.1.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
+ezKit-1.10.1.dist-info/RECORD,,

ezKit/cls.py

@@ -1,313 +0,0 @@
-"""财联社数据"""
-import re
-
-import pandas as pd
-import requests
-from loguru import logger
-
-from . import stock, utils
-
-
-def up_down_analysis(
-    target: str = "up_pool",
-    df: bool = False
-) -> list | pd.DataFrame | None:
-    """涨停跌停数据"""
-
-    # 判断参数是否正确
-    match True:
-        case True if not utils.isTrue(target, str):
-            logger.error("argument error: target")
-            return None
-        case _:
-            pass
-
-    info: str = "获取涨停池股票"
-    match True:
-        case True if target == "up_pool":
-            info = "获取涨停池股票"
-        case True if target == "continuous_up_pool":
-            info = "获取连板池股票"
-        case True if target == "up_open_pool":
-            info = "获取炸板池股票"
-        case True if target == "down_pool":
-            info = "获取跌停池股票"
-        case _:
-            pass
-
-    try:
-        logger.info(f"{info} ......")
-
-        user_agent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36"
-        headers = {"User-Agent": user_agent}
-
-        # 涨停池: https://x-quote.cls.cn/quote/index/up_down_analysis?rever=1&way=last_px&type=up_pool
-        # 连板池: https://x-quote.cls.cn/quote/index/up_down_analysis?rever=1&way=last_px&type=continuous_up_pool
-        # 炸板池: https://x-quote.cls.cn/quote/index/up_down_analysis?rever=1&way=last_px&type=up_open_pool
-        # 跌停池: https://x-quote.cls.cn/quote/index/up_down_analysis?rever=1&way=last_px&type=down_pool
-        api = f"https://x-quote.cls.cn/quote/index/up_down_analysis?rever=1&way=last_px&type={target}"
-
-        response = requests.get(api, headers=headers, timeout=10)
-
-        response_dict: dict = response.json()
-
-        result: list = []
-
-        for i in response_dict["data"]:
-
-            # if re.match(r"^(sz00|sh60)", i["secu_code"]):
-            #     print(i["secu_code"])
-
-            # if re.search(r"ST|银行", i["secu_name"]):
-            #     print(i["secu_name"])
-
-            # 主板, 非ST, 非银行, 非证券
-            if (not re.match(r"^(sz00|sh60)", i["secu_code"])) or re.search(r"ST|银行|证券", i["secu_name"]):
-                continue
-
-            if target in ["up_pool", "up_pool"]:
-                result.append({
-                    "code": stock.coderename(i["secu_code"], restore=True),
-                    "name": i["secu_name"],
-                    "up_days": i["limit_up_days"],
-                    "reason": i["up_reason"]
-                })
-
-            if target in ["up_open_pool", "down_pool"]:
-                result.append({
-                    "code": stock.coderename(i["secu_code"], restore=True),
-                    "name": i["secu_name"]
-                })
-
-        if not utils.isTrue(df, bool):
-            logger.success(f"{info} [成功]")
-            return result
-
-        # data: pd.DataFrame = pd.DataFrame(response_dict["data"], columns=["secu_code", "secu_name", "limit_up_days", "up_reason"])
-        # data = data.rename(columns={"secu_code": "code", "secu_name": "name", "limit_up_days": "up_days", "up_reason": "reason"})
-
-        return pd.DataFrame(data=pd.DataFrame(result))
-
-    except Exception as e:
-        logger.error(f"{info} [失败]")
-        logger.exception(e)
-        return None
-
-
-# --------------------------------------------------------------------------------------------------
-
-
-def latest_data(
-    payload: str | dict,
-    data_type: str = "stock",
-    df: bool = False
-) -> list | pd.DataFrame | None:
-    """股票或板块的最新数据"""
-
-    # 热门板块
-    #   https://www.cls.cn/hotPlate
-    # 行业板块
-    #   https://x-quote.cls.cn/web_quote/plate/plate_list?rever=1&way=change&type=industry
-    # 概念板块
-    #   https://x-quote.cls.cn/web_quote/plate/plate_list?rever=1&way=change&type=concept
-    # 地域板块
-    #   https://x-quote.cls.cn/web_quote/plate/plate_list?rever=1&way=change&type=area
-
-    # ----------------------------------------------------------------------------------------------
-
-    # 判断参数类型
-    match True:
-        case True if not utils.isTrue(payload, (str, dict)):
-            logger.error("argument error: payload")
-            return None
-        case True if not utils.isTrue(data_type, str):
-            logger.error("argument error: data_type")
-            return None
-        case _:
-            pass
-
-    # ----------------------------------------------------------------------------------------------
-
-    # 判断数据类型. 数据类型: 个股, 板块 (产业链: industry)
-    if data_type not in ["stock", "plate"]:
-        logger.error("data_type error")
-        return None
-
-    # ----------------------------------------------------------------------------------------------
-
-    # 日志信息
-
-    # 个股 (默认)
-    info: str = "获取股票最新数据"
-
-    # 板块
-    if data_type == "plate":
-        info = "获取板块最新数据"
-
-    # match True:
-    #     case True if data_type == "plate":
-    #         info = "获取板块最新数据"
-    #     case True if data_type == "industry":
-    #         info = "获取产业链最新数据"
-    #     case _:
-    #         pass
-
-    # ----------------------------------------------------------------------------------------------
-
-    try:
-
-        logger.info(f"{info} ......")
-
-        # ------------------------------------------------------------------------------------------
-
-        # HTTP User Agent
-        user_agent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36"
-
-        # HTTP Headers
-        headers = {"User-Agent": user_agent}
-
-        # ------------------------------------------------------------------------------------------
-
-        # 请求参数
-        params: dict = {}
-
-        # 默认请求参数
-        if isinstance(payload, str) and utils.isTrue(payload, str):
-            params = {"secu_code": payload}
-
-        # 请求参数
-        if isinstance(payload, dict) and utils.isTrue(payload, dict):
-            params = payload
-
-        # ------------------------------------------------------------------------------------------
-
-        # 不直接在API后面跟参数, 使用 params 传递参数
-
-        # API: 股票
-        # api: str = f"https://x-quote.cls.cn/quote/stock/basic?secu_code={code}"
-        api: str = "https://x-quote.cls.cn/quote/stock/basic"
-
-        # API: 板块
-        if data_type == "plate":
-            # api = f"https://x-quote.cls.cn/web_quote/plate/stocks?secu_code={code}"
-            api = "https://x-quote.cls.cn/web_quote/plate/stocks"
-
-        # match True:
-        #     case True if data_type == "plate":
-        #         # 板块
-        #         # api = f"https://x-quote.cls.cn/web_quote/plate/stocks?secu_code={code}"
-        #         api = "https://x-quote.cls.cn/web_quote/plate/stocks"
-        #     case True if data_type == "industry":
-        #         # 产业链
-        #         # api = f"https://x-quote.cls.cn/web_quote/plate/industry?secu_code={code}"
-        #         api = "https://x-quote.cls.cn/web_quote/plate/industry"
-        #     case _:
-        #         pass
-
-        # ------------------------------------------------------------------------------------------
-
-        # 获取数据
-        # response = requests.get(api, headers=headers, timeout=10)
-        response = requests.get(api, headers=headers, params=params, timeout=10)
-
-        # 转换数据类型
-        response_dict: dict = response.json()
-
-        # 判断数据是否正确
-        if True not in [utils.isTrue(response_dict["data"], dict), utils.isTrue(response_dict["data"], list)]:
-            logger.error(f"{info} [失败]")
-            return None
-
-        # ------------------------------------------------------------------------------------------
-
-        # 个股
-
-        if data_type == "stock":
-
-            # 停牌, 返回 None
-            if response_dict["data"]["trade_status"] == "STOPT":
-                logger.error(f"{info} [停牌]")
-                return None
-
-            # pd.DataFrame 数据
-            if utils.isTrue(df, bool):
-                df_data = {
-                    # "date": [pd.to_datetime(date_today)],
-                    "open": [float(response_dict["data"]["open_px"])],
-                    "close": [float(response_dict["data"]["last_px"])],
-                    "high": [float(response_dict["data"]["high_px"])],
-                    "low": [float(response_dict["data"]["low_px"])],
-                    "volume": [int(response_dict["data"]["business_amount"])],
-                    "turnover": [float(response_dict["data"]["tr"])]
-                }
-                logger.success(f"{info} [成功]")
-                return pd.DataFrame(data=df_data)
-
-            # 默认返回的数据
-            logger.success(f"{info} [成功]")
-            return response_dict["data"]
-
-        # ------------------------------------------------------------------------------------------
-
-        # 板块
-
-        # 板块数据不能转换为 pd.DataFrame
-        if (data_type == "plate") and utils.isTrue(df, bool):
-            logger.error(f"{info} [错误]")
-            return None
-
-        # 数据结果
-        result: list = []
-
-        # 筛选 主板, 非ST, 非银行, 非证券 的股票
-        for i in response_dict["data"]["stocks"]:
-            if (re.match(r"^(sz00|sh60)", i["secu_code"])) and (not re.search(r"ST|银行|证券", i["secu_name"])):
-                result.append(i)
-
-        # 返回数据
-        logger.success(f"{info} [成功]")
-        return result
-
-    except Exception as e:
-        logger.error(f"{info} [失败]")
-        logger.exception(e)
-        return None
-
-
-# --------------------------------------------------------------------------------------------------
-
-
-def plate_codes(
-    plate: str
-) -> list | None:
-    """获取板块成分股代码"""
-
-    # 判断参数是否正确
-    match True:
-        case True if not utils.isTrue(plate, str):
-            logger.error("argument error: plate")
-            return None
-        case _:
-            pass
-
-    info: str = "获取板块成分股代码"
-
-    try:
-
-        logger.info(f"{info} ......")
-
-        items = latest_data(payload=plate, data_type="plate")
-
-        if isinstance(items, list):
-            codes: list = [stock.coderename(i["secu_code"], restore=True) for i in items]
-            codes.sort()
-            logger.success(f"{info} [成功]")
-            return codes
-
-        logger.error(f"{info} [失败]")
-        return None
-
-    except Exception as e:
-        logger.error(f"{info} [失败]")
-        logger.exception(e)
-        return None

ezKit/stock.py

@@ -1,355 +0,0 @@
-"""股票"""
-import re
-from copy import deepcopy
-
-import akshare as ak
-import numpy as np
-import talib as ta
-from loguru import logger
-from pandas import DataFrame
-from sqlalchemy.engine import Engine
-
-from . import utils
-
-
-def coderename(
-    target: str | dict,
-    restore: bool = False
-) -> str | dict | None:
-    """代码重命名"""
-
-    # 正向:
-    #     coderename('000001') => 'sz000001'
-    #     coderename({'code': '000001', 'name': '平安银行'}) => {'code': 'sz000001', 'name': '平安银行'}
-    # 反向:
-    #     coderename('sz000001', restore=True) => '000001'
-    #     coderename({'code': 'sz000001', 'name': '平安银行'}) => {'code': '000001', 'name': '平安银行'}
-
-    # 判断参数是否正确
-    match True:
-        case True if not utils.isTrue(target, (str, dict)):
-            logger.error("argument error: target")
-            return None
-        case _:
-            pass
-
-    try:
-
-        # 初始化
-        code_object: dict = {}
-        code_name: str | dict = ""
-
-        # 判断 target 是 string 还是 dictionary
-        if isinstance(target, str) and utils.isTrue(target, str):
-            code_name = target
-        elif isinstance(target, dict) and utils.isTrue(target, dict):
-            code_object = deepcopy(target)
-            code_name = str(deepcopy(target["code"]))
-        else:
-            return None
-
-        # 是否还原
-        if utils.isTrue(restore, bool):
-            if len(code_name) == 8 and re.match(r"^(sz|sh)", code_name):
-                code_name = deepcopy(code_name[2:8])
-            else:
-                return None
-        else:
-            if code_name[0:2] == "00":
-                code_name = f"sz{code_name}"
-            elif code_name[0:2] == "60":
-                code_name = f"sh{code_name}"
-            else:
-                return None
-
-        # 返回结果
-        if utils.isTrue(target, str):
-            return code_name
-
-        if utils.isTrue(target, dict):
-            code_object["code"] = code_name
-            return code_object
-
-        return None
-
-    except Exception as e:
-        logger.exception(e)
-        return None
-
-
-# --------------------------------------------------------------------------------------------------
-
-
-def kdj_vector(
-    df: DataFrame,
-    kdj_options: tuple[int, int, int] = (9, 3, 3)
-) -> DataFrame | None:
-    """KDJ计算器"""
-
-    # 计算周期：Calculation Period, 也可使用 Lookback Period 表示回溯周期, 指用于计算指标值的时间周期.
-    # 移动平均周期: Smoothing Period 或 Moving Average Period, 指对指标进行平滑处理时采用的周期.
-    # 同花顺默认参数: 9 3 3
-    # https://www.daimajiaoliu.com/daima/4ed4ffa26100400
-    # 说明: KDJ 指标的中文名称又叫随机指标, 融合了动量观念、强弱指标和移动平均线的一些优点, 能够比较迅速、快捷、直观地研判行情, 被广泛用于股市的中短期趋势分析.
-    # 有采用 ewm 使用 com=2 的, 但是如果使用 com=2 在默认值的情况下KDJ值是正确的.
-    # 但是非默认值, 比如调整参数, 尝试慢速 KDJ 时就不对了, 最终采用 alpha = 1/m 的情况, 对比同花顺数据, 是正确的.
-
-    # 检查参数
-    if isinstance(df, DataFrame) and df.empty:
-        logger.error("argument error: df")
-        return None
-
-    if not utils.check_arguments([(kdj_options, tuple, "kdj_options")]):
-        return None
-
-    if not all(utils.isTrue(item, int) for item in kdj_options):
-        logger.error("argument error: kdj_options")
-        return None
-
-    try:
-        low_list = df['low'].rolling(kdj_options[0]).min()
-        high_list = df['high'].rolling(kdj_options[0]).max()
-        rsv = (df['close'] - low_list) / (high_list - low_list) * 100
-        df['K'] = rsv.ewm(alpha=1 / kdj_options[1], adjust=False).mean()
-        df['D'] = df['K'].ewm(alpha=1 / kdj_options[2], adjust=False).mean()
-        df['J'] = (3 * df['K']) - (2 * df['D'])
-        return df
-    except Exception as e:
-        logger.exception(e)
-        return None
-
-
-# --------------------------------------------------------------------------------------------------
-
-
-def data_vector(
-    df: DataFrame,
-    macd_options: tuple[int, int, int] = (12, 26, 9),
-    kdj_options: tuple[int, int, int] = (9, 3, 3)
-) -> DataFrame | None:
-    """数据运算"""
-
-    # 检查参数
-    if isinstance(df, DataFrame) and df.empty:
-        logger.error("argument error: df")
-        return None
-
-    if not utils.check_arguments([(macd_options, tuple, "macd_options"), (kdj_options, tuple, "kdj_options")]):
-        return None
-
-    if not all(utils.isTrue(item, int) for item in macd_options):
-        logger.error("argument error: macd_options")
-        return None
-
-    if not all(utils.isTrue(item, int) for item in kdj_options):
-        logger.error("argument error: kdj_options")
-        return None
-
-    try:
-
-        # ------------------------------------------------------------------------------------------
-
-        # 计算均线: 3,7日均线
-        # pylint: disable=E1101
-        # df['SMA03'] = ta.SMA(df['close'], timeperiod=3)  # type: ignore
-        # df['SMA07'] = ta.SMA(df['close'], timeperiod=7)  # type: ignore
-
-        # 3,7日均线金叉: 0 无, 1 金叉, 2 死叉
-        # df['SMA37_X'] = 0
-        # sma37_position = df['SMA03'] > df['SMA07']
-        # df.loc[sma37_position[(sma37_position is True) & (sma37_position.shift() is False)].index, 'SMA37_X'] = 1  # type: ignore
-        # df.loc[sma37_position[(sma37_position is False) & (sma37_position.shift() is True)].index, 'SMA37_X'] = 2  # type: ignore
-
-        # 计算均线: 20,25日均线
-        # df['SMA20'] = ta.SMA(df['close'], timeperiod=20)  # type: ignore
-        # df['SMA25'] = ta.SMA(df['close'], timeperiod=25)  # type: ignore
-
-        # 20,25日均线金叉: 0 无, 1 金叉, 2 死叉
-        # df['SMA225_X'] = 0
-        # sma225_position = df['SMA20'] > df['SMA25']
-        # df.loc[sma225_position[(sma225_position is True) & (sma225_position.shift() is False)].index, 'SMA225_X'] = 1  # type: ignore
-        # df.loc[sma225_position[(sma225_position is False) & (sma225_position.shift() is True)].index, 'SMA225_X'] = 2  # type: ignore
-
-        # ------------------------------------------------------------------------------------------
-
-        # 计算 MACD: 默认参数 12 26 9
-        macd_dif, macd_dea, macd_bar = ta.MACD(  # type: ignore
-            df['close'].values,
-            fastperiod=macd_options[0],
-            slowperiod=macd_options[1],
-            signalperiod=macd_options[2]
-        )
-
-        macd_dif[np.isnan(macd_dif)], macd_dea[np.isnan(macd_dea)], macd_bar[np.isnan(macd_bar)] = 0, 0, 0
-
-        # https://www.bilibili.com/read/cv10185856
-        df['MACD'] = 2 * (macd_dif - macd_dea)
-        df['MACD_DIF'] = macd_dif
-        df['MACD_DEA'] = macd_dea
-
-        # 初始化 MACD_X 列(0 无, 1 金叉, 2 死叉)
-        df['MACD_X'] = 0
-
-        # 计算 MACD 条件
-        macd_position = df['MACD_DIF'] > df['MACD_DEA']
-
-        # 设置 MACD_X = 1: 从 False 变为 True 的位置
-        df.loc[macd_position & ~macd_position.shift(fill_value=False), 'MACD_X'] = 1
-
-        # 设置 MACD_X = 2: 从 True 变为 False 的位置
-        df.loc[~macd_position & macd_position.shift(fill_value=False), 'MACD_X'] = 2
-
-        # 将浮点数限制为小数点后两位
-        df['MACD'] = df['MACD'].round(2)
-        df['MACD_DIF'] = df['MACD_DIF'].round(2)
-        df['MACD_DEA'] = df['MACD_DEA'].round(2)
-
-        # ------------------------------------------------------------------------------------------
-
-        # # 计算 KDJ: : 默认参数 9 3 3
-        kdj_data = kdj_vector(df, kdj_options)
-
-        if kdj_data is not None:
-
-            # KDJ 数据
-            df['K'] = kdj_data['K'].values
-            df['D'] = kdj_data['D'].values
-            df['J'] = kdj_data['J'].values
-
-            # 初始化 KDJ_X 列(0 无, 1 金叉, 2 死叉)
-            df['KDJ_X'] = 0
-
-            # 计算 MACD 条件
-            kdj_position = df['J'] > df['D']
-
-            # 设置 KDJ_X = 1: 从 False 变为 True 的位置
-            df.loc[kdj_position & ~kdj_position.shift(fill_value=False), 'KDJ_X'] = 1
-
-            # 设置 KDJ_X = 2: 从 True 变为 False 的位置
-            df.loc[~kdj_position & kdj_position.shift(fill_value=False), 'KDJ_X'] = 2
-
-            # 将浮点数限制为小数点后两位
-            df['K'] = df['K'].round(2)
-            df['D'] = df['D'].round(2)
-            df['J'] = df['J'].round(2)
-
-        # ------------------------------------------------------------------------------------------
-
-        return df
-
-    except Exception as e:
-        logger.exception(e)
-        return None
-
-
-# --------------------------------------------------------------------------------------------------
-
-
-def get_code_name_from_akshare() -> DataFrame | None:
-    """获取股票代码和名称"""
-    info = "获取股票代码和名称"
-    try:
-        logger.info(f"{info} ......")
-        df: DataFrame = ak.stock_info_a_code_name()
-        if df.empty:
-            logger.error(f"{info} [失败]")
-            return None
-        # 排除 ST、证券和银行
-        # https://towardsdatascience.com/8-ways-to-filter-pandas-dataframes-d34ba585c1b8
-        df = df[df.code.str.contains("^00|^60") & ~df.name.str.contains("ST|证券|银行")]
-        logger.success(f"{info} [成功]")
-        return df
-    except Exception as e:
-        logger.error(f"{info} [失败]")
-        logger.exception(e)
-        return None
-
-
-# --------------------------------------------------------------------------------------------------
-
-
-def get_stock_data_from_akshare(
-    code: str,
-    adjust: str = "qfq",
-    period: str = "daily",
-    start_date: str = "19700101",
-    end_date: str = "20500101",
-    timeout: float = 10
-) -> DataFrame | None:
-    """从 akshare 获取股票数据"""
-    info = f"获取股票数据: {code}"
-    try:
-        logger.info(f"{info} ......")
-        # https://akshare.akfamily.xyz/data/stock/stock.html#id22
-        df: DataFrame = ak.stock_zh_a_hist(symbol=code, adjust=adjust, period=period, start_date=start_date, end_date=end_date, timeout=timeout)
-        df = df.rename(columns={
-            "日期": "date",
-            "开盘": "open",
-            "收盘": "close",
-            "最高": "high",
-            "最低": "low",
-            "成交量": "volume"
-        })
-        logger.success(f"{info} [成功]")
-        return df[['date', 'open', 'close', 'high', 'low', 'volume']].copy()
-    except Exception as e:
-        logger.error(f"{info} [失败]")
-        logger.exception(e)
-        return None
-
-
-# --------------------------------------------------------------------------------------------------
-
-
-def save_data_to_database(engine: Engine, code: str, latest: bool = False) -> bool:
-    """保存股票所有数据到数据库"""
-
-    # 默认将所有数据保存到数据库中的表里
-    # 如果 latest 为 True, 插入最新的数据到数据库中的表里
-    #   即: 将最后一条数据插入到数据库中的表里
-
-    info: str = "保存股票所有数据到数据库"
-
-    if utils.isTrue(latest, bool):
-        info = "保存股票最新数据到数据库"
-
-    try:
-
-        logger.info(f"{info} ......")
-
-        # 代码名称转换
-        name = coderename(code)
-
-        if not isinstance(name, str):
-            logger.error(f"{info} [代码名称转换错误]")
-            return False
-
-        # 获取数据
-        df: DataFrame | None = get_stock_data_from_akshare(code)
-
-        if df is None:
-            logger.error(f"{info} [获取数据错误]")
-            return False
-
-        # 计算数据
-        df: DataFrame | None = data_vector(df)
-
-        if df is None:
-            logger.error(f"{info} [计算数据错误]")
-            return False
-
-        # 保存到数据库
-        if utils.isTrue(latest, bool):
-            df = df.tail(1)
-            df.to_sql(name=name, con=engine, if_exists="append", index=False)
-        else:
-            df.to_sql(name=name, con=engine, if_exists="replace", index=False)
-
-        logger.success(f"{info} [成功]")
-
-        return True
-
-    except Exception as e:
-        logger.success(f"{info} [失败]")
-        logger.exception(e)
-        return False