rome_mobilities quick dirty fix

StartupsPoleEmploi · Jan 25, 2018 · 16a70a4 · 16a70a4
1 parent cbf19a7
commit 16a70a4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/labonneboite/common/search.py b/labonneboite/common/search.py
@@ -276,7 +276,10 @@ def get_companies(self, add_suggestions=False):
         if self.company_count < 10 and add_suggestions:
 
             # Suggest other jobs.
-            alternative_rome_codes = ROME_MOBILITIES[self.rome]
+            try:
+                alternative_rome_codes = ROME_MOBILITIES[self.rome]
+            except KeyError:  # FIXME regenerate rome_mobilities dataset after rome slicing
+                alternative_rome_codes = []
             for rome in alternative_rome_codes:
                 if not rome == self.rome:
                     company_count = self._get_company_count(rome, self.distance)