Jelajahi Sumber

HUE-8738 [task] Protect against missing query key in check status

Romain 5 tahun lalu
induk
melakukan
8740cc04cc
1 mengubah file dengan 4 tambahan dan 1 penghapusan
  1. 4 1
      desktop/libs/notebook/src/notebook/tasks.py

+ 4 - 1
desktop/libs/notebook/src/notebook/tasks.py

@@ -487,7 +487,10 @@ def _cleanup(notebook, snippet):
   caches[CACHES_CELERY_KEY].delete(_fetch_progress_key(notebook, snippet))
 
 def _get_query_key(notebook, snippet):
-  return snippet['executor']['executables'][0]['history']['uuid'] if ENABLE_NOTEBOOK_2.get() else notebook['uuid']
+  if ENABLE_NOTEBOOK_2.get():
+    return snippet['executor']['executables'][0].get('history', {}).get('uuid')
+  else:
+    return notebook['uuid']
 
 def _log_key(notebook, snippet):
   return _get_query_key(notebook, snippet) + '_log'