cxppython 0.2.9__tar.gz → 0.2.10__tar.gz

{cxppython-0.2.9 → cxppython-0.2.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cxppython
-Version: 0.2.9
+Version: 0.2.10
 Summary: A python utils package
 Home-page: https://github.com/yourusername/my_package
 Author: cxp

{cxppython-0.2.9 → cxppython-0.2.10}/cxppython/utils/database/mysql.py

@@ -48,12 +48,59 @@ class MysqlDB:
         self.async_engine: Optional[AsyncEngine] = None
         self.async_session_factory: Optional[Any] = None
         self._async_engine_created: bool = False
+        self._ctx_session: Optional[Session] = None
+        self._async_ctx_session: Optional[AsyncSession] = None
 
         logger.info("MysqlDB 同步引擎已创建。")
 
         if async_engine:
             self._ensure_async_engine()
 
+    # --- 上下文管理协议，便于 with 使用 ---
+    def __enter__(self) -> Session:
+        """支持 `with MysqlDB as session` 便捷获取会话。"""
+        self._ctx_session = self.session_factory()
+        return self._ctx_session
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        session = self._ctx_session
+        if session is None:
+            return False
+        try:
+            if exc_type:
+                if session.in_transaction():
+                    session.rollback()
+            else:
+                if session.in_transaction():
+                    session.commit()
+        finally:
+            session.close()
+            self._ctx_session = None
+        # 返回 False 以便异常向外继续传播
+        return False
+
+    async def __aenter__(self) -> AsyncSession:
+        """支持 `async with MysqlDB as session`。"""
+        self._ensure_async_engine()
+        self._async_ctx_session = self.async_session_factory()
+        return self._async_ctx_session
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        session = self._async_ctx_session
+        if session is None:
+            return False
+        try:
+            if exc_type:
+                if session.in_transaction():
+                    await session.rollback()
+            else:
+                if session.in_transaction():
+                    await session.commit()
+        finally:
+            await session.close()
+            self._async_ctx_session = None
+        return False
+
     def _ensure_async_engine(self):
         """延迟创建异步引擎（线程安全）"""
         if not self._async_engine_created:

{cxppython-0.2.9 → cxppython-0.2.10}/cxppython.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cxppython
-Version: 0.2.9
+Version: 0.2.10
 Summary: A python utils package
 Home-page: https://github.com/yourusername/my_package
 Author: cxp

{cxppython-0.2.9 → cxppython-0.2.10}/setup.py

@@ -4,7 +4,7 @@ with open("README.md", "r", encoding="utf-8") as f:
     long_description = f.read()
 setup(
     name="cxppython",              # 包名
-    version="0.2.9",                # 版本号
+    version="0.2.10",                # 版本号
     packages=find_packages(exclude=["tests", "tests.*"]),       # 自动找到所有包
     description="A python utils package",
     long_description=long_description,