databend-driver 0.28.2__cp37-abi3-macosx_10_12_x86_64.whl → 0.29.0__cp37-abi3-macosx_10_12_x86_64.whl

databend_driver/__init__.py

@@ -15,3 +15,31 @@
 # flake8: noqa
 
 from ._databend_driver import *
+
+# Export for convenience at module level
+__all__ = [
+    # Exception classes - PEP 249 compliant
+    "Warning",
+    "Error",
+    "InterfaceError",
+    "DatabaseError",
+    "DataError",
+    "OperationalError",
+    "IntegrityError",
+    "InternalError",
+    "ProgrammingError",
+    "NotSupportedError",
+    # Client classes
+    "AsyncDatabendClient",
+    "AsyncDatabendConnection",
+    "BlockingDatabendClient",
+    "BlockingDatabendConnection",
+    "BlockingDatabendCursor",
+    # Data types
+    "ConnectionInfo",
+    "Schema",
+    "Field",
+    "Row",
+    "RowIterator",
+    "ServerStats",
+]

databend_driver/__init__.pyi

@@ -14,6 +14,18 @@
 
 # flake8: noqa
 
+# Exception classes - PEP 249 compliant
+class Warning(Exception): ...
+class Error(Exception): ...
+class InterfaceError(Error): ...
+class DatabaseError(Error): ...
+class DataError(DatabaseError): ...
+class OperationalError(DatabaseError): ...
+class IntegrityError(DatabaseError): ...
+class InternalError(DatabaseError): ...
+class ProgrammingError(DatabaseError): ...
+class NotSupportedError(DatabaseError): ...
+
 class ServerStats:
     @property
     def total_rows(self) -> int: ...
@@ -71,6 +83,9 @@ class RowIterator:
 class AsyncDatabendConnection:
     async def info(self) -> ConnectionInfo: ...
     async def version(self) -> str: ...
+    async def close(self) -> None: ...
+    def last_query_id(self) -> str | None: ...
+    async def kill_query(self, query_id: str) -> None: ...
     async def exec(self, sql: str) -> int: ...
     async def query_row(self, sql: str) -> Row: ...
     async def query_iter(self, sql: str) -> RowIterator: ...
@@ -86,6 +101,9 @@ class AsyncDatabendClient:
 class BlockingDatabendConnection:
     def info(self) -> ConnectionInfo: ...
     def version(self) -> str: ...
+    def close(self) -> None: ...
+    def last_query_id(self) -> str | None: ...
+    def kill_query(self, query_id: str) -> None: ...
     def exec(self, sql: str) -> int: ...
     def query_row(self, sql: str) -> Row: ...
     def query_iter(self, sql: str) -> RowIterator: ...

