Browse Source

[sqlalchemy] fix presto progress error message when query is short

penggongkui 4 years ago
parent
commit
d462f3010b
1 changed files with 1 additions and 1 deletions
  1. 1 1
      desktop/libs/notebook/src/notebook/connectors/sql_alchemy.py

+ 1 - 1
desktop/libs/notebook/src/notebook/connectors/sql_alchemy.py

@@ -296,7 +296,7 @@ class SqlAlchemyApi(Api):
 
 
     if connection:
     if connection:
       cursor = connection['result'].cursor
       cursor = connection['result'].cursor
-      if self.options['url'].startswith('presto://') and cursor and cursor.poll():
+      if self.options['url'].startswith('presto://') and cursor and cursor._state == cursor._STATE_RUNNING:
         response['status'] = 'running'
         response['status'] = 'running'
       elif snippet['result']['handle']['has_result_set']:
       elif snippet['result']['handle']['has_result_set']:
         response['status'] = 'available'
         response['status'] = 'available'