[enrich] Fix empty value of 'author_domain' field

This code fixes the empty value of `author_domain` field on the `refreshing identities` process and sets it to `None`. Signed-off-by: Quan Zhou <[email protected]>
chaoss · Jan 26, 2022 · a8f3b29 · a8f3b29
1 parent fda45e9
commit a8f3b29
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/grimoire_elk/enriched/enrich.py b/grimoire_elk/enriched/enrich.py
@@ -447,7 +447,7 @@ def get_email_domain(self, email):
         domain = None
         try:
             domain = email.split("@")[1]
-        except IndexError:
+        except (IndexError, AttributeError):
             # logger.warning("Bad email format: %s" % (identity['email']))
             pass
         return domain
@@ -801,8 +801,7 @@ def get_item_sh_fields(self, identity=None, item_date=None, sh_id=None,
             eitem_sh[rol + "_name"] = profile.get('name', eitem_sh[rol + "_name"])
 
             email = profile.get('email', None)
-            if email:
-                eitem_sh[rol + "_domain"] = self.get_email_domain(email)
+            eitem_sh[rol + "_domain"] = self.get_email_domain(email)
 
             eitem_sh[rol + "_gender"] = profile.get('gender', self.unknown_gender)
             eitem_sh[rol + "_gender_acc"] = profile.get('gender_acc', 0)