{databend_driver-0.28.2.dist-info → databend_driver-0.29.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databend-driver
-Version: 0.28.2
+Version: 0.29.0
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
@@ -136,6 +136,26 @@ row = await context.conn.query_row(
 row = await context.conn.query_row("SELECT ?, ?, ?, ?", params = (3, False, 4, "55"))
 ```
 
+### Query ID tracking and query management
+
+```python
+# Get the last executed query ID
+query_id = conn.last_query_id()
+print(f"Last query ID: {query_id}")
+
+# Execute a query and get its ID
+await conn.query_row("SELECT 1")
+query_id = conn.last_query_id()
+print(f"Query ID: {query_id}")
+
+# Kill a running query (if needed)
+try:
+    await conn.kill_query("some-query-id")
+    print("Query killed successfully")
+except Exception as e:
+    print(f"Failed to kill query: {e}")
+```
+
 ## Type Mapping
 
 [Databend Types](https://docs.databend.com/sql/sql-reference/data-types/)
@@ -188,6 +208,32 @@ print(value)
 
 ## APIs
 
+### Exception Classes (PEP 249 Compliant)
+
+The driver provides a complete set of exception classes that follow the PEP 249 standard for database interfaces:
+
+```python
+# Base exceptions
+class Warning(Exception): ...
+class Error(Exception): ...
+
+# Interface errors
+class InterfaceError(Error): ...
+
+# Database errors
+class DatabaseError(Error): ...
+
+# Specific database error types
+class DataError(DatabaseError): ...
+class OperationalError(DatabaseError): ...
+class IntegrityError(DatabaseError): ...
+class InternalError(DatabaseError): ...
+class ProgrammingError(DatabaseError): ...
+class NotSupportedError(DatabaseError): ...
+```
+
+These exceptions are automatically mapped from Databend error codes to appropriate PEP 249 exception types based on the nature of the error.
+
 ### AsyncDatabendClient
 
 ```python
@@ -203,6 +249,8 @@ class AsyncDatabendConnection:
     async def info(self) -> ConnectionInfo: ...
     async def version(self) -> str: ...
     async def close(self) -> None: ...
+    def last_query_id(self) -> str | None: ...
+    async def kill_query(self, query_id: str) -> None: ...
     async def exec(self, sql: str, params: list[string] | tuple[string] | any = None) -> int: ...
     async def query_row(self, sql: str, params: list[string] | tuple[string] | any = None) -> Row: ...
     async def query_iter(self, sql: str, params: list[string] | tuple[string] | any = None) -> RowIterator: ...
@@ -226,6 +274,8 @@ class BlockingDatabendConnection:
     def info(self) -> ConnectionInfo: ...
     def version(self) -> str: ...
     def close(self) -> None: ...
+    def last_query_id(self) -> str | None: ...
+    def kill_query(self, query_id: str) -> None: ...
     def exec(self, sql: str, params: list[string] | tuple[string] | any = None) -> int: ...
     def query_row(self, sql: str, params: list[string] | tuple[string] | any = None) -> Row: ...
     def query_iter(self, sql: str, params: list[string] | tuple[string] | any = None) -> RowIterator: ...

databend_driver-0.29.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+databend_driver-0.29.0.dist-info/METADATA,sha256=spT4mbCgPxm42V8AdS6GKsMYxrZA4bjXo9qFESIhYak,10964
+databend_driver-0.29.0.dist-info/WHEEL,sha256=EXWYGIlQOGI1SAWXpNs-tKqcTH553LQS8liEUK4J-U8,104
+databend_driver/__init__.py,sha256=p5VDMqNyirYvuomUSZZk4aRDfw76XNqmnIgvsqiGFKE,1218
+databend_driver/__init__.pyi,sha256=FOf7g35hJzZ6Ugf4op3k-0ZKt1Hq8lufJJElanp7VIk,4665
+databend_driver/_databend_driver.abi3.so,sha256=GxzuBSS7rZPsdJ_ly2aqZ1uL2_c2IG4x11o_BBG_MG0,18032064
+databend_driver-0.29.0.dist-info/RECORD,,

{databend_driver-0.28.2.dist-info → databend_driver-0.29.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: maturin (1.9.3)
+Generator: maturin (1.9.4)
 Root-Is-Purelib: false
 Tag: cp37-abi3-macosx_10_12_x86_64

databend_driver-0.28.2.dist-info/RECORD

@@ -1,6 +0,0 @@
-databend_driver-0.28.2.dist-info/METADATA,sha256=HS-QygPix18c3JyLs2f4CgZqpRBMNz4iUZ9mWZ1JyTE,9507
-databend_driver-0.28.2.dist-info/WHEEL,sha256=aZuW2QT8h7dxnk_E9cUNpI1w3saVkxUgks0fBGyHTZQ,104
-databend_driver/__init__.py,sha256=sNam6xQNlyVYUm7qBdX4cDxsxlgHgr3zYavOp8uielw,626
-databend_driver/__init__.pyi,sha256=14ys4K9rDVX-eRUcwcpYB9qi8hCoERQNoaAJf1SLg80,3976
-databend_driver/_databend_driver.abi3.so,sha256=7Z1lA8Wfphv2yE2OMrRKC7m3Pg7HRwk_2Phnti__JlE,17969592
-databend_driver-0.28.2.dist-info/RECORD,,