Explorar o código

[desktop] Protect against get_default_mrcluster() returning None

Erick Tryzelaar %!s(int64=10) %!d(string=hai) anos
pai
achega
9376f14
Modificáronse 1 ficheiros con 5 adicións e 1 borrados
  1. 5 1
      desktop/libs/hadoop/src/hadoop/cluster.py

+ 5 - 1
desktop/libs/hadoop/src/hadoop/cluster.py

@@ -117,7 +117,11 @@ def get_next_ha_mrcluster():
   candidates = all_mrclusters()
   has_ha = sum([conf.MR_CLUSTERS[name].SUBMIT_TO.get() for name in conf.MR_CLUSTERS.keys()]) >= 2
 
-  current_user = get_default_mrcluster().user
+  mrcluster = get_default_mrcluster()
+  if mrcluster is None:
+    return None
+
+  current_user = mrcluster.user
 
   for name in conf.MR_CLUSTERS.keys():
     config = conf.MR_CLUSTERS[name]