فهرست منبع

[editor] Prevent List index out of range when number of queries changed

Romain Rigaux 9 سال پیش
والد
کامیت
92c7d201a5
1فایلهای تغییر یافته به همراه3 افزوده شده و 0 حذف شده
  1. 3 0
      desktop/libs/notebook/src/notebook/connectors/hiveserver2.py

+ 3 - 0
desktop/libs/notebook/src/notebook/connectors/hiveserver2.py

@@ -106,6 +106,7 @@ class HS2Api(Api):
 
     # Multiquery, if not first statement or arrived to the last query
     statement_id = snippet['result']['handle'].get('statement_id', 0)
+    statements_count = snippet['result']['handle'].get('statements_count', 1)
     if snippet['result']['handle'].get('has_more_statements'):
       try:
         handle = self._get_handle(snippet)
@@ -117,6 +118,8 @@ class HS2Api(Api):
       statement_id = 0
 
     statements = self._get_statements(snippet['statement'])
+    if statements_count != len(statements):
+      statement_id = 0
     statement = statements[statement_id]
 
     settings = snippet['properties'].get('settings', None)