ezKit 1.12.6__py3-none-any.whl → 1.12.7__py3-none-any.whl

ezKit/database.py

@@ -6,8 +6,10 @@
 #     https://docs.sqlalchemy.org/en/20/core/connections.html#sqlalchemy.engine.CursorResult
 # PostgreSQL 14 Data Types
 #     https://www.postgresql.org/docs/14/datatype.html
+
 import csv
 import json
+from datetime import date, datetime
 from typing import Any, Type
 
 import pandas as pd
@@ -20,6 +22,51 @@ from . import utils
 Base = declarative_base()
 
 
+def orm_object_to_dict(obj, include: list | None = None) -> dict | None:
+    """
+    将 ORM 对象转为 dict, 可选择只包含部分字段.
+    :param obj: SQLAlchemy ORM 实例
+    :param include: 要保留的字段列表(白名单)
+    """
+    if obj is None:
+        return None
+
+    data = {}
+
+    for column in obj.__table__.columns:
+        key = column.name
+        if include and key not in include:
+            continue
+        value = getattr(obj, key)
+        if isinstance(value, (datetime, date)):
+            # data[key] = value.isoformat()
+            data[key] = value.strftime("%Y-%m-%d %H:%M:%S")
+        else:
+            data[key] = value
+
+    return data
+
+
+def orm_list_to_dict(obj_list, include: list | None = None) -> list:
+    return [orm_object_to_dict(obj, include) for obj in obj_list]
+
+
+def orm_to_dict(obj, include: list | None = None) -> dict | list | None:
+    """
+    ORM 对象 (单个或列表) 转 JSON 字符串.
+    :param include: 仅导出指定字段（白名单）
+    """
+    try:
+        if isinstance(obj, list):
+            data = orm_list_to_dict(obj, include)
+        else:
+            data = orm_object_to_dict(obj, include)
+        return data
+    except Exception as e:
+        logger.exception(e)
+        return None
+
+
 class Database:
     """Database"""
 

{ezkit-1.12.6.dist-info → ezkit-1.12.7.dist-info}/METADATA

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

{ezkit-1.12.6.dist-info → ezkit-1.12.7.dist-info}/RECORD

@@ -3,7 +3,7 @@ ezKit/_file.py,sha256=0qRZhwYuagTgTGrhm-tzAMvEQT4HTJA_xZKqF2bo0ho,1207
 ezKit/bottle.py,sha256=43h4v1kzz6qrLvCt5IMN0H-gFtaT0koG9wETqteXsps,181666
 ezKit/bottle_extensions.py,sha256=27rogmfK7mL2qUSjXH79IMGZbCVULtYEikql_N9O6Zs,1165
 ezKit/cipher.py,sha256=7jBarRH7ukSYzkz-Anl8B8JzluhnRz4CLHidPRRj_cg,2939
-ezKit/database.py,sha256=JbaBLoG0wUeu86kkroiliFB69bEZvzpt9BWRxzsy07c,14242
+ezKit/database.py,sha256=1_RmCwLA6kzigcMcewkoOYStkhFd_w-Qn84vGy-tInI,15553
 ezKit/dockerhub.py,sha256=j-wQO-71BsOgExHZhYynuy2k_hCX3on-vg0TH7QCit4,1996
 ezKit/http.py,sha256=zhNxJF-x91UqGncXWxVXnhZVpFo_wmmpGnMXVT11y9E,1832
 ezKit/markdown_to_html.template,sha256=21G2sSVGJn6aJvHd0NN4zY5YiDteKe4UtW36AzBwSdk,22274
@@ -15,8 +15,8 @@ ezKit/token.py,sha256=Ac-i9xfq4TqpGyfCzakjrh4NYzxHiN2sCQrMk1tzVi8,1716
 ezKit/utils.py,sha256=VyAw2yEs15VER9RCpMgskx1l0zrSpRlV34X___eOia0,42416
 ezKit/xftp.py,sha256=-XQXyhMqeigT63P6sXkSS7r4GROXyqqlkzKxITLWG-g,8278
 ezKit/zabbix.py,sha256=PkMnfu7mcuotwwIIsHaC9FsNg-gap6hD1xvm0AwSL1Y,33777
-ezkit-1.12.6.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-ezkit-1.12.6.dist-info/METADATA,sha256=tTGxSiL8d5cXfFIqIWd8IR8RKiXEC8Jw0r0IbLOvEC8,316
-ezkit-1.12.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ezkit-1.12.6.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
-ezkit-1.12.6.dist-info/RECORD,,
+ezkit-1.12.7.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+ezkit-1.12.7.dist-info/METADATA,sha256=1uzq8FpjYS1rIP8iewgixWFGzahSaQe32exFrp2xYyc,316
+ezkit-1.12.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ezkit-1.12.7.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
+ezkit-1.12.7.dist-info/RECORD,,