email validation

introduck/routes.py

@@ -5,6 +5,8 @@ from fastapi import FastAPI
 from gradio.routes import App as GradioApp
 from introduck.inference import extract_contacts_from_text
 from introduck.utils import dump_email_to_string
+from introduck.utils import validate_email
+from introduck.utils import validate_multiple_emails
 
 _INTRO_SUBJECT_EXAMPLE: str = "Could you make an intro?"
 _INTRO_MESSAGE_EXAMPLE: str = """\
@@ -39,7 +41,17 @@ def _analyze_message(
 ) -> (pd.DataFrame, dict):
     default_outputs = None, None
 
-    # TODO: validate sender and recipients
+    # validate sender email (if present)
+    if sender:
+        sender = validate_email(email=sender)
+        if not sender:
+            return default_outputs
+
+    # validate recipient emails (if present)
+    if recipients:
+        recipients = validate_multiple_emails(emails=recipients)
+        if not recipients:
+            return default_outputs
 
     if not subject:
         return default_outputs
@@ -77,7 +89,17 @@ def _validate_message(
 ) -> dict:
     errors: list[str] = []
 
-    # ...
+    # validate sender email (if present)
+    if sender:
+        sender = validate_email(email=sender)
+        if not sender:
+            errors.append("- sender email is not valid;")
+
+    # validate recipient emails (if present)
+    if recipients:
+        recipients = validate_multiple_emails(emails=recipients)
+        if not recipients:
+            errors.append("- one or more recipient emails are not valid;")
 
     if not subject:
         errors.append("- subject can't be empty;")

introduck/utils.py

@@ -1,3 +1,7 @@
+from email_validator import EmailNotValidError
+from email_validator import validate_email as validate_email_address
+
+
 def dump_email_to_string(data: dict) -> str:
     msg_from: str = data.get("from", "") or "*"
     msg_to: str = data.get("to", "") or "*"
@@ -13,6 +17,33 @@ def dump_email_to_string(data: dict) -> str:
     return msg
 
 
-def validate_email_address(data: str) -> bool:
-    # TODO: https://pypi.org/project/email-validator/
-    return False
+def validate_email(email: str) -> str:
+    if not email:
+        return ""
+
+    try:
+        validation = validate_email_address(email, check_deliverability=False)
+
+        # Take the normalized form of the email address for all logic beyond
+        # this point (especially before going to a database query where equality
+        # may not take into account Unicode normalization).
+        email = validation.email
+    except EmailNotValidError as e:
+        # Email is not valid. The exception message is human-readable.
+        email = ""
+        print(str(e))
+
+    return email
+
+
+def validate_multiple_emails(emails: str) -> str:
+    if not emails:
+        return ""
+
+    for email in emails.split(sep=","):
+        email = validate_email(email=email.strip(" "))
+        if not email:
+            emails = ""
+            break
+
+    return emails

requirements.txt

@@ -1,3 +1,4 @@
+email-validator==1.3.0
 fastapi[all]
 gradio==3.4.0  # keep in sync with version from README.md metadata
 pandas==1.5.0