Эх сурвалжийг харах

[hplsql] using right method for checking key presence (#2684)

Ayush Goyal 3 жил өмнө
parent
commit
9ec1b48e6a

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

@@ -737,7 +737,7 @@ DROP TABLE IF EXISTS `%(table)s`;
       functions = next((prop['value'] for prop in properties if prop['key'] == 'functions'), None)
 
     database = snippet.get('database') or 'default'
-    query_type = QUERY_TYPES[4] if hasattr(snippet, 'dialect') and snippet['dialect'] == 'hplsql' else QUERY_TYPES[0]
+    query_type = QUERY_TYPES[4] if 'dialect' in snippet and snippet['dialect'] == 'hplsql' else QUERY_TYPES[0]
 
     return hql_query(
       statement,

+ 1 - 1
desktop/libs/notebook/src/notebook/sql_utils.py

@@ -57,7 +57,7 @@ def get_current_statement(snippet):
   statement_id = handle.get('statement_id', 0)
   statements_count = handle.get('statements_count', 1)
 
-  statements = get_statements(snippet['statement'], snippet['dialect'] if hasattr(snippet, 'dialect') else None)
+  statements = get_statements(snippet['statement'], snippet['dialect'] if 'dialect' in snippet else None)
 
   statement_id = min(statement_id, len(statements) - 1) # In case of removal of statements
   previous_statement_hash = compute_statement_hash(statements[statement_id]['statement'])