Improve resilience.

Antoine Goutenoir
1 parent e144bb1b
Showing 2 changed files with 7 additions and 4 deletions Show diff stats
flaskr/controllers/main_controller.py
flaskr/models.py
@@ -603,8 +603,9 @@ def consult_estimation(public_id, extension):
         else:
             estimation_output = estimation.get_output_dict()
             estimation_sum = 0
-            for city in estimation_output['cities']:
-                estimation_sum += city['footprint']
+            if estimation_output:
+                for city in estimation_output['cities']:
+                    estimation_sum += city['footprint']
  
             return render_template(
                 "estimation.html",
@@ -71,9 +71,11 @@ class Estimation(db.Model):
  
     def get_output_dict(self):
         if self._output_dict is None:
-            self._output_dict = yaml_load(self.output_yaml)
+            if self._output_yaml is None:
+                self._output_dict = None
+            else:
+                self._output_dict = yaml_load(self.output_yaml)
             return self._output_dict
-        pass
  
     def is_one_to_one(self):
         return self.scenario == ScenarioEnum.one_to_one