ezKit 1.12.27__py3-none-any.whl → 1.12.29__py3-none-any.whl

ezKit/database.py

@@ -15,15 +15,17 @@ from typing import Any, Dict, List, Optional, Tuple, Type
 
 import pandas as pd
 from loguru import logger
-from sqlalchemy import CursorResult, Engine, Index, bindparam, create_engine, text
+from sqlalchemy import CursorResult, Engine, Index, bindparam, create_engine, insert, text
 from sqlalchemy.engine import Result
 from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker
 from sqlalchemy.orm import DeclarativeBase, Session, declarative_base
 
 from . import utils
+from .errors import database_error, sqlalchemy_error
 
-Base = declarative_base()
+# --------------------------------------------------------------------------------------------------
 
+Base = declarative_base()
 
 # --------------------------------------------------------------------------------------------------
 
@@ -698,3 +700,175 @@ class DatabaseAsyncSession:
             return []
         rows = result.mappings().all()
         return [dict(row) for row in rows]
+
+
+# --------------------------------------------------------------------------------------------------
+
+
+async def db_async_create(db: DatabaseAsyncSession, schema: Type[DeclarativeBase], data: list) -> dict:
+
+    # 插入数据
+    result = await db.operater(insert(schema), data)
+
+    # 返回结果
+    if result is None:
+        return {"code": 1001, "data": 0, "message": database_error["create"]["message"]}
+
+    return {"code": 200, "data": len(data), "message": "create data success"}
+
+
+# --------------------------------------------------------------------------------------------------
+
+
+async def db_async_read(db: DatabaseAsyncSession, table_name: str, data: dict, statement_prefix: str, statement_end: str = "") -> dict:
+
+    # 初始返回结果 (适用于 count 等于 0)
+    result = {"count": 0, "data": None, "pageIndex": data["pageIndex"], "pageSize": 0}
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 构建 WHERE
+
+    where_collections = build_sqlalchemy_select_where(data)
+
+    if where_collections is None:
+        return {"code": 1001, "data": None, "message": sqlalchemy_error["build_where"]["message"]}
+
+    where_clause, statement_params, bind_params = where_collections
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 统计 WHERE
+
+    count_statement = text(f"SELECT COUNT(*) FROM {table_name} {where_clause}").bindparams(*bind_params)
+
+    count_result = await db.operate_return_scalar_one(count_statement, statement_params)
+
+    if count_result is None:
+        return {"code": 1002, "data": None, "message": sqlalchemy_error["statement_params"]["message"]}
+
+    if count_result == 0:
+        return {"code": 200, "data": result, "message": "read data success"}
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 读取数据
+
+    read_statement = text(
+        f"""
+            {statement_prefix}
+            {where_clause}
+            {statement_end}
+            LIMIT :limit
+            OFFSET :offset;
+        """
+    ).bindparams(*bind_params)
+
+    read_result = await db.operate_return_mappings_all(read_statement, statement_params)
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 数据整合
+    result["count"] = count_result
+    result["data"] = read_result
+    result["pageSize"] = len(read_result)
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 返回结果
+    return {"code": 200, "data": result, "message": "read data success"}
+
+
+# --------------------------------------------------------------------------------------------------
+
+
+async def db_async_update(db: DatabaseAsyncSession, table_name: str, data: dict) -> dict:
+
+    # 构建 WHERE
+
+    where_collections = build_sqlalchemy_select_where(data["where"])
+
+    if where_collections is None:
+        return {"code": 1001, "data": 0, "message": sqlalchemy_error["build_where"]["message"]}
+
+    where_clause, statement_params, bind_params = where_collections
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 统计 WHERE
+
+    count_statement = text(f"SELECT COUNT(*) FROM {table_name} {where_clause}").bindparams(*bind_params)
+
+    count_result = await db.operate_return_scalar_one(count_statement, statement_params)
+
+    if count_result is None:
+        return {"code": 1002, "data": None, "message": sqlalchemy_error["statement_params"]["message"]}
+
+    if count_result == 0:
+        return {"code": 1003, "data": 0, "message": database_error["update"]["message"]}
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 更新数据
+
+    update_collections = build_sqlalchemy_update(table_name, data["where"], data["values"])
+
+    if update_collections is None:
+        return {"code": 1004, "data": None, "message": sqlalchemy_error["build_statement"]["message"]}
+
+    update_statement, update_params = update_collections
+
+    update_result = await db.operater(update_statement, update_params)
+
+    if update_result is None:
+        return {"code": 1005, "data": None, "message": database_error["update"]["message"]}
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 返回结果
+    return {"code": 200, "data": count_result, "message": "update data success"}
+
+
+# --------------------------------------------------------------------------------------------------
+
+
+async def db_async_delete(db: DatabaseAsyncSession, table_name: str, data: dict) -> dict:
+
+    # 构建 WHERE
+
+    where_collections = build_sqlalchemy_select_where(data)
+
+    if where_collections is None:
+        return {"code": 1001, "data": 0, "message": sqlalchemy_error["build_where"]["message"]}
+
+    where_clause, statement_params, bind_params = where_collections
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 统计 WHERE
+
+    count_statement = text(f"SELECT COUNT(*) FROM {table_name} {where_clause}").bindparams(*bind_params)
+
+    count_result = await db.operate_return_scalar_one(count_statement, statement_params)
+
+    if count_result is None:
+        return {"code": 1002, "data": None, "message": sqlalchemy_error["statement_params"]["message"]}
+
+    if count_result == 0:
+        return {"code": 1003, "data": 0, "message": database_error["delete"]["message"]}
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 删除数据
+
+    delete_statement = text(f"DELETE FROM {table_name} {where_clause}").bindparams(*bind_params)
+
+    delete_result = await db.operater(delete_statement, statement_params)
+
+    if delete_result is None:
+        return {"code": 1004, "data": 0, "message": database_error["delete"]["message"]}
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 返回结果
+    return {"code": 200, "data": count_result, "message": "delete data success"}

ezKit/errors.py

@@ -48,8 +48,16 @@ sqlalchemy_error = {
         "code": 4003,
         "message": "SQLAlchemy statement or params error",
     },
-    "build_where": {
+    "build_statement": {
         "code": 4004,
+        "message": "SQLAlchemy build statement error",
+    },
+    "build_params": {
+        "code": 4005,
+        "message": "SQLAlchemy build params error",
+    },
+    "build_where": {
+        "code": 4006,
         "message": "SQLAlchemy build WHERE error",
     },
 }

{ezkit-1.12.27.dist-info → ezkit-1.12.29.dist-info}/METADATA

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

{ezkit-1.12.27.dist-info → ezkit-1.12.29.dist-info}/RECORD

@@ -3,9 +3,9 @@ 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=yxi8HZw9ijbfKYhCl8OApoLDwLW7tYjPPucsTw6D_Is,23761
+ezKit/database.py,sha256=5MsIYDhyN6VtQ8MJy7pbMGqeaXguy9DzRHgJ6NJD5Gs,30126
 ezKit/dockerhub.py,sha256=j-wQO-71BsOgExHZhYynuy2k_hCX3on-vg0TH7QCit4,1996
-ezKit/errors.py,sha256=P23_yiQJNA6jJpNpzXb2o27uqmGHM-mRKa5171hQWu8,1122
+ezKit/errors.py,sha256=wLLzJ-YWCdAUqdLzsHM-jfnfZJL9pHtZoG_O_bVNI_A,1334
 ezKit/http.py,sha256=zhNxJF-x91UqGncXWxVXnhZVpFo_wmmpGnMXVT11y9E,1832
 ezKit/markdown_to_html.template,sha256=21G2sSVGJn6aJvHd0NN4zY5YiDteKe4UtW36AzBwSdk,22274
 ezKit/mongo.py,sha256=vsRCjJ2uWbNp-1bnGKICPohNYx25El8XnHp10o-lsM4,2397
@@ -16,8 +16,8 @@ ezKit/token.py,sha256=Ac-i9xfq4TqpGyfCzakjrh4NYzxHiN2sCQrMk1tzVi8,1716
 ezKit/utils.py,sha256=U457ahFkxIXuB-qWvS3995xJs-LlkFIX5_ZWVgmL5cY,43130
 ezKit/xftp.py,sha256=-XQXyhMqeigT63P6sXkSS7r4GROXyqqlkzKxITLWG-g,8278
 ezKit/zabbix.py,sha256=PkMnfu7mcuotwwIIsHaC9FsNg-gap6hD1xvm0AwSL1Y,33777
-ezkit-1.12.27.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-ezkit-1.12.27.dist-info/METADATA,sha256=WhxULotC-6zBWIc6OGXN6prPRVjqDWXsPfyIF3CSbiY,317
-ezkit-1.12.27.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ezkit-1.12.27.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
-ezkit-1.12.27.dist-info/RECORD,,
+ezkit-1.12.29.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+ezkit-1.12.29.dist-info/METADATA,sha256=8eEFRDzl6V1laRNNs2E1maD3xcSGOXBlNBVTr9s2geQ,317
+ezkit-1.12.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ezkit-1.12.29.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
+ezkit-1.12.29.dist-info/RECORD,,