Skip to content

Commit 2e07453

Browse files
committed
Fix issue of queries not reaching completed state
1 parent e260273 commit 2e07453

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

trino/sqlalchemy/dialect.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,7 @@ def get_view_definition(self, connection: Connection, view_name: str, schema: st
231231
"""
232232
).strip()
233233
res = connection.execute(sql.text(query), schema=schema, view=view_name)
234-
return res.scalar()
234+
return res.scalar_one()
235235

236236
def get_indexes(self, connection: Connection, table_name: str, schema: str = None, **kw) -> List[Dict[str, Any]]:
237237
if not self.has_table(connection, table_name, schema):
@@ -275,7 +275,7 @@ def get_table_comment(self, connection: Connection, table_name: str, schema: str
275275
).strip()
276276
try:
277277
res = connection.execute(sql.text(query), schema=schema, table_name=table_name)
278-
return dict(text=res.scalar())
278+
return dict(text=res.scalar_one())
279279
except error.TrinoQueryError as e:
280280
if e.error_name in (
281281
error.PERMISSION_DENIED,
@@ -317,7 +317,7 @@ def _get_server_version_info(self, connection: Connection) -> Any:
317317
query = "SELECT version()"
318318
try:
319319
res = connection.execute(sql.text(query))
320-
version = res.scalar()
320+
version = res.scalar_one()
321321
return tuple([version])
322322
except exc.ProgrammingError as e:
323323
logger.debug(f"Failed to get server version: {e.orig.message}")

0 commit comments

Comments
 (0)