fix street_and_number and apartment of volgistics

Author: urirot

src/server/datasource_manager.py

@@ -71,22 +71,23 @@ def __clean_csv_headers(header):
 }
 
 
-def volgistics_address(index, street):
+def volgistics_address(street, index):
     result = ""
 
-    for item in street:
-        if isinstance(item, str):
-            if " " in item:
-                result = item.split()[index]
+    if isinstance(street, str):
+        if " " in street:
+            result = street.split()[index]
 
     return result
 
+
 def normalize_phone_number(number):
     if str(number) == 'nan':
         return ""
     parsed_number = phonenumbers.parse(number, "US")
     return phonenumbers.format_number(parsed_number, phonenumbers.PhoneNumberFormat.NATIONAL)
 
+
 SOURCE_NORMALIZATION_MAPPING = {
     "salesforcecontacts": {
         "source_id": "contact_id",
@@ -131,8 +132,8 @@ def normalize_phone_number(number):
         "last_name": "last_name",
         "email": "email",
         "mobile": lambda df: df["cell"].combine_first(df["home"]).apply(normalize_phone_number),
-        "street_and_number": lambda df: volgistics_address(1, df["street_1"]),
-        "apartment": lambda df: volgistics_address(0, df["street_1"]),
+        "street_and_number": lambda df: df["street_1"].apply(volgistics_address, index=1),
+        "apartment": lambda df: df["street_1"].apply(volgistics_address, index=0),
         "city": "city",
         "state": "state",
         "zip": "zip",