datahub_binary 1.9.4__cp312-cp312-win_amd64.whl → 1.9.6__cp312-cp312-win_amd64.whl

datahub/dbo/database.pyi

@@ -1,4 +1,5 @@
 import polars as pl
+import types
 from _typeshed import Incomplete
 from contextlib import contextmanager
 from datahub.utils.logger import logger as logger
@@ -12,93 +13,59 @@ class Database:
     schema: Incomplete
     session_factory: Incomplete
     def __init__(self, connection_string: str, pool_size: int = 3, max_overflow: int = 10, pool_timeout: int = 30, pool_recycle: int = 3600) -> None: ...
-    def insert_many(self, table_name: str, data: list[dict[str, Any]]) -> int:
+    def begin(self) -> Database:
+        """开启事务"""
+    def commit(self) -> None:
+        """提交事务"""
+    def rollback(self) -> None:
+        """回滚事务"""
+    @property
+    def in_transaction(self) -> bool:
+        """是否在事务中"""
+    @contextmanager
+    def transaction(self) -> Generator['Database', Any, None]:
         """
-        批量插入数据
+        事务上下文管理器，自动 commit/rollback
 
-        :param table_name: 表名
-        :param data: 要插入的数据列表
-        :return: 影响的行数
+        Usage:
+            with db.transaction():
+                db.insert(...)
+                db.update(...)
         """
+    def insert(self, table_name: str, data: dict[str, Any]) -> int:
+        """插入单条数据"""
+    def insert_many(self, table_name: str, data: list[dict[str, Any]]) -> int:
+        """批量插入数据"""
     def insert_ignore(self, table_name: str, data: dict, keys: list) -> bool:
         """
         插入忽略,不存在就插入,存在就忽略（单条）
-        :param table_name: 表名
-        :param data: 插入数据
-        :param keys: 关键字,关键字相同则认为数据相同
-        :return: 是否成功
         """
     def insert_many_ignore(self, table_name: str, data: list[dict], keys: list) -> bool:
         """
         批量插入忽略,不存在就插入,存在就忽略
-        :param table_name: 表名
-        :param data: 插入数据 [{row1:v1, row2:v2...}...]
-        :param keys: 关键字,关键字相同则认为数据相同
-        :return: 是否成功
         """
     def upsert(self, table_name: str, data: dict, keys: list) -> bool:
-        """
-        插入更新,不存在就插入,存在就更新
-        :param table_name: 表名
-        :param data: 插入数据
-        :param keys: 关键字,关键字相同则认为数据相同
-        :return: 是否成功
-        """
+        """插入更新,不存在就插入,存在就更新"""
     def upsert_many(self, table_name: str, data: list[dict], keys: list) -> bool:
-        """
-        批量插入更新,不存在就插入,存在就更新
-        :param table_name: 表名
-        :param data: 插入数据[{row1:v1,row2:v2...}...]
-        :param keys: 关键字,关键字相同则认为数据相同
-        :return: 是否成功
-        """
+        """批量插入更新"""
     def update(self, table_name: str, data: dict, keys: list) -> int:
-        '''
-        更新数据
-        :param table_name: 表名
-        :param data: 字典数据{"字段名":值}
-        :param keys: 索引字段["ID",...]
-        :return: 更新的行数
-        '''
+        """更新数据"""
     def update_many(self, table_name: str, rows: list, keys: list) -> bool:
-        '''
-        批量更新数据
-        :param table_name: 表名
-        :param rows: [{"字段名":值}...]
-        :param keys: 索引字段["ID",...]
-        :return: 更新结果
-        '''
+        """批量更新数据"""
     def delete(self, table_name: str, **filters) -> int:
-        '''
-        删除数据
-        :param table_name: 表名
-        :param filters: "字段名"=值
-        :return: 删除行数
-        '''
+        """删除数据"""
     def select(self, table_name: str, **filters) -> list[dict]:
-        '''
-        查询数据
-        :param table_name: 表名
-        :param filters: "字段名"=值（支持单值或列表）
-        :return: 查询结果列表
-        '''
+        """查询数据"""
     def query(self, sql: str, return_format: Literal['dataframe', 'records'] = 'records') -> pl.DataFrame | list[dict] | None:
-        """
-        执行原生SQL查询
-
-        :param sql: SQL查询语句
-        :param return_format: 返回格式
-        :return: 查询结果
-        """
+        """执行原生SQL查询"""
+    def execute(self, sql: str) -> int:
+        """执行原生SQL语句（INSERT/UPDATE/DELETE等）"""
     @contextmanager
     def get_session(self) -> Generator[Session, Any, None]:
-        """获取一个数据库会话，支持用户手动管理 session 生命周期"""
+        """获取一个数据库会话"""
     def query_with_session(self, sql: str, session: Session, return_format: Literal['dataframe', 'records'] = 'records'):
-        """
-        执行原生SQL查询
-
-        :param sql: SQL查询语句
-        :param session: 数据库会话
-        :param return_format: 返回格式
-        :return: 查询结果
-        """
+        """使用指定 session 执行原生SQL查询"""
+    def close(self) -> None:
+        """关闭数据库连接"""
+    def __enter__(self) -> Database: ...
+    def __exit__(self, exc_type: type[BaseException] | None, exc_val: BaseException | None, exc_tb: types.TracebackType | None) -> None: ...

{datahub_binary-1.9.4.dist-info → datahub_binary-1.9.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datahub_binary
-Version: 1.9.4
+Version: 1.9.6
 Summary: A comprehensive Python library for data processing, integration, and management.
 Requires-Python: <3.13,>=3.10
 Description-Content-Type: text/markdown

{datahub_binary-1.9.4.dist-info → datahub_binary-1.9.6.dist-info}/RECORD

@@ -1,11 +1,11 @@
-datahub.cp312-win_amd64.pyd,sha256=T8Ut9F3uEZpJfo5gViK7Ky6IcVbAd8gW9daCPWiWOt4,1923584
+datahub.cp312-win_amd64.pyd,sha256=jDEIFNy9XCFW9AZ3V3saEyzvCqxWo0ESOAwGqULRo9c,1949696
 datahub.pyi,sha256=biltT_CgztnMyL9mbp332i5vT7LUMws0bN6fdQ3WYH4,1555
 datahub/__init__.pyi,sha256=hX8D9Qd6uPjZiUBd6rIq7Llkt_H4l_RvRXsws3sdYbI,630
 datahub/datacache.pyi,sha256=Z2b3ktYv5t2y9Z1yXn0axbdoV06-GhjJhEIl_Ms-FjU,5044
 datahub/datahub.pyi,sha256=NKfhw00iPMsp0Z_q4usO_D7xB6-Hr50wRxpe3KGYH7E,51370
 datahub/setting.pyi,sha256=u6gYjjF6r1thSpp2XNMvg_1JUcqB64dRTkNoHlOoKUQ,889
 datahub/ats/client_sdk.pyi,sha256=RScBpyY2ML84RBrYt60RnHtodOwWOe5-LbnJ3nFFBGs,8710
-datahub/dbo/database.pyi,sha256=zGncpT29ygIdMu3i8SzApdx2X08GBBspHor_Wy7PzjU,4100
+datahub/dbo/database.pyi,sha256=oD5Iw3B3FPFQqbFUufhWlwgHqnT8hsHvpDtX85A5XL8,3203
 datahub/dbo/pg.pyi,sha256=diMoeSPBqOswXYvVjtpfF5XklWqa_f20vQ1TPiHZ7WA,2077
 datahub/dbo/redis_stream.pyi,sha256=geVI1BoTHRlxodS44MquM-FY_aJt27eDqmAKf2a4PyM,1938
 datahub/dbo/sr.pyi,sha256=ZKcUUxGP-s4xGr_zF1qgVlRZcL4fkyROAOiJHuX-Fwk,28644
@@ -25,7 +25,7 @@ datahub/utils/sftp.pyi,sha256=w8S-cxJWMeWJ6--RjT95qVgng_IhwPImiyoqdCpOCQI,2107
 datahub/utils/monitor/__init__.pyi,sha256=EOwgf8CbJ4g3iUAaFoR6O-mYemJ9xjP42zlBj2KeA9c,76
 datahub/utils/monitor/base.pyi,sha256=n1dKYK73JJ_7QPkCWFc-6Aj_a16j2z_VBaQ9zkN9A3E,310
 datahub/utils/monitor/feishu.pyi,sha256=GgvJXYeX3cPOQjqpV0GJUr_Ri1_cZe2-viRBkpe6O_g,402
-datahub_binary-1.9.4.dist-info/METADATA,sha256=WhZ8OR0YNi_Mxrxth7-omQuKEboU846UkU99qBTqK_c,7903
-datahub_binary-1.9.4.dist-info/WHEEL,sha256=mktjIo72eqyqXY-lxdbITmUE47ZU2O9WYzyv0PnFKrU,96
-datahub_binary-1.9.4.dist-info/top_level.txt,sha256=J0rzqYfZghMgpObWaPGhQtV88I-KXrgArDvi_ddf6ZA,8
-datahub_binary-1.9.4.dist-info/RECORD,,
+datahub_binary-1.9.6.dist-info/METADATA,sha256=PL5Lrq8jWmnnB5J0kOaEtqlxEUDa1_UR_WwLOS8qPO4,7903
+datahub_binary-1.9.6.dist-info/WHEEL,sha256=mktjIo72eqyqXY-lxdbITmUE47ZU2O9WYzyv0PnFKrU,96
+datahub_binary-1.9.6.dist-info/top_level.txt,sha256=J0rzqYfZghMgpObWaPGhQtV88I-KXrgArDvi_ddf6ZA,8
+datahub_binary-1.9.6.dist-info/RECORD,,