Expose query_id property on dbapi.Cursor

Author: mdesmet

tests/integration/test_dbapi_integration.py

@@ -77,6 +77,8 @@ def test_select_query(trino_connection):
     assert columns["node_version"] == "varchar"
     assert columns["coordinator"] == "boolean"
     assert columns["state"] == "varchar"
+    assert cur.query_id is not None
+    assert cur.stats is not None
 
 
 def test_select_query_result_iteration(trino_connection):

trino/dbapi.py

@@ -21,7 +21,7 @@
 import math
 import uuid
 from decimal import Decimal
-from typing import Any, List, Optional  # NOQA for mypy types
+from typing import Any, Dict, List, Optional  # NOQA for mypy types
 
 import trino.client
 import trino.exceptions
@@ -289,6 +289,12 @@ def stats(self):
             return self._query.stats
         return None
 
+    @property
+    def query_id(self) -> Optional[str]:
+        if self._query is not None:
+            return self._query.query_id
+        return None
+
     @property
     def warnings(self):
         if self._query is not None: