ezKit 1.12.14__py3-none-any.whl → 1.12.16__py3-none-any.whl

ezKit/database.py

@@ -15,6 +15,8 @@ from typing import Any, Dict, Optional, Tuple, Type
 import pandas as pd
 from loguru import logger
 from sqlalchemy import CursorResult, Engine, Index, bindparam, create_engine, 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
@@ -586,3 +588,64 @@ class Database:
                 logger.error(f"{info} [failed]")
                 logger.exception(e)
                 return False
+
+
+# ----------------------------------------------------------------------------------------------
+
+
+class DatabaseAsyncSession:
+    """Database Async Session"""
+
+    AsyncSessionLocal: async_sessionmaker[AsyncSession]
+
+    def __init__(self, session: async_sessionmaker[AsyncSession]):
+        """Initiation"""
+        self.AsyncSessionLocal = session
+
+    # 执行器
+    async def operater(self, stmt, params, **kwargs) -> Result | None:
+
+        async with self.AsyncSessionLocal() as session:
+            try:
+                await session.begin()
+                result = await session.execute(stmt, params, **kwargs)
+                await session.commit()
+                return result
+            except Exception as e:
+                await session.rollback()
+                logger.exception(e)
+                return None
+
+    # 精确返回一个标量结果
+    async def operate_return_scalar_one(self, stmt, params, **kwargs) -> Any | None:
+        result = await self.operater(stmt, params, **kwargs)
+        if result is None:
+            return None
+        return result.scalar_one()
+
+    # 返回所有结果
+    async def operate_return_mappings_all(self, stmt, params, **kwargs) -> list:
+        result = await self.operater(stmt, params, **kwargs)
+        if result is None:
+            return []
+        rows = result.mappings().all()
+        return [dict(row) for row in rows]
+
+    # 创建数据
+    async def create_with_schema(self, schema: Type, data: list) -> bool:
+
+        async with self.AsyncSessionLocal() as session:
+            try:
+                await session.begin()
+                for item in data:
+                    if not utils.isTrue(item, dict):
+                        logger.error(f"data error: {item}")
+                        await session.rollback()
+                        return False
+                    await session.add(schema(**item))
+                await session.commit()
+                return True
+            except Exception as e:
+                await session.rollback()
+                logger.exception(e)
+                return False

{ezkit-1.12.14.dist-info → ezkit-1.12.16.dist-info}/METADATA

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

{ezkit-1.12.14.dist-info → ezkit-1.12.16.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=HK-TquMmlzPlRsiqmnXTGxQrrSHiRx65oOHoE6c4M84,19926
+ezKit/database.py,sha256=_nrxF_iqmHSNcRFvN-eODlDPdjcn1oMNxj4DygXvai0,22134
 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.14.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-ezkit-1.12.14.dist-info/METADATA,sha256=aXP1h-OH37Q4mhrUOAW9WnN6sUkEiz4Mvf5swf--IbY,317
-ezkit-1.12.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ezkit-1.12.14.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
-ezkit-1.12.14.dist-info/RECORD,,
+ezkit-1.12.16.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+ezkit-1.12.16.dist-info/METADATA,sha256=bnTT052ZZdLQfBS8USzxs5ebinHglJTo1KitZdKNdz0,317
+ezkit-1.12.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ezkit-1.12.16.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
+ezkit-1.12.16.dist-info/RECORD,,