Fix small bug

akretion · Sep 15, 2023 · 0836797 · 0836797
1 parent 4ff88c0
commit 0836797
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/partner_import_helper/models/res_partner.py b/partner_import_helper/models/res_partner.py
@@ -157,8 +157,9 @@ def _import_prepare_vals(self, vals, speedy, email_check_deliverability=True, cr
         country_id = country_code = False
         if vals.get('country_name') and isinstance(vals['country_name'], str) and not vals.get('country_id'):
             country_id = self._match_country(vals, speedy)
+            # Warning: country_id can be False
             vals['country_id'] = country_id
-            country_code = speedy['country']['id2code'][country_id]
+            country_code = speedy['country']['id2code'].get(country_id)
         # TITLE
         if not vals.get('is_company') and vals.get('title_code') and isinstance(vals['title_code'], str) and not vals.get('title'):
             title_id = self._match_title(vals, speedy)