folio-data-import 0.2.2__tar.gz → 0.2.4__tar.gz

{folio_data_import-0.2.2 → folio_data_import-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: folio_data_import
-Version: 0.2.2
+Version: 0.2.4
 Summary: A python module to interact with the data importing capabilities of the open-source FOLIO ILS
 License: MIT
 Author: Brooks Travis

{folio_data_import-0.2.2 → folio_data_import-0.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "folio_data_import"
-version = "0.2.2"
+version = "0.2.4"
 description = "A python module to interact with the data importing capabilities of the open-source FOLIO ILS"
 authors = ["Brooks Travis <brooks.travis@gmail.com>"]
 license = "MIT"

{folio_data_import-0.2.2 → folio_data_import-0.2.4}/src/folio_data_import/UserImport.py

@@ -181,12 +181,15 @@ class UserImporter:  # noqa: R0902
             KeyError: If an address type name in the user object is not found in address_type_map.
 
         """
-        if "personal" in user_obj and "addresses" in user_obj["personal"]:
-            for address in user_obj["personal"]["addresses"]:
+        if "personal" in user_obj:
+            addresses = user_obj["personal"].pop("addresses", [])
+            mapped_addresses = []
+            for address in addresses:
                 try:
                     address["addressTypeId"] = self.address_type_map[
                         address["addressTypeId"]
                     ]
+                    mapped_addresses.append(address)
                 except KeyError:
                     if address["addressTypeId"] not in self.address_type_map.values():
                         print(
@@ -197,9 +200,8 @@ class UserImporter:  # noqa: R0902
                             f"Row {line_number}: Address type {address['addressTypeId']} not found"
                             f", removing address\n"
                         )
-                        del address
-            if len(user_obj["personal"]["addresses"]) == 0:
-                del user_obj["personal"]["addresses"]
+            if mapped_addresses:
+                user_obj["personal"]["addresses"] = mapped_addresses
 
     async def map_patron_groups(self, user_obj, line_number) -> None:
         """
@@ -594,7 +596,9 @@ class UserImporter:  # noqa: R0902
                         await self.logfile.write(message + "\n")
                     tasks = []
             if tasks:
+                start = time.time()
                 await asyncio.gather(*tasks)
+                duration = time.time() - start
                 async with self.lock:
                     message = (
                         f"{dt.now().isoformat(sep=' ', timespec='milliseconds')}: "