ezKit 1.9.5__tar.gz → 1.9.7__tar.gz

{ezkit-1.9.5/ezKit.egg-info → ezkit-1.9.7}/PKG-INFO

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

{ezkit-1.9.5 → ezkit-1.9.7}/ezKit/stock.py

@@ -99,7 +99,7 @@ def kdj_vector(
         case True if not utils.isTrue(df, DataFrame):
             logger.error("argument error: df")
             return None
-        case True if True not in [utils.isTrue(kdj_options, tuple), all(utils.isTrue(item, int) for item in kdj_options)]:
+        case True if not any([utils.isTrue(kdj_options, tuple), all(utils.isTrue(item, int) for item in kdj_options)]):
             logger.error("argument error: kdj_options")
             return None
         case _:
@@ -133,10 +133,10 @@ def data_vector(
         case True if not utils.isTrue(df, DataFrame):
             logger.error("argument error: df")
             return None
-        case True if True not in [utils.isTrue(macd_options, tuple), all(utils.isTrue(item, int) for item in macd_options)]:
+        case True if not any([utils.isTrue(macd_options, tuple), all(utils.isTrue(item, int) for item in macd_options)]):
             logger.error("argument error: macd_options")
             return None
-        case True if True not in [utils.isTrue(kdj_options, tuple), all(utils.isTrue(item, int) for item in kdj_options)]:
+        case True if not any([utils.isTrue(kdj_options, tuple), all(utils.isTrue(item, int) for item in kdj_options)]):
             logger.error("argument error: kdj_options")
             return None
         case _:
@@ -249,8 +249,8 @@ def get_stock_data_from_akshare(
     end_date: str = "20500101",
     timeout: float = 10
 ) -> DataFrame | None:
-    """从 akshare 获取数据"""
-    info = f"获取股票所有数据: {code}"
+    """从 akshare 获取股票数据"""
+    info = f"获取股票数据: {code}"
     try:
         logger.info(f"{info} ......")
         # https://akshare.akfamily.xyz/data/stock/stock.html#id22
@@ -276,17 +276,44 @@ def get_stock_data_from_akshare(
 # --------------------------------------------------------------------------------------------------
 
 
-def save_data_to_database(engine: Engine, code: str) -> bool:
-    """保存股票数据到数据库"""
-    info: str = "保存股票数据到数据库"
+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} ......")
+
         df: DataFrame | None = get_stock_data_from_akshare(code)
+
         if df is None:
+            logger.error(f"{info} [获取数据失败]")
             return False
-        df.to_sql(name=code, con=engine, if_exists="replace", index=False)
+
+        name = coderename(code)
+
+        if not isinstance(name, str):
+            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)

{ezkit-1.9.5 → ezkit-1.9.7}/ezKit/utils.py

@@ -55,6 +55,10 @@ def isTrue(
     #
     #   isTrue("abc", (str, int))
     #   isTrue("abc", (str | int))
+    #
+    # all() 用于检查一个可迭代对象(如列表、元组、集合等)中的 所有 元素是否为 真值 (truthy), 所有元素为真, 返回 True
+    # any() 用于检查一个可迭代对象(如列表、元组、集合等)中的 某个 元素是否为 真值 (truthy), 某个元素为真, 返回 True
+    # 与 all() 作用相反的 not any(), 可以用来检查所有元素是否为 假值 (falsy), any() 中所有元素为假, not any() 返回 True
 
     try:
 
@@ -165,7 +169,7 @@ def mam_of_numbers(
         case True if not isTrue(numbers, (list, tuple)):
             logger.error("argument error: numbers")
             return None
-        case True if True not in [isTrue(dest_type, str), dest_type is None]:
+        case True if not any([isTrue(dest_type, str), dest_type is None]):
             logger.error("argument error: dest_type")
             return None
         case _:

{ezkit-1.9.5 → ezkit-1.9.7/ezKit.egg-info}/PKG-INFO

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

{ezkit-1.9.5 → ezkit-1.9.7}/setup.py

@@ -3,7 +3,7 @@ from setuptools import find_packages, setup
 
 setup(
     name='ezKit',
-    version='1.9.5',
+    version='1.9.7',
     author='septvean',
     author_email='septvean@gmail.com',
     description='Easy Kit',