ezKit 2.0.5__tar.gz → 2.0.7__tar.gz

{ezkit-2.0.5/ezKit.egg-info → ezkit-2.0.7}/PKG-INFO

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

ezkit-2.0.7/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.5 → ezkit-2.0.7}/ezKit/utils.py

@@ -3,6 +3,7 @@
 import re
 import tomllib
 
+from datetime import datetime, timezone
 from os import environ
 from typing import Iterable
 
@@ -143,3 +144,20 @@ def check_dict_values_incorrect(
             logger.error(e)
 
         return False
+
+
+# --------------------------------------------------------------------------------------------------
+
+
+def timestamp_to_datetime(timestamp: int | float, tz: timezone = timezone.utc) -> datetime | None:
+    """Unix Timestamp 转换为 Datatime"""
+    try:
+        if not isinstance(timestamp, (int, float)):
+            return None
+        return (datetime.fromtimestamp(timestamp, tz=tz)).replace(tzinfo=None)
+    except Exception as e:
+        if DEBUG:
+            logger.exception(e)
+        else:
+            logger.error(e)
+        return None

{ezkit-2.0.5 → ezkit-2.0.7/ezKit.egg-info}/PKG-INFO

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

{ezkit-2.0.5 → ezkit-2.0.7}/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.5 → ezkit-2.0.7}/setup.py

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