From 1d41e9fbc1e1f8f67b54cd80361ffebf8ab0f03a Mon Sep 17 00:00:00 2001
From: Austin Weisgrau <austinweisgrau@gmail.com>
Date: Wed, 20 Dec 2023 13:51:36 -0800
Subject: [PATCH] Enable email dict to be passed to VAN.upsert_person()

This allows for greater configurability of email data when upserting
person records in EveryAction. This change is backwards compatible,
but allows, for example, setting emails to not be subscribed by
default when loading into EveryAction.
---
 parsons/ngpvan/people.py | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/parsons/ngpvan/people.py b/parsons/ngpvan/people.py
index 34478570f9..f530cca17b 100644
--- a/parsons/ngpvan/people.py
+++ b/parsons/ngpvan/people.py
@@ -1,4 +1,5 @@
 from parsons.utilities import json_format
+from typing import Union, List, Dict
 import logging
 
 logger = logging.getLogger(__name__)
@@ -197,7 +198,7 @@ def upsert_person(
         first_name=None,
         last_name=None,
         date_of_birth=None,
-        email=None,
+        email: Union[str, List[Dict[str, Union[str, bool]]], None] = None,
         phone=None,
         phone_type=None,
         street_number=None,
@@ -227,8 +228,10 @@ def upsert_person(
                 The person's last name
             dob: str
                 ISO 8601 formatted date of birth (e.g. ``1981-02-01``)
-            email: str
-                The person's email address
+            email: Union[str, List[Dict[str, Union[str, bool]]], None]
+                The person's email address or a list of email dicts.
+                e.g. [{'email': 'abcd@gmail.com', 'isSubscribed': False}]
+                See https://docs.everyaction.com/reference/people-common-models#email
             phone: str
                 Phone number of any type (Work, Cell, Home)
             phone_type: str
@@ -298,7 +301,7 @@ def _people_search(
         first_name=None,
         last_name=None,
         date_of_birth=None,
-        email=None,
+        email: Union[str, List[Dict[str, Union[str, bool]]], None] = None,
         phone=None,
         phone_type="H",
         street_number=None,
@@ -320,7 +323,14 @@ def _people_search(
 
             # Will fail if empty dicts are provided, hence needed to add if exist
             if email:
-                json["emails"] = [{"email": email}]
+                if isinstance(email, str):
+                    json["emails"] = [{"email": email}]
+                elif isinstance(email, list):
+                    json["emails"] = email
+                else:
+                    raise ValueError(
+                        f"Unexpected data type for email argument: {type(email)}"
+                    )
             if phone:  # To Do: Strip out non-integers from phone
                 json["phones"] = [{"phoneNumber": phone, "phoneType": phone_type}]
             if date_of_birth: