Check cursor _query is None

trinodb · Nov 3, 2023 · f99ef24 · f99ef24
1 parent e4249f4
commit f99ef24
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/tests/unit/test_dbapi.py b/tests/unit/test_dbapi.py
@@ -309,3 +309,9 @@ def test_hostname_parsing():
     assert only_hostname_with_path.host == "mytrinoserver.domain/some_path"
     assert only_hostname_with_path.port == 8080
     assert only_hostname_with_path.http_scheme == constants.HTTP
+
+
+def test_description_is_none_when_cursor_is_not_executed():
+    connection = Connection("sample_trino_cluster:443")
+    cursor = connection.cursor()
+    assert hasattr(cursor, 'description')
diff --git a/trino/dbapi.py b/trino/dbapi.py
@@ -376,7 +376,7 @@ def update_type(self):
 
     @property
     def description(self) -> List[ColumnDescription]:
-        if self._query.columns is None:
+        if self._query is None or self._query.columns is None:
             return None
 
         # [ (name, type_code, display_size, internal_size, precision, scale, null_ok) ]