浏览代码

use url.startswith('presto://') honor the other code

penggongkui 5 年之前
父节点
当前提交
9cd09280d0
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      desktop/libs/notebook/src/notebook/connectors/sql_alchemy.py

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

@@ -297,7 +297,7 @@ class SqlAlchemyApi(Api):
 
 
     if connection:
     if connection:
       cursor = connection['result'].cursor
       cursor = connection['result'].cursor
-      if snippet.get('dialect') == 'presto' and cursor and cursor.poll():
+      if self.options['url'].startswith('presto://') and cursor and cursor.poll():
         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'
@@ -310,7 +310,7 @@ class SqlAlchemyApi(Api):
 
 
   @query_error_handler
   @query_error_handler
   def progress(self, notebook, snippet, logs=''):
   def progress(self, notebook, snippet, logs=''):
-    if snippet.get('dialect') == 'presto':
+    if self.options['url'].startswith('presto://'):
       guid = snippet['result']['handle']['guid']
       guid = snippet['result']['handle']['guid']
       handle = CONNECTIONS.get(guid)
       handle = CONNECTIONS.get(guid)
       if not handle:
       if not handle: