folio-migration-tools 1.9.0rc4__py3-none-any.whl → 1.9.0rc5__py3-none-any.whl

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)

{folio_migration_tools-1.9.0rc4.dist-info → folio_migration_tools-1.9.0rc5.dist-info}/METADATA

@@ -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.dist-info → folio_migration_tools-1.9.0rc5.dist-info}/RECORD

@@ -37,7 +37,7 @@ folio_migration_tools/marc_rules_transformation/rules_mapper_holdings.py,sha256=
 folio_migration_tools/migration_report.py,sha256=BkRspM1hwTBnWeqsHamf7yVEofzLj560Q-9G--O00hw,4258
 folio_migration_tools/migration_tasks/__init__.py,sha256=ZkbY_yGyB84Ke8OMlYUzyyBj4cxxNrhMTwQlu_GbdDs,211
 folio_migration_tools/migration_tasks/authority_transformer.py,sha256=AoXg9s-GLO3yEEDCrQV7hc4YVXxwxsdxDdpj1zhHydE,4251
-folio_migration_tools/migration_tasks/batch_poster.py,sha256=xwrjx8Lh--B1tGUcIzzvDi6r_5g3up6rq-6-Z2DY6YQ,35285
+folio_migration_tools/migration_tasks/batch_poster.py,sha256=8Oksp7uGjr6eAq-aqSpelQDTum14V9VzYGa_ZfLJQCM,35451
 folio_migration_tools/migration_tasks/bibs_transformer.py,sha256=XzlPo-0uuugJA4SM80xOlOj5nDK6OMDXFnAYg80hOBc,7791
 folio_migration_tools/migration_tasks/courses_migrator.py,sha256=CzXnsu-KGP7B4zcINJzLYUqz47D16NuFfzu_DPqRlTQ,7061
 folio_migration_tools/migration_tasks/holdings_csv_transformer.py,sha256=WT-RlDRm2ILr2-2shfG3TZ3nlSfqxEXT3TklZSqtJCM,22311
@@ -60,8 +60,8 @@ folio_migration_tools/transaction_migration/legacy_request.py,sha256=1ulyFzPQw_I
 folio_migration_tools/transaction_migration/legacy_reserve.py,sha256=d0qbh2fWpwlVSYRL6wZyZG20__NAYNxh7sPSsB-LAes,1804
 folio_migration_tools/transaction_migration/transaction_result.py,sha256=cTdCN0BnlI9_ZJB2Z3Fdkl9gpymIi-9mGZsRFlQcmDk,656
 folio_migration_tools/translations/en.json,sha256=HOVpkb_T-SN_x0NpDp8gyvV1hMLCui3SsG7ByyIv0OU,38669
-folio_migration_tools-1.9.0rc4.dist-info/LICENSE,sha256=PhIEkitVi3ejgq56tt6sWoJIG_zmv82cjjd_aYPPGdI,1072
-folio_migration_tools-1.9.0rc4.dist-info/METADATA,sha256=zh8mko180EbmuZjoSJ9ElvXWmQQVg0TW_NGgjRzS_Qc,7415
-folio_migration_tools-1.9.0rc4.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-folio_migration_tools-1.9.0rc4.dist-info/entry_points.txt,sha256=Hbe-HjqMcU8FwVshVIkeWyZd9XwgT1CCMNf06EpHQu8,77
-folio_migration_tools-1.9.0rc4.dist-info/RECORD,,
+folio_migration_tools-1.9.0rc5.dist-info/LICENSE,sha256=PhIEkitVi3ejgq56tt6sWoJIG_zmv82cjjd_aYPPGdI,1072
+folio_migration_tools-1.9.0rc5.dist-info/METADATA,sha256=6MxKZNM3wUsGNheI_nzS9YpJjR2XqDTigIedpeAcvzA,7415
+folio_migration_tools-1.9.0rc5.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+folio_migration_tools-1.9.0rc5.dist-info/entry_points.txt,sha256=Hbe-HjqMcU8FwVshVIkeWyZd9XwgT1CCMNf06EpHQu8,77
+folio_migration_tools-1.9.0rc5.dist-info/RECORD,,