ezKit 1.9.2__py3-none-any.whl → 1.9.3__py3-none-any.whl

ezKit/stock.py

@@ -2,8 +2,12 @@
 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
 
@@ -111,3 +115,147 @@ def kdj_vector(df: DataFrame, cp: int = 9, sp1: int = 3, sp2: int = 3) -> DataFr
     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:
+    """数据运算"""
+
+    try:
+
+        # 数据为空
+        if isinstance(df, DataFrame) and df.empty:
+            return None
+
+        # ------------------------------------------------------------------------------------------
+
+        # 计算均线: 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(df['close'].values, fastperiod=macd_options[0], slowperiod=macd_options[1], signalperiod=macd_options[2])  # type: ignore
+        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 金叉死叉: 0 无, 1 金叉, 2 死叉
+        df['MACD_X'] = 0
+        macd_position = df['MACD_DIF'] > df['MACD_DEA']
+        df.loc[macd_position[(macd_position is True) & (macd_position.shift() is False)].index, 'MACD_X'] = 1  # type: ignore
+        df.loc[macd_position[(macd_position is False) & (macd_position.shift() is True)].index, 'MACD_X'] = 2  # type: ignore
+
+        # ------------------------------------------------------------------------------------------
+
+        # 计算 KDJ: : 默认参数 9 3 3
+        kdj_data = kdj_vector(df, kdj_options[0], kdj_options[1], kdj_options[2])
+
+        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 金叉死叉: 0 无, 1 金叉, 2 死叉
+            df['KDJ_X'] = 0
+            kdj_position = df['J'] > df['D']
+            df.loc[kdj_position[(kdj_position is True) & (kdj_position.shift() is False)].index, 'KDJ_X'] = 1  # type: ignore
+            df.loc[kdj_position[(kdj_position is False) & (kdj_position.shift() is True)].index, 'KDJ_X'] = 2  # type: ignore
+
+        # ------------------------------------------------------------------------------------------
+
+        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) -> DataFrame | None:
+    """从 akshare 获取数据"""
+    info = f"获取股票所有数据: {code}"
+    try:
+        logger.info(f"{info} ......")
+        df: DataFrame = ak.stock_zh_a_daily(symbol=code, adjust="qfq")
+        df = df.round({'turnover': 4})
+        logger.success(f"{info} [成功]")
+        return df[['date', 'open', 'close', 'high', 'low', 'volume', 'turnover']].copy()
+    except Exception as e:
+        logger.error(f"{info} [失败]")
+        logger.exception(e)
+        return None
+
+
+# --------------------------------------------------------------------------------------------------
+
+
+def save_data_to_database(engine: Engine, code: str) -> bool:
+    """保存股票数据到数据库"""
+    info: str = "保存股票数据到数据库"
+    try:
+        logger.info(f"{info} ......")
+        df: DataFrame | None = get_stock_data_from_akshare(code)
+        if df is None:
+            return False
+        df.to_sql(name=code, 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

{ezKit-1.9.2.dist-info → ezKit-1.9.3.dist-info}/METADATA

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

{ezKit-1.9.2.dist-info → ezKit-1.9.3.dist-info}/RECORD

@@ -9,12 +9,12 @@ ezKit/mongo.py,sha256=5KDDYQnOE3lw4QJlLp-EAuzbKM6VzbU0w0GR9iklldA,2379
 ezKit/qywx.py,sha256=X_H4fzP-iEqeDEbumr7D1bXi6dxczaxfO8iyutzy02s,7171
 ezKit/redis.py,sha256=5TQ7s8fkc-ArXuHT5gOlkR0EBgm3Z-9yTTtS1BgI5Qk,1951
 ezKit/sendemail.py,sha256=3YyBqFHPDZl9CwgWtmGABubARE_ih8P473mIcEp5VsY,8193
-ezKit/stock.py,sha256=_gWighuVzqSub4GRnN-X_9TVY7kUEYvKHKdbYJPYSYE,4001
+ezKit/stock.py,sha256=BWs3G2N3xRfva9yf6K40brnH31V2TROeo9pnl8kgEK8,9954
 ezKit/token.py,sha256=HKREyZj_T2S8-aFoFIrBXTaCKExQq4zE66OHXhGHqQg,1750
 ezKit/utils.py,sha256=q6DbXJNgC6sqzjqUPmxbiSiiXqiQFCVnI3jp_zi1ujs,42927
 ezKit/xftp.py,sha256=XyIdr_2rxRVLqPofG6fIYWhAMVsFwTyp46dg5P9FLW4,7774
-ezKit-1.9.2.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-ezKit-1.9.2.dist-info/METADATA,sha256=ZoiSVL8T2mtsb9ZxR61cHS8dGvixGvGS5lUui0Y5e8M,190
-ezKit-1.9.2.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-ezKit-1.9.2.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
-ezKit-1.9.2.dist-info/RECORD,,
+ezKit-1.9.3.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+ezKit-1.9.3.dist-info/METADATA,sha256=i9lzRl59vhpJ4az8LovuKDj_NjDISMpuBQW9zk_gf6w,190
+ezKit-1.9.3.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+ezKit-1.9.3.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
+ezKit-1.9.3.dist-info/RECORD,,