folio-migration-tools 1.9.9__py3-none-any.whl → 1.10.0__py3-none-any.whl

folio_migration_tools/migration_tasks/migration_task_base.py

@@ -47,15 +47,14 @@ class MigrationTaskBase:
         logging.info("MigrationTaskBase init")
         self.start_datetime = datetime.now(timezone.utc)
         self.task_configuration = task_configuration
-        logging.info(self.task_configuration.json(indent=4))
+        logging.info(self.task_configuration.model_dump_json(indent=4))
         self.folio_client: FolioClient = folio_client
         self.ecs_tenant_id = (
             task_configuration.ecs_tenant_id or library_configuration.ecs_tenant_id
         )
-        self.ecs_tenant_header = (
-            {"x-okapi-tenant": self.ecs_tenant_id} if self.ecs_tenant_id else {}
-        )
-        self.folio_client.okapi_headers.update(self.ecs_tenant_header)
+        if self.ecs_tenant_id:
+            self.folio_client.tenant_id = self.ecs_tenant_id
+
         self.central_folder_structure: Optional[FolderStructure] = None
         if library_configuration.is_ecs and library_configuration.ecs_central_iteration_identifier:
             self.central_folder_structure = FolderStructure(
@@ -100,6 +99,7 @@ class MigrationTaskBase:
         raise NotImplementedError()
 
     def clean_out_empty_logs(self):
+        _close_handler(self.data_issue_file_handler)
         if (
             self.folder_structure.data_issue_file_path.is_file()
             and os.stat(self.folder_structure.data_issue_file_path).st_size == 0
@@ -133,11 +133,7 @@ class MigrationTaskBase:
             TransformationProcessError: _description_
 
         """
-        files = [
-            source_path / f.file_name
-            for f in file_defs
-            if isfile(source_path / f.file_name)
-        ]
+        files = [source_path / f.file_name for f in file_defs if isfile(source_path / f.file_name)]
         ret_str = ", ".join(f.file_name for f in file_defs)
 
         if files and len(files) < len(file_defs):
@@ -162,12 +158,13 @@ class MigrationTaskBase:
         This is in the base class because multiple tasks need it. It exists because instances in an ECS environment
         are transformed for the central and data tenants separately, but the data tenants need to know about
         the central tenant instance ids. This is a bit of a hack, but it works for now.
-        """
+        """  # noqa: E501
         map_files = []
         instance_id_map = {}
         if self.library_configuration.is_ecs and self.central_folder_structure:
             logging.info(
-                "Loading ECS central tenant instance id map from %s", self.central_folder_structure.instance_id_map_path
+                "Loading ECS central tenant instance id map from %s",
+                self.central_folder_structure.instance_id_map_path,
             )
             instance_id_map = self.load_id_map(
                 self.central_folder_structure.instance_id_map_path,
@@ -176,7 +173,7 @@ class MigrationTaskBase:
             map_files.append(str(self.central_folder_structure.instance_id_map_path))
             logging.info(
                 "Loading member tenant isntance id map from %s",
-                self.folder_structure.instance_id_map_path
+                self.folder_structure.instance_id_map_path,
             )
         instance_id_map = self.load_id_map(
             self.folder_structure.instance_id_map_path,
@@ -190,13 +187,11 @@ class MigrationTaskBase:
         return instance_id_map
 
     @staticmethod
-    def load_id_map(map_path, raise_if_empty=False, existing_id_map={}):
+    def load_id_map(map_path, raise_if_empty=False, existing_id_map=None):
         if not isfile(map_path):
-            logging.warning(
-                "No legacy id map found at %s. Will build one from scratch", map_path
-            )
+            logging.warning("No legacy id map found at %s. Will build one from scratch", map_path)
             return {}
-        id_map = existing_id_map
+        id_map = existing_id_map or {}
         loaded_rows = len(id_map)
         with open(map_path) as id_map_file:
             for index, json_string in enumerate(id_map_file, start=1):
@@ -247,9 +242,7 @@ class MigrationTaskBase:
         else:
             logger.setLevel(logging.INFO)
             stream_handler.setLevel(logging.INFO)
-            stream_handler.addFilter(
-                ExcludeLevelFilter(30)
-            )  # Exclude warnings from pymarc
+            stream_handler.addFilter(ExcludeLevelFilter(30))  # Exclude warnings from pymarc
         stream_handler.setFormatter(formatter)
         logger.addHandler(stream_handler)
 
@@ -268,34 +261,28 @@ class MigrationTaskBase:
 
         # Data issue file formatter
         data_issue_file_formatter = logging.Formatter("%(message)s")
-        data_issue_file_handler = logging.FileHandler(
+        self.data_issue_file_handler = logging.FileHandler(
             filename=str(self.folder_structure.data_issue_file_path), mode="w"
         )
-        data_issue_file_handler.addFilter(LevelFilter(26))
-        data_issue_file_handler.setFormatter(data_issue_file_formatter)
-        data_issue_file_handler.setLevel(26)
-        logging.getLogger().addHandler(data_issue_file_handler)
+        self.data_issue_file_handler.addFilter(LevelFilter(26))
+        self.data_issue_file_handler.setFormatter(data_issue_file_formatter)
+        self.data_issue_file_handler.setLevel(26)
+        logging.getLogger().addHandler(self.data_issue_file_handler)
         logger.info("Logging set up")
 
     def setup_records_map(self, mapping_file_path):
         with open(mapping_file_path) as mapping_file:
             field_map = json.load(mapping_file)
-            logging.info(
-                "%s fields present in record mapping file", len(field_map["data"])
-            )
+            logging.info("%s fields present in record mapping file", len(field_map["data"]))
             mapped_fields = (
                 f
                 for f in field_map["data"]
                 if f["legacy_field"] and f["legacy_field"] != "Not mapped"
             )
-            logging.info(
-                "%s fields mapped in record mapping file", len(list(mapped_fields))
-            )
+            logging.info("%s fields mapped in record mapping file", len(list(mapped_fields)))
             return field_map
 
-    def log_and_exit_if_too_many_errors(
-        self, error: TransformationRecordFailedError, idx
-    ):
+    def log_and_exit_if_too_many_errors(self, error: TransformationRecordFailedError, idx):
         self.num_exeptions += 1
         error.log_it()
         if self.num_exeptions / (1 + idx) > 0.2 and self.num_exeptions > 5000:
@@ -311,9 +298,7 @@ class MigrationTaskBase:
         if num_processed > 1 and num_processed % 10000 == 0:
             elapsed = num_processed / (time.time() - start_time)
             elapsed_formatted = "{0:.4g}".format(elapsed)
-            logging.info(
-                f"{num_processed:,} records processed. Recs/sec: {elapsed_formatted} "
-            )
+            logging.info(f"{num_processed:,} records processed. Recs/sec: {elapsed_formatted} ")
 
     def do_work_marc_transformer(
         self,
@@ -322,9 +307,7 @@ class MigrationTaskBase:
         if self.folder_structure.failed_marc_recs_file.is_file():
             os.remove(self.folder_structure.failed_marc_recs_file)
             logging.info("Removed failed marc records file to prevent duplicating data")
-        with open(
-            self.folder_structure.created_objects_path, "w+"
-        ) as created_records_file:
+        with open(self.folder_structure.created_objects_path, "w+") as created_records_file:
             self.processor = MarcFileProcessor(
                 self.mapper, self.folder_structure, created_records_file
             )
@@ -377,7 +360,7 @@ class MigrationTaskBase:
 
         Returns:
             None
-        """
+        """  # noqa: E501
         current_pos = map_file.tell()
         try:
             map_file.seek(0)
@@ -391,13 +374,12 @@ class MigrationTaskBase:
                     "",
                     (
                         f"Mapping file {map_file.name} has rows with different number "
-                        f"of columns ({'Row' if len(invalid_lines) == 1 else 'Rows'} {', '.join(invalid_lines)})"
+                        f"of columns ({'Row' if len(invalid_lines) == 1 else 'Rows'} "
+                        f"{', '.join(invalid_lines)})"
                     ),
                 )
             if not valid_lines:
-                raise TransformationProcessError(
-                    "", f"Map has no rows: {map_file.name}"
-                )
+                raise TransformationProcessError("", f"Map has no rows: {map_file.name}")
         finally:
             map_file.seek(current_pos)
 
@@ -422,7 +404,8 @@ class MigrationTaskBase:
             or required
             or folio_property_name.startswith("statisticalCodeIds")
             or folio_property_name.startswith("locationMap")
-        ):
+            or folio_property_name.startswith("fundsMap")
+        ) and map_file_path.is_file():
             try:
                 with open(map_file_path) as map_file:
                     # Validate the structure of the mapping file
@@ -482,7 +465,7 @@ class MarcTaskConfigurationBase(task_configuration.AbstractTaskConfiguration):
         deactivate035_from001 (bool):
             Disables the default FOLIO functionality of moving the previous 001 field into a 035 field, prefixed with the value from 003.
             Default is False, meaning the functionality remains active.
-    """
+    """  # noqa: E501
 
     files: Annotated[
         List[library_configuration.FileDefinition],
@@ -496,8 +479,7 @@ class MarcTaskConfigurationBase(task_configuration.AbstractTaskConfiguration):
         Field(
             title="Create source records",
             description=(
-                "Controls whether or not to retain the MARC records in "
-                "Source Record Storage."
+                "Controls whether or not to retain the MARC records in Source Record Storage."
             ),
         ),
     ] = False
@@ -538,12 +520,14 @@ class MarcTaskConfigurationBase(task_configuration.AbstractTaskConfiguration):
             title="Statistical code mapping fields",
             description=(
                 "List of fields + subfields to be used for mapping statistical codes. "
-                "Subfields should be delimited by a \"$\" (eg. 907$a). Single repeating subfields "
-                "will be treated as unique values. Multiple subfields will be concatenated together with a space."
+                'Subfields should be delimited by a "$" (eg. 907$a). Single repeating subfields '
+                "will be treated as unique values. Multiple subfields will be concatenated "
+                "together with a space."
             ),
         ),
     ] = []
 
+
 class ExcludeLevelFilter(logging.Filter):
     def __init__(self, level):
         super().__init__()
@@ -570,3 +554,10 @@ class LevelFilter(logging.Filter):
 
     def filter(self, record):
         return record.levelno == self.level
+
+
+def _close_handler(handler: logging.Handler | None):
+    if handler is None:
+        return
+    handler.flush()
+    handler.close()

folio_migration_tools/migration_tasks/orders_transformer.py

@@ -4,7 +4,6 @@ import json
 import logging
 import sys
 import time
-from os.path import isfile
 from typing import List, Optional, Annotated
 from pydantic import Field
 
@@ -17,6 +16,7 @@ from folio_migration_tools.custom_exceptions import (
     TransformationRecordFailedError,
 )
 from folio_migration_tools.helper import Helper
+from folio_migration_tools.i18n_cache import i18n_t
 from folio_migration_tools.library_configuration import (
     FileDefinition,
     LibraryConfiguration,
@@ -84,50 +84,35 @@ class OrdersTransformer(MigrationTaskBase):
             Optional[str],
             Field(
                 title="Payment Status Map File Name",
-                description=(
-                    "File name for payment status mapping. "
-                    "By default is empty string."
-                ),
+                description=("File name for payment status mapping. By default is empty string."),
             ),
         ] = ""
         receipt_status_map_file_name: Annotated[
             Optional[str],
             Field(
                 title="Receipt Status Map File Name",
-                description=(
-                    "File name for receipt status mapping. "
-                    "By default is empty string."
-                ),
+                description=("File name for receipt status mapping. By default is empty string."),
             ),
         ] = ""
         workflow_status_map_file_name: Annotated[
             Optional[str],
             Field(
                 title="Workflow Status Map File Name",
-                description=(
-                    "File name for workflow status mapping. "
-                    "By default is empty string."
-                ),
+                description=("File name for workflow status mapping. By default is empty string."),
             ),
         ] = ""
         location_map_file_name: Annotated[
             Optional[str],
             Field(
                 title="Location Map File Name",
-                description=(
-                    "File name for location mapping. "
-                    "By default is empty string."
-                ),
+                description=("File name for location mapping. By default is empty string."),
             ),
         ] = ""
         funds_map_file_name: Annotated[
             Optional[str],
             Field(
                 title="Funds Map File Name",
-                description=(
-                    "File name for funds mapping. "
-                    "By default is empty string."
-                ),
+                description=("File name for funds mapping. By default is empty string."),
             ),
         ] = ""
         funds_expense_class_map_file_name: Annotated[
@@ -135,8 +120,7 @@ class OrdersTransformer(MigrationTaskBase):
             Field(
                 title="Funds Expense Class Map File Name",
                 description=(
-                    "File name for funds expense class mapping. "
-                    "By default is empty string."
+                    "File name for funds expense class mapping. By default is empty string."
                 ),
             ),
         ] = ""
@@ -218,7 +202,7 @@ class OrdersTransformer(MigrationTaskBase):
                 "fundsMap",
                 self.folder_structure.mapping_files_folder / self.task_config.funds_map_file_name,
                 self.folio_keys,
-                False,
+                True,
             ),
             self.load_ref_data_mapping_file(  # Todo: The property in the schema has no type
                 "fundsExpenseClassMap",
@@ -230,28 +214,28 @@ class OrdersTransformer(MigrationTaskBase):
         )
 
     def list_source_files(self):
-        files = [
-            self.folder_structure.data_folder / self.object_type_name / f.file_name
-            for f in self.task_config.files
-            if isfile(self.folder_structure.data_folder / self.object_type_name / f.file_name)
-        ]
-        if not any(files):
-            ret_str = ",".join(f.file_name for f in self.task_config.files)
-            raise TransformationProcessError(
-                f"Files {ret_str} not found in"
-                "{self.folder_structure.data_folder} / {self.object_type_name}"
-            )
+        files = []
+        for f in self.task_config.files:
+            file_path = self.folder_structure.data_folder / self.object_type_name / f.file_name
+
+            if not file_path.is_file():
+                print(f"\n\nERROR: File defined in task not found - {f.file_name}")
+                raise TransformationProcessError(
+                    f"\n\nERROR: File defined in task not found - {f.file_name}"
+                )
+            files.append(file_path)
         logging.info("Files to process:")
         for filename in files:
             logging.info("\t%s", filename)
         return files
 
     def process_single_file(self, filename):
-        with open(filename, encoding="utf-8-sig") as records_file, open(
-            self.folder_structure.created_objects_path, "w+"
-        ) as results_file:
+        with (
+            open(filename, encoding="utf-8-sig") as records_file,
+            open(self.folder_structure.created_objects_path, "w+") as results_file,
+        ):
             self.mapper.migration_report.add_general_statistics(
-                i18n.t("Number of files processed")
+                i18n_t("Number of files processed")
             )
             start = time.time()
             records_processed = 0
@@ -270,7 +254,7 @@ class OrdersTransformer(MigrationTaskBase):
                     self.mapper.perform_additional_mapping(legacy_id, folio_rec)
 
                     self.mapper.migration_report.add_general_statistics(
-                        i18n.t("TOTAL Purchase Order Lines created")
+                        i18n_t("TOTAL Purchase Order Lines created")
                     )
                     self.mapper.report_folio_mapping(folio_rec, self.mapper.composite_order_schema)
                     self.mapper.notes_mapper.map_notes(
@@ -333,7 +317,7 @@ class OrdersTransformer(MigrationTaskBase):
                 self.folder_structure.migration_reports_file,
             )
             self.mapper.migration_report.write_migration_report(
-                i18n.t("Pruchase Orders and Purchase Order Lines Transformation Report"),
+                i18n_t("Pruchase Orders and Purchase Order Lines Transformation Report"),
                 migration_report_file,
                 self.start_datetime,
             )

folio_migration_tools/migration_tasks/organization_transformer.py

@@ -50,27 +50,21 @@ class OrganizationTransformer(MigrationTaskBase):
             str,
             Field(
                 title="Migration task type",
-                description=(
-                    "The type of migration task you want to perform"
-                ),
+                description=("The type of migration task you want to perform"),
             ),
         ]
         files: Annotated[
             List[FileDefinition],
             Field(
                 title="Source files",
-                description=(
-                    "List of MARC21 files with holdings records"
-                ),
+                description=("List of MARC21 files with holdings records"),
             ),
         ]
         organization_map_path: Annotated[
             str,
             Field(
                 title="Organization map path",
-                description=(
-                    "Path to the organization map file"
-                ),
+                description=("Path to the organization map file"),
             ),
         ]
         organization_types_map_path: Annotated[
@@ -95,18 +89,14 @@ class OrganizationTransformer(MigrationTaskBase):
             Optional[str],
             Field(
                 title="Email categories map path",
-                description=(
-                    "Path to the email categories map file. By default is empty string"
-                ),
+                description=("Path to the email categories map file. By default is empty string"),
             ),
         ] = ""
         phone_categories_map_path: Annotated[
             Optional[str],
             Field(
                 title="Phone categories map path",
-                description=(
-                    "Path to the phone categories map file. By default is empty string"
-                ),
+                description=("Path to the phone categories map file. By default is empty string"),
             ),
         ] = ""
 
@@ -201,9 +191,10 @@ class OrganizationTransformer(MigrationTaskBase):
         return files
 
     def process_single_file(self, filename):
-        with open(filename, encoding="utf-8-sig") as records_file, open(
-            self.folder_structure.created_objects_path, "w+"
-        ) as results_file:
+        with (
+            open(filename, encoding="utf-8-sig") as records_file,
+            open(self.folder_structure.created_objects_path, "w+") as results_file,
+        ):
             self.mapper.migration_report.add_general_statistics(
                 i18n.t("Number of files processed")
             )

folio_migration_tools/migration_tasks/requests_migrator.py

@@ -13,6 +13,7 @@ from zoneinfo import ZoneInfo
 from folio_migration_tools.circulation_helper import CirculationHelper
 from folio_migration_tools.custom_dict import InsensitiveDictReader
 from folio_migration_tools.helper import Helper
+from folio_migration_tools.i18n_cache import i18n_t
 from folio_migration_tools.library_configuration import (
     FileDefinition,
     LibraryConfiguration,
@@ -32,7 +33,7 @@ class RequestsMigrator(MigrationTaskBase):
                 description=(
                     "Name of this migration task. The name is being used to call "
                     "the specific task, and to distinguish tasks of similar types"
-                )
+                ),
             ),
         ]
         migration_task_type: Annotated[
@@ -54,8 +55,7 @@ class RequestsMigrator(MigrationTaskBase):
             Field(
                 title="Starting row",
                 description=(
-                    "Row number to start processing data from. "
-                    "Optional, by default is first row"
+                    "Row number to start processing data from. Optional, by default is first row"
                 ),
             ),
         ] = 1
@@ -64,8 +64,7 @@ class RequestsMigrator(MigrationTaskBase):
             Field(
                 title="Item files",
                 description=(
-                    "List of files containing item data. "
-                    "Optional, by default is empty list"
+                    "List of files containing item data. Optional, by default is empty list"
                 ),
             ),
         ] = []
@@ -74,8 +73,7 @@ class RequestsMigrator(MigrationTaskBase):
             Field(
                 title="Patron files",
                 description=(
-                    "List of files containing patron data. "
-                    "Optional, by default is empty list"
+                    "List of files containing patron data. Optional, by default is empty list"
                 ),
             ),
         ] = []
@@ -88,7 +86,7 @@ class RequestsMigrator(MigrationTaskBase):
         self,
         task_configuration: TaskConfiguration,
         library_config: LibraryConfiguration,
-        folio_client
+        folio_client,
     ):
         csv.register_dialect("tsv", delimiter="\t")
         self.migration_report = MigrationReport()
@@ -135,8 +133,7 @@ class RequestsMigrator(MigrationTaskBase):
             )
         else:
             logging.info(
-                "No item or user files supplied. Not validating against"
-                "previously migrated objects"
+                "No item or user files supplied. Not validating againstpreviously migrated objects"
             )
             self.valid_legacy_requests = self.semi_valid_legacy_requests
 
@@ -151,7 +148,7 @@ class RequestsMigrator(MigrationTaskBase):
 
     def prepare_legacy_request(self, legacy_request: LegacyRequest):
         patron = self.circulation_helper.get_user_by_barcode(legacy_request.patron_barcode)
-        self.migration_report.add_general_statistics(i18n.t("Patron lookups performed"))
+        self.migration_report.add_general_statistics(i18n_t("Patron lookups performed"))
 
         if not patron:
             logging.error(f"No user with barcode {legacy_request.patron_barcode} found in FOLIO")
@@ -161,18 +158,18 @@ class RequestsMigrator(MigrationTaskBase):
                 f"{legacy_request.patron_barcode}",
             )
             self.migration_report.add_general_statistics(
-                i18n.t("No user with barcode found in FOLIO")
+                i18n_t("No user with barcode found in FOLIO")
             )
             self.failed_requests.add(legacy_request)
             return False, legacy_request
         legacy_request.patron_id = patron.get("id")
 
         item = self.circulation_helper.get_item_by_barcode(legacy_request.item_barcode)
-        self.migration_report.add_general_statistics(i18n.t("Item lookups performed"))
+        self.migration_report.add_general_statistics(i18n_t("Item lookups performed"))
         if not item:
             logging.error(f"No item with barcode {legacy_request.item_barcode} found in FOLIO")
             self.migration_report.add_general_statistics(
-                i18n.t("No item with barcode found in FOLIO")
+                i18n_t("No item with barcode found in FOLIO")
             )
             Helper.log_data_issue(
                 f"{legacy_request.item_barcode}",
@@ -182,22 +179,22 @@ class RequestsMigrator(MigrationTaskBase):
             self.failed_requests.add(legacy_request)
             return False, legacy_request
         holding = self.circulation_helper.get_holding_by_uuid(item.get("holdingsRecordId"))
-        self.migration_report.add_general_statistics(i18n.t("Holdings lookups performed"))
+        self.migration_report.add_general_statistics(i18n_t("Holdings lookups performed"))
         legacy_request.item_id = item.get("id")
         legacy_request.holdings_record_id = item.get("holdingsRecordId")
         legacy_request.instance_id = holding.get("instanceId")
         if item["status"]["name"] in ["Available"]:
             legacy_request.request_type = "Page"
-            logging.info(f'Setting request to Page, since the status is {item["status"]["name"]}')
+            logging.info(f"Setting request to Page, since the status is {item['status']['name']}")
         self.migration_report.add_general_statistics(
-            i18n.t("Valid, prepared requests, ready for posting")
+            i18n_t("Valid, prepared requests, ready for posting")
         )
         return True, legacy_request
 
     def do_work(self):
         logging.info("Starting")
         if self.task_configuration.starting_row > 1:
-            logging.info(f"Skipping {(self.task_configuration.starting_row-1)} records")
+            logging.info(f"Skipping {(self.task_configuration.starting_row - 1)} records")
         for num_requests, legacy_request in enumerate(
             self.valid_legacy_requests[self.task_configuration.starting_row - 1 :],
             start=1,
@@ -210,11 +207,11 @@ class RequestsMigrator(MigrationTaskBase):
                         self.folio_client, legacy_request, self.migration_report
                     ):
                         self.migration_report.add_general_statistics(
-                            i18n.t("Successfully migrated requests")
+                            i18n_t("Successfully migrated requests")
                         )
                     else:
                         self.migration_report.add_general_statistics(
-                            i18n.t("Unsuccessfully migrated requests")
+                            i18n_t("Unsuccessfully migrated requests")
                         )
                         self.failed_requests.add(legacy_request)
                 if num_requests == 1:
@@ -237,7 +234,7 @@ class RequestsMigrator(MigrationTaskBase):
 
         with open(self.folder_structure.migration_reports_file, "w+") as report_file:
             self.migration_report.write_migration_report(
-                i18n.t("Requests migration report"), report_file, self.start_datetime
+                i18n_t("Requests migration report"), report_file, self.start_datetime
             )
         self.clean_out_empty_logs()
 
@@ -281,7 +278,8 @@ class RequestsMigrator(MigrationTaskBase):
                 self.migration_report.add(
                     "DiscardedLoans",
                     i18n.t(
-                        "Requests discarded. Had migrated item barcode: %{item_barcode}.\n Had migrated user barcode: %{patron_barcode}",
+                        "Requests discarded. Had migrated item barcode: %{item_barcode}.\n "
+                        "Had migrated user barcode: %{patron_barcode}",
                         item_barcode=has_item_barcode,
                         patron_barcode=has_patron_barcode,
                     ),
@@ -333,8 +331,7 @@ class RequestsMigrator(MigrationTaskBase):
             except ValueError as ve:
                 logging.exception(ve)
         logging.info(
-            f"Done validating {legacy_reques_count} "
-            f"legacy requests with {num_bad} rotten apples"
+            f"Done validating {legacy_reques_count} legacy requests with {num_bad} rotten apples"
         )
         if num_bad > 0 and (num_bad / legacy_reques_count) > 0.5:
             q = num_bad / legacy_reques_count

folio_migration_tools/migration_tasks/reserves_migrator.py

@@ -14,6 +14,7 @@ from folio_uuid.folio_namespaces import FOLIONamespaces
 
 from folio_migration_tools.custom_dict import InsensitiveDictReader
 from folio_migration_tools.custom_exceptions import TransformationProcessError
+from folio_migration_tools.i18n_cache import i18n_t
 from folio_migration_tools.library_configuration import (
     FileDefinition,
     LibraryConfiguration,
@@ -59,7 +60,7 @@ class ReservesMigrator(MigrationTaskBase):
         self,
         task_configuration: TaskConfiguration,
         library_config: LibraryConfiguration,
-        folio_client
+        folio_client,
     ):
         csv.register_dialect("tsv", delimiter="\t")
         self.migration_report = MigrationReport()
@@ -90,7 +91,7 @@ class ReservesMigrator(MigrationTaskBase):
         logging.info("Starting")
         for num_reserves, legacy_reserve in enumerate(self.valid_reserves, start=1):
             t0_migration = time.time()
-            self.migration_report.add_general_statistics(i18n.t("Processed reserves"))
+            self.migration_report.add_general_statistics(i18n_t("Processed reserves"))
             try:
                 self.post_single_reserve(legacy_reserve)
             except Exception as ee:
@@ -107,10 +108,10 @@ class ReservesMigrator(MigrationTaskBase):
                 path, legacy_reserve.to_dict(), "POST", i18n.t("Posted reserves")
             ):
                 self.migration_report.add_general_statistics(
-                    i18n.t("Successfully posted reserves")
+                    i18n_t("Successfully posted reserves")
                 )
             else:
-                self.migration_report.add_general_statistics(i18n.t("Failure to post reserve"))
+                self.migration_report.add_general_statistics(i18n_t("Failure to post reserve"))
         except Exception as ee:
             logging.error(ee)
 
@@ -123,7 +124,7 @@ class ReservesMigrator(MigrationTaskBase):
 
         with open(self.folder_structure.migration_reports_file, "w+") as report_file:
             self.migration_report.write_migration_report(
-                i18n.t("Reserves migration report"), report_file, self.start_datetime
+                i18n_t("Reserves migration report"), report_file, self.start_datetime
             )
         self.clean_out_empty_logs()