folio-migration-tools 1.9.0rc4__tar.gz → 1.9.0rc5__tar.gz

{folio_migration_tools-1.9.0rc4 → folio_migration_tools-1.9.0rc5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: folio_migration_tools
-Version: 1.9.0rc4
+Version: 1.9.0rc5
 Summary: A tool allowing you to migrate data from legacy ILS:s (Library systems) into FOLIO LSP
 License: MIT
 Keywords: FOLIO,ILS,LSP,Library Systems,MARC21,Library data

{folio_migration_tools-1.9.0rc4 → folio_migration_tools-1.9.0rc5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "folio_migration_tools"
-version = "1.9.0rc4"
+version = "1.9.0rc5"
 description =  "A tool allowing you to migrate data from legacy ILS:s (Library systems) into FOLIO LSP"
 authors = [
 	{name = "Theodor Tolstoy", email = "github.teddes@tolstoy.se"},

{folio_migration_tools-1.9.0rc4 → folio_migration_tools-1.9.0rc5}/src/folio_migration_tools/migration_tasks/batch_poster.py

@@ -260,11 +260,15 @@ class BatchPoster(MigrationTaskBase):
         """
         Synchronous wrapper for set_version_async
         """
-        loop = asyncio.get_event_loop()
-        if loop.is_running():
+        try:
+            loop = asyncio.get_running_loop()
+        except RuntimeError:
+            loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(loop)
             loop.run_until_complete(self.set_version_async(batch, query_api, object_type))
+            asyncio.set_event_loop(None)  # Reset the event loop
         else:
-            asyncio.run(self.set_version_async(batch, query_api, object_type))
+            loop.run_until_complete(self.set_version_async(batch, query_api, object_type))
 
     async def set_version_async(self, batch, query_api, object_type) -> None:
         """
@@ -298,7 +302,7 @@ class BatchPoster(MigrationTaskBase):
 
             for response in responses:
                 if response.status_code == 200:
-                    response_json = await response.json()
+                    response_json = response.json()
                     for record in response_json[object_type]:
                         updates[record["id"]] = {
                             "_version": record["_version"],
@@ -325,8 +329,6 @@ class BatchPoster(MigrationTaskBase):
             logging.info(json.dumps(json_rec, indent=True))
         batch.append(json_rec)
         if len(batch) == int(self.batch_size):
-            if self.query_params.get("upsert", False) and self.api_info.get("query_endpoint", ""):
-                self.set_version(batch, self.api_info['query_endpoint'], self.api_info['object_name'])
             self.post_batch(batch, failed_recs_file, self.processed)
             batch = []
         return batch
@@ -451,6 +453,8 @@ class BatchPoster(MigrationTaskBase):
         logging.info("=======================", flush=True)
 
     def post_batch(self, batch, failed_recs_file, num_records, recursion_depth=0):
+        if self.query_params.get("upsert", False) and self.api_info.get("query_endpoint", ""):
+            self.set_version(batch, self.api_info['query_endpoint'], self.api_info['object_name'])
         response = self.do_post(batch)
         if response.status_code == 401:
             logging.error("Authorization failed (%s). Fetching new auth token...", response.text)