ezKit 2.0.4__tar.gz → 2.0.6__tar.gz

{ezkit-2.0.4/ezKit.egg-info → ezkit-2.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ezKit
-Version: 2.0.4
+Version: 2.0.6
 Summary: Easy Kit
 Author: septvean
 Author-email: septvean@gmail.com

ezkit-2.0.6/ezKit/stock.py

@@ -0,0 +1,85 @@
+"""股票"""
+
+from copy import deepcopy
+from os import environ
+
+from loguru import logger
+
+# --------------------------------------------------------------------------------------------------
+
+DEBUG = environ.get("DEBUG")
+
+# --------------------------------------------------------------------------------------------------
+
+
+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': '平安银行'}
+
+    try:
+
+        # 初始化
+
+        code_object: dict = {}
+        code_name: str = ""
+
+        # 判断 target
+
+        if isinstance(target, str) and target:
+            code_name = target
+        elif isinstance(target, dict) and target:
+            code_object = deepcopy(target)
+            code_name = code_object.get("code", "")
+        else:
+            logger.error("target value or type error")
+            return None
+
+        if not code_name:
+            logger.error("code name error")
+            return None
+
+        # 是否还原
+
+        if restore:
+
+            if len(code_name) != 8 or not code_name.startswith(("sz", "sh")):
+                logger.error("code name error")
+                return None
+
+            code_name = code_name[-6:]
+
+        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:
+                logger.error("code name error")
+                return None
+
+        # 返回结果
+
+        if isinstance(target, str):
+            return code_name
+
+        if isinstance(target, dict):
+            code_object["code"] = code_name
+            return code_object
+
+        return None
+
+    except Exception as e:
+
+        if DEBUG:
+            logger.exception(e)
+        else:
+            logger.error(e)
+
+        return None

ezkit-2.0.6/ezKit/utils.py

@@ -0,0 +1,145 @@
+"""Utils Library"""
+
+import re
+import tomllib
+
+from os import environ
+from typing import Iterable
+
+from loguru import logger
+
+# --------------------------------------------------------------------------------------------------
+
+DEBUG = environ.get("DEBUG")
+
+# --------------------------------------------------------------------------------------------------
+
+
+def load_toml_file(file: str) -> dict:
+    """Load TOML file"""
+
+    info: str = "load toml file"
+
+    try:
+
+        logger.info(f"{info} [ start ]")
+
+        # 不要加 encoding="utf-8" 参数, 否则会报错:
+        # binary mode doesn't take an encoding argument
+        with open(file, "rb") as _file:
+            config = tomllib.load(_file)
+
+        logger.success(f"{info} [ success ]")
+
+        return config
+
+    except Exception as e:
+
+        logger.error(f"{info} [ error ]")
+
+        if DEBUG:
+            logger.exception(e)
+        else:
+            logger.error(e)
+
+        return {}
+
+
+# --------------------------------------------------------------------------------------------------
+
+
+def check_dict_values_incorrect(
+    data: dict[str, object],
+    errors: object | Iterable[object] | None = None,
+    include_keys: list[str] | None = None,
+    exclude_keys: list[str] | None = None,
+    regex: str | None = None,
+) -> bool:
+    """检查字典的值是否存在错误值"""
+
+    # 支持:
+    #
+    #   target: 单值 或 多值（列表/元组）
+    #   regex: 值是否匹配某正则表达式
+    #   include_keys / exclude_keys: 指定检查的 key 范围
+    #
+    # 匹配规则(满足任一条件即可):
+    #
+    #   value 等于 target 或 在 target 列表里
+    #   value 与 regex 匹配
+    #
+    # 参数:
+    #
+    #   data            dict[str, object]       待检查的字典
+    #   errors          object|Iterable|None    指定匹配值或值列表
+    #   include_keys    list[str]|None          只检查这些 key
+    #   exclude_keys    list[str]|None          排除指定 key
+    #   regex           str|None                正则匹配（自动转换成 re.Pattern）
+    #
+    # 返回:
+    #
+    #   bool
+    #
+    # 所有最终 keys 的值满足任一匹配条件返回 False, 否则 True
+
+    try:
+
+        # -----------------------
+        # 处理 include / exclude 逻辑
+        # -----------------------
+        if include_keys is not None:
+            keys = set(include_keys)
+        else:
+            keys = set(data.keys())
+
+        if exclude_keys is not None:
+            keys -= set(exclude_keys)
+
+        # 若没有 key 需要检查 -> 默认 True
+        if not keys:
+            return False
+
+        # -----------------------
+        # 处理 target（单值 or 列表）
+        # -----------------------
+        if isinstance(errors, Iterable) and not isinstance(errors, (str, bytes)):
+            error_set = set(errors)
+        else:
+            error_set = {errors}
+
+        # -----------------------
+        # 处理正则
+        # -----------------------
+        pattern = re.compile(regex) if regex else None
+
+        # -----------------------
+        # 核心检查逻辑
+        # -----------------------
+        for k in keys:
+
+            value = data.get(k)
+
+            # 匹配 target
+            if value in error_set:
+                return False
+
+            # 匹配 类型
+            # if isinstance(types, list) and types:
+            #     if any(isinstance(value, type) for type in types):
+            #         return False
+
+            # 匹配正则
+            if pattern is not None:
+                if isinstance(value, str) and pattern.search(value):
+                    return False
+
+        return True
+
+    except Exception as e:
+
+        if DEBUG:
+            logger.exception(e)
+        else:
+            logger.error(e)
+
+        return False

{ezkit-2.0.4 → ezkit-2.0.6/ezKit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ezKit
-Version: 2.0.4
+Version: 2.0.6
 Summary: Easy Kit
 Author: septvean
 Author-email: septvean@gmail.com

{ezkit-2.0.4 → ezkit-2.0.6}/ezKit.egg-info/SOURCES.txt

@@ -7,6 +7,7 @@ ezKit/__init__.py
 ezKit/database.py
 ezKit/http.py
 ezKit/redix.py
+ezKit/stock.py
 ezKit/utils.py
 ezKit.egg-info/PKG-INFO
 ezKit.egg-info/SOURCES.txt

{ezkit-2.0.4 → ezkit-2.0.6}/setup.py

@@ -4,7 +4,7 @@ from setuptools import find_packages, setup
 
 setup(
     name="ezKit",
-    version="2.0.4",
+    version="2.0.6",
     author="septvean",
     author_email="septvean@gmail.com",
     description="Easy Kit",

ezkit-2.0.4/ezKit/utils.py

@@ -1,46 +0,0 @@
-"""Utils Library"""
-
-import tomllib
-
-from os import environ
-
-from loguru import logger
-
-# --------------------------------------------------------------------------------------------------
-
-DEBUG = environ.get("DEBUG")
-
-# --------------------------------------------------------------------------------------------------
-
-
-def load_toml_file(file: str) -> dict:
-    """Load TOML file"""
-
-    info: str = "load toml file"
-
-    try:
-
-        logger.info(f"{info} [ start ]")
-
-        # 不要加 encoding="utf-8" 参数, 否则会报错:
-        # binary mode doesn't take an encoding argument
-        with open(file, "rb") as _file:
-            config = tomllib.load(_file)
-
-        logger.success(f"{info} [ success ]")
-
-        return config
-
-    except Exception as e:
-
-        logger.error(f"{info} [ error ]")
-
-        if DEBUG:
-            logger.exception(e)
-        else:
-            logger.error(e)
-
-        return {}
-
-
-# --------------------------------------------------------------------------------------------------