Fix GB country type

eea · Feb 15, 2021 · bc5a3f2 · bc5a3f2
1 parent 5048668
commit bc5a3f2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/notifications/__init__.py b/notifications/__init__.py
@@ -1,4 +1,4 @@
-FGASES_EU = 'EU_TYPE'
+FGASES_EU = ['EU_TYPE', 'AMBIGUOUS_TYPE']
 FGASES_NONEU = 'NONEU_TYPE'
 
 FGASES_EU_GROUP_CODE = 'f-gases-eu'

diff --git a/notifications/management/commands/fetch_ecr.py b/notifications/management/commands/fetch_ecr.py
@@ -40,7 +40,7 @@ def set_all_persons_to_current_false(self):
     def get_group(self, company):
         if company['domain'] == 'ODS':
             return self.group_ods
-        elif company['address']['country']['type'] == FGASES_EU:
+        elif company['address']['country']['type'] in FGASES_EU:
             return self.group_eu
         elif company['address']['country']['type'] == FGASES_NONEU:
             return self.group_noneu

diff --git a/notifications/views/actions.py b/notifications/views/actions.py
@@ -126,7 +126,7 @@ def get(self, request, *args, **kwargs):
         counter_companies = 0
         errors_companies = []
         for item in registry.get_companies():
-            if item['address']['country']['type'] == FGASES_EU:
+            if item['address']['country']['type'] in FGASES_EU:
                 group = group_eu
             elif item['address']['country']['type'] == FGASES_NONEU:
                 group = group_noneu