folio-migration-tools 1.9.0a2__py3-none-any.whl → 1.9.0a4__py3-none-any.whl

folio_migration_tools/migration_tasks/batch_poster.py

@@ -5,8 +5,7 @@ import sys
 import time
 import traceback
 from datetime import datetime
-from typing import Annotated
-from typing import List
+from typing import Annotated, List
 from uuid import uuid4
 
 import httpx
@@ -14,10 +13,14 @@ import i18n
 from folio_uuid.folio_namespaces import FOLIONamespaces
 from pydantic import Field
 
-from folio_migration_tools.custom_exceptions import TransformationProcessError
-from folio_migration_tools.custom_exceptions import TransformationRecordFailedError
-from folio_migration_tools.library_configuration import FileDefinition
-from folio_migration_tools.library_configuration import LibraryConfiguration
+from folio_migration_tools.custom_exceptions import (
+    TransformationProcessError,
+    TransformationRecordFailedError,
+)
+from folio_migration_tools.library_configuration import (
+    FileDefinition,
+    LibraryConfiguration,
+)
 from folio_migration_tools.migration_report import MigrationReport
 from folio_migration_tools.migration_tasks.migration_task_base import MigrationTaskBase
 from folio_migration_tools.task_configuration import AbstractTaskConfiguration
@@ -71,6 +74,24 @@ class BatchPoster(MigrationTaskBase):
                 )
             ),
         ] = True
+        extradata_endpoints: Annotated[
+            dict,
+            Field(
+                description=(
+                    "A dictionary of extradata endpoints. "
+                    "The key is the object type and the value is the endpoint"
+                )
+            ),
+        ] = {}
+        upsert: Annotated[
+            bool,
+            Field(
+                description=(
+                    "Toggles whether or not to use the upsert feature of the Inventory storage "
+                    "endpoints. Defaults to False"
+                )
+            ),
+        ] = False
 
     @staticmethod
     def get_object_type() -> FOLIONamespaces:
@@ -80,9 +101,10 @@ class BatchPoster(MigrationTaskBase):
         self,
         task_config: TaskConfiguration,
         library_config: LibraryConfiguration,
+        folio_client,
         use_logging: bool = True,
     ):
-        super().__init__(library_config, task_config, use_logging)
+        super().__init__(library_config, task_config, folio_client, use_logging)
         self.migration_report = MigrationReport()
         self.performing_rerun = False
         self.failed_ids: list = []
@@ -91,6 +113,11 @@ class BatchPoster(MigrationTaskBase):
             self.task_configuration.object_type,
             self.task_configuration.use_safe_inventory_endpoints,
         )
+        self.query_params = {}
+        if self.api_info["supports_upsert"]:
+            self.query_params["upsert"] = self.task_configuration.upsert
+        elif self.task_configuration.upsert and not self.api_info["supports_upsert"]:
+            logging.info("Upsert is not supported for this object type. Query parameter will not be set.")
         self.snapshot_id = str(uuid4())
         self.failed_objects: list = []
         self.batch_size = self.task_configuration.batch_size
@@ -107,7 +134,7 @@ class BatchPoster(MigrationTaskBase):
         self.http_client = None
 
     def do_work(self):
-        with httpx.Client(timeout=None) as httpx_client:
+        with self.folio_client.get_folio_http_client() as httpx_client:
             self.http_client = httpx_client
             try:
                 batch = []
@@ -193,7 +220,7 @@ class BatchPoster(MigrationTaskBase):
 
     def post_extra_data(self, row: str, num_records: int, failed_recs_file):
         (object_name, data) = row.split("\t")
-        endpoint = get_extradata_endpoint(object_name, data)
+        endpoint = self.get_extradata_endpoint(self.task_configuration, object_name, data)
         url = f"{self.folio_client.okapi_url}/{endpoint}"
         body = data
         response = self.post_objects(url, body)
@@ -216,6 +243,35 @@ class BatchPoster(MigrationTaskBase):
                 self.num_failures,
             )
 
+    @staticmethod
+    def get_extradata_endpoint(
+        task_configuration: TaskConfiguration, object_name: str, string_object: str
+    ):
+        object_types = {
+            "precedingSucceedingTitles": "preceding-succeeding-titles",
+            "precedingTitles": "preceding-succeeding-titles",
+            "succeedingTitles": "preceding-succeeding-titles",
+            "boundwithPart": "inventory-storage/bound-with-parts",
+            "notes": "notes",
+            "course": "coursereserves/courses",
+            "courselisting": "coursereserves/courselistings",
+            "contacts": "organizations-storage/contacts",
+            "interfaces": "organizations-storage/interfaces",
+            "account": "accounts",
+            "feefineaction": "feefineactions",
+            "bankInfo": "organizations/banking-information",
+        }
+        object_types.update(task_configuration.extradata_endpoints)
+        if object_name == "instructor":
+            instructor = json.loads(string_object)
+            return f'coursereserves/courselistings/{instructor["courseListingId"]}/instructors'
+
+        if object_name == "interfaceCredential":
+            credential = json.loads(string_object)
+            return f'organizations-storage/interfaces/{credential["interfaceId"]}/credentials'
+
+        return object_types[object_name]
+
     def post_single_records(self, row: str, num_records: int, failed_recs_file):
         if self.api_info["is_batch"]:
             raise TypeError("This record type supports batch processing, use post_batch method")
@@ -345,7 +401,7 @@ class BatchPoster(MigrationTaskBase):
         elif response.status_code == 400:
             # Likely a json parsing error
             logging.error(response.text)
-            raise TransformationProcessError("", "HTTP 400. Somehting is wrong. Quitting")
+            raise TransformationProcessError("", "HTTP 400. Something is wrong. Quitting")
         elif self.task_configuration.object_type == "SRS" and response.status_code >= 500:
             logging.info(
                 "Post failed. Size: %s Waiting 30s until reposting. Number of tries: %s of 5",
@@ -398,10 +454,10 @@ class BatchPoster(MigrationTaskBase):
             payload = {self.api_info["object_name"]: batch}
         if self.http_client and not self.http_client.is_closed:
             return self.http_client.post(
-                url, json=payload, headers=self.folio_client.okapi_headers
+                url, json=payload, headers=self.folio_client.okapi_headers, params=self.query_params
             )
         else:
-            return httpx.post(url, headers=self.okapi_headers, json=payload, timeout=None)
+            return httpx.post(url, headers=self.okapi_headers, json=payload, params=self.query_params, timeout=None)
 
     def wrap_up(self):
         logging.info("Done. Wrapping up")
@@ -448,7 +504,7 @@ class BatchPoster(MigrationTaskBase):
                 temp_report = copy.deepcopy(self.migration_report)
                 temp_start = self.start_datetime
                 self.task_configuration.rerun_failed_records = False
-                self.__init__(self.task_configuration, self.library_configuration)
+                self.__init__(self.task_configuration, self.library_configuration, self.folio_client)
                 self.performing_rerun = True
                 self.migration_report = temp_report
                 self.start_datetime = temp_start
@@ -529,6 +585,7 @@ def get_api_info(object_type: str, use_safe: bool = True):
             "api_endpoint": "",
             "total_records": False,
             "addSnapshotId": False,
+            "supports_upsert": False,
         },
         "Items": {
             "object_name": "items",
@@ -540,6 +597,7 @@ def get_api_info(object_type: str, use_safe: bool = True):
             "is_batch": True,
             "total_records": False,
             "addSnapshotId": False,
+            "supports_upsert": True,
         },
         "Holdings": {
             "object_name": "holdingsRecords",
@@ -551,6 +609,7 @@ def get_api_info(object_type: str, use_safe: bool = True):
             "is_batch": True,
             "total_records": False,
             "addSnapshotId": False,
+            "supports_upsert": True,
         },
         "Instances": {
             "object_name": "instances",
@@ -562,6 +621,7 @@ def get_api_info(object_type: str, use_safe: bool = True):
             "is_batch": True,
             "total_records": False,
             "addSnapshotId": False,
+            "supports_upsert": True,
         },
         "Authorities": {
             "object_name": "",
@@ -569,6 +629,7 @@ def get_api_info(object_type: str, use_safe: bool = True):
             "is_batch": False,
             "total_records": False,
             "addSnapshotId": False,
+            "supports_upsert": False,
         },
         "SRS": {
             "object_name": "records",
@@ -576,6 +637,7 @@ def get_api_info(object_type: str, use_safe: bool = True):
             "is_batch": True,
             "total_records": True,
             "addSnapshotId": True,
+            "supports_upsert": False,
         },
         "Users": {
             "object_name": "users",
@@ -583,6 +645,7 @@ def get_api_info(object_type: str, use_safe: bool = True):
             "is_batch": True,
             "total_records": True,
             "addSnapshotId": False,
+            "supports_upsert": False,
         },
         "Organizations": {
             "object_name": "",
@@ -590,6 +653,7 @@ def get_api_info(object_type: str, use_safe: bool = True):
             "is_batch": False,
             "total_records": False,
             "addSnapshotId": False,
+            "supports_upsert": False,
         },
         "Orders": {
             "object_name": "",
@@ -597,6 +661,7 @@ def get_api_info(object_type: str, use_safe: bool = True):
             "is_batch": False,
             "total_records": False,
             "addSnapshotId": False,
+            "supports_upsert": False,
         },
     }
 
@@ -623,31 +688,6 @@ def chunks(records, number_of_chunks):
         yield records[i : i + number_of_chunks]
 
 
-def get_extradata_endpoint(object_name: str, string_object: str):
-    object_types = {
-        "precedingSucceedingTitles": "preceding-succeeding-titles",
-        "precedingTitles": "preceding-succeeding-titles",
-        "succeedingTitles": "preceding-succeeding-titles",
-        "boundwithPart": "inventory-storage/bound-with-parts",
-        "notes": "notes",
-        "course": "coursereserves/courses",
-        "courselisting": "coursereserves/courselistings",
-        "contacts": "organizations-storage/contacts",
-        "interfaces": "organizations-storage/interfaces",
-        "account": "accounts",
-        "feefineaction": "feefineactions",
-    }
-    if object_name == "instructor":
-        instructor = json.loads(string_object)
-        return f'coursereserves/courselistings/{instructor["courseListingId"]}/instructors'
-
-    if object_name == "interfaceCredential":
-        credential = json.loads(string_object)
-        return f'organizations-storage/interfaces/{credential["interfaceId"]}/credentials'
-
-    return object_types[object_name]
-
-
 def get_human_readable(size, precision=2):
     suffixes = ["B", "KB", "MB", "GB", "TB"]
     suffix_index = 0

folio_migration_tools/migration_tasks/bibs_transformer.py

@@ -1,16 +1,17 @@
 import logging
-from typing import Annotated
-from typing import List
-import i18n
+from typing import Annotated, List
 
+import i18n
 from folio_uuid.folio_namespaces import FOLIONamespaces
 from pydantic import Field
 
 from folio_migration_tools.helper import Helper
-from folio_migration_tools.library_configuration import FileDefinition
-from folio_migration_tools.library_configuration import HridHandling
-from folio_migration_tools.library_configuration import IlsFlavour
-from folio_migration_tools.library_configuration import LibraryConfiguration
+from folio_migration_tools.library_configuration import (
+    FileDefinition,
+    HridHandling,
+    IlsFlavour,
+    LibraryConfiguration,
+)
 from folio_migration_tools.marc_rules_transformation.marc_file_processor import (
     MarcFileProcessor,
 )
@@ -96,6 +97,17 @@ class BibsTransformer(MigrationTaskBase):
                 ),
             ),
         ] = True
+        data_import_marc: Annotated[
+            bool,
+            Field(
+                title="Generate a MARC file for data import overlay of instances",
+                description=(
+                    "If set to true, the process will generate a file of binary MARC records that can"
+                    "be imported into FOLIO using the Data Import APIs. If set to false, only a file"
+                    "of FOLIO instance records (and optional SRS records) will be generated."
+                ),
+            )
+        ] = False
         parse_cataloged_date: Annotated[
             bool,
             Field(
@@ -153,9 +165,10 @@ class BibsTransformer(MigrationTaskBase):
         self,
         task_config: TaskConfiguration,
         library_config: LibraryConfiguration,
+        folio_client,
         use_logging: bool = True,
     ):
-        super().__init__(library_config, task_config, use_logging)
+        super().__init__(library_config, task_config, folio_client, use_logging)
         self.processor: MarcFileProcessor
         self.check_source_files(
             self.folder_structure.legacy_records_folder, self.task_configuration.files

folio_migration_tools/migration_tasks/courses_migrator.py

@@ -4,15 +4,19 @@ import logging
 import sys
 import time
 import traceback
-import i18n
 from typing import Optional
 
+import i18n
 from folio_uuid.folio_namespaces import FOLIONamespaces
 
-from folio_migration_tools.custom_exceptions import TransformationProcessError
-from folio_migration_tools.custom_exceptions import TransformationRecordFailedError
-from folio_migration_tools.library_configuration import FileDefinition
-from folio_migration_tools.library_configuration import LibraryConfiguration
+from folio_migration_tools.custom_exceptions import (
+    TransformationProcessError,
+    TransformationRecordFailedError,
+)
+from folio_migration_tools.library_configuration import (
+    FileDefinition,
+    LibraryConfiguration,
+)
 from folio_migration_tools.mapping_file_transformation.courses_mapper import (
     CoursesMapper,
 )
@@ -41,10 +45,11 @@ class CoursesMigrator(MigrationTaskBase):
         self,
         task_configuration: TaskConfiguration,
         library_config: LibraryConfiguration,
+        folio_client
     ):
         csv.register_dialect("tsv", delimiter="\t")
         self.task_configuration = task_configuration
-        super().__init__(library_config, task_configuration)
+        super().__init__(library_config, task_configuration, folio_client)
         self.t0 = time.time()
         self.courses_map = self.setup_records_map(
             self.folder_structure.mapping_files_folder

folio_migration_tools/migration_tasks/holdings_csv_transformer.py

@@ -5,23 +5,24 @@ import logging
 import sys
 import time
 import traceback
-from typing import Annotated
-from typing import List
-from typing import Optional
+from typing import Annotated, List, Optional
 
 import i18n
 from folio_uuid.folio_namespaces import FOLIONamespaces
-from folio_uuid.folio_uuid import FolioUUID
 from httpx import HTTPError
 from pydantic import Field
 
-from folio_migration_tools.custom_exceptions import TransformationProcessError
-from folio_migration_tools.custom_exceptions import TransformationRecordFailedError
+from folio_migration_tools.custom_exceptions import (
+    TransformationProcessError,
+    TransformationRecordFailedError,
+)
 from folio_migration_tools.helper import Helper
 from folio_migration_tools.holdings_helper import HoldingsHelper
-from folio_migration_tools.library_configuration import FileDefinition
-from folio_migration_tools.library_configuration import HridHandling
-from folio_migration_tools.library_configuration import LibraryConfiguration
+from folio_migration_tools.library_configuration import (
+    FileDefinition,
+    HridHandling,
+    LibraryConfiguration,
+)
 from folio_migration_tools.mapping_file_transformation.holdings_mapper import (
     HoldingsMapper,
 )
@@ -80,9 +81,10 @@ class HoldingsCsvTransformer(MigrationTaskBase):
         self,
         task_config: TaskConfiguration,
         library_config: LibraryConfiguration,
+        folio_client,
         use_logging: bool = True,
     ):
-        super().__init__(library_config, task_config, use_logging)
+        super().__init__(library_config, task_config, folio_client, use_logging)
         self.fallback_holdings_type = None
         try:
             self.task_config = task_config
@@ -250,9 +252,9 @@ class HoldingsCsvTransformer(MigrationTaskBase):
                         # Prevent the first item in a boundwith to be overwritten
                         # TODO: Find out why not
                         # if legacy_id not in self.holdings_id_map:
-                        self.holdings_id_map[
-                            FolioUUID.clean_iii_identifiers(legacy_id)
-                        ] = self.mapper.get_id_map_tuple(legacy_id, holding, self.object_type)
+                        self.holdings_id_map[legacy_id] = self.mapper.get_id_map_tuple(
+                            legacy_id, holding, self.object_type
+                        )
                     Helper.write_to_file(holdings_file, holding)
                     self.mapper.migration_report.add_general_statistics(
                         i18n.t("Holdings Records Written to disk")
@@ -306,7 +308,7 @@ class HoldingsCsvTransformer(MigrationTaskBase):
                     folio_rec, legacy_id = self.mapper.do_map(
                         legacy_record, f"row # {idx}", FOLIONamespaces.holdings
                     )
-                    self.post_process_holding(folio_rec, legacy_id)
+                    self.post_process_holding(folio_rec, legacy_id, file_def)
                 except TransformationProcessError as process_error:
                     self.mapper.handle_transformation_process_error(idx, process_error)
                 except TransformationRecordFailedError as error:
@@ -326,7 +328,7 @@ class HoldingsCsvTransformer(MigrationTaskBase):
                 f"Total records processed: {self.total_records:,}"
             )
 
-    def post_process_holding(self, folio_rec: dict, legacy_id: str):
+    def post_process_holding(self, folio_rec: dict, legacy_id: str, file_def: FileDefinition):
         HoldingsHelper.handle_notes(folio_rec)
         HoldingsHelper.remove_empty_holdings_statements(folio_rec)
 
@@ -348,6 +350,7 @@ class HoldingsCsvTransformer(MigrationTaskBase):
             raise TransformationRecordFailedError(legacy_id, "No instance id in parsed record", "")
 
         for folio_holding in holdings_from_row:
+            self.mapper.perform_additional_mappings(folio_holding, file_def)
             self.merge_holding_in(folio_holding, all_instance_ids, legacy_id)
         self.mapper.report_folio_mapping(folio_holding, self.mapper.schema)
 
@@ -375,7 +378,10 @@ class HoldingsCsvTransformer(MigrationTaskBase):
         """
         if len(instance_ids) > 1:
             # Is boundwith
-            bw_key = f"bw_{incoming_holding['instanceId']}_{'_'.join(sorted(instance_ids))}"
+            bw_key = (
+                f"bw_{incoming_holding['instanceId']}_{incoming_holding['permanentLocationId']}_"
+                f"{incoming_holding.get('callNumber', '')}_{'_'.join(sorted(instance_ids))}"
+            )
             if bw_key not in self.bound_with_keys:
                 self.bound_with_keys.add(bw_key)
                 self.holdings[bw_key] = incoming_holding

folio_migration_tools/migration_tasks/holdings_marc_transformer.py

@@ -1,18 +1,19 @@
 '''Main "script."'''
 import csv
 import logging
-import i18n
-from typing import Annotated
-from typing import List
+from typing import Annotated, List
 
+import i18n
 from folio_uuid.folio_namespaces import FOLIONamespaces
 from pydantic import Field
 
 from folio_migration_tools.custom_exceptions import TransformationProcessError
 from folio_migration_tools.helper import Helper
-from folio_migration_tools.library_configuration import FileDefinition
-from folio_migration_tools.library_configuration import HridHandling
-from folio_migration_tools.library_configuration import LibraryConfiguration
+from folio_migration_tools.library_configuration import (
+    FileDefinition,
+    HridHandling,
+    LibraryConfiguration,
+)
 from folio_migration_tools.marc_rules_transformation.rules_mapper_holdings import (
     RulesMapperHoldings,
 )
@@ -162,10 +163,11 @@ class HoldingsMarcTransformer(MigrationTaskBase):
         self,
         task_config: TaskConfiguration,
         library_config: LibraryConfiguration,
+        folio_client,
         use_logging: bool = True,
     ):
         csv.register_dialect("tsv", delimiter="\t")
-        super().__init__(library_config, task_config, use_logging)
+        super().__init__(library_config, task_config, folio_client, use_logging)
         self.task_config = task_config
         self.holdings_types = list(
             self.folio_client.folio_get_all("/holdings-types", "holdingsTypes")

folio_migration_tools/migration_tasks/items_transformer.py

@@ -7,20 +7,22 @@ import sys
 import time
 import traceback
 import uuid
-import i18n
-from typing import Annotated
-from typing import List
-from typing import Optional
+from typing import Annotated, List, Optional
 
+import i18n
 from folio_uuid.folio_namespaces import FOLIONamespaces
 from pydantic import Field
 
-from folio_migration_tools.custom_exceptions import TransformationProcessError
-from folio_migration_tools.custom_exceptions import TransformationRecordFailedError
+from folio_migration_tools.custom_exceptions import (
+    TransformationProcessError,
+    TransformationRecordFailedError,
+)
 from folio_migration_tools.helper import Helper
-from folio_migration_tools.library_configuration import FileDefinition
-from folio_migration_tools.library_configuration import HridHandling
-from folio_migration_tools.library_configuration import LibraryConfiguration
+from folio_migration_tools.library_configuration import (
+    FileDefinition,
+    HridHandling,
+    LibraryConfiguration,
+)
 from folio_migration_tools.mapping_file_transformation.item_mapper import ItemMapper
 from folio_migration_tools.mapping_file_transformation.mapping_file_mapper_base import (
     MappingFileMapperBase,
@@ -75,10 +77,11 @@ class ItemsTransformer(MigrationTaskBase):
         self,
         task_config: TaskConfiguration,
         library_config: LibraryConfiguration,
+        folio_client,
         use_logging: bool = True,
     ):
         csv.register_dialect("tsv", delimiter="\t")
-        super().__init__(library_config, task_config, use_logging)
+        super().__init__(library_config, task_config, folio_client, use_logging)
         self.task_config = task_config
         self.check_source_files(
             self.folder_structure.legacy_records_folder, self.task_config.files

folio_migration_tools/migration_tasks/loans_migrator.py

@@ -5,22 +5,22 @@ import logging
 import sys
 import time
 import traceback
-from datetime import datetime
-from datetime import timedelta
+from datetime import datetime, timedelta
 from typing import Optional
 from urllib.error import HTTPError
-from zoneinfo import ZoneInfo
 
-import httpx
 import i18n
 from dateutil import parser as du_parser
 from folio_uuid.folio_namespaces import FOLIONamespaces
+from zoneinfo import ZoneInfo
 
 from folio_migration_tools.circulation_helper import CirculationHelper
 from folio_migration_tools.helper import Helper
-from folio_migration_tools.library_configuration import FileDefinition
-from folio_migration_tools.library_configuration import FolioRelease
-from folio_migration_tools.library_configuration import LibraryConfiguration
+from folio_migration_tools.library_configuration import (
+    FileDefinition,
+    FolioRelease,
+    LibraryConfiguration,
+)
 from folio_migration_tools.mapping_file_transformation.mapping_file_mapper_base import (
     MappingFileMapperBase,
 )
@@ -51,6 +51,7 @@ class LoansMigrator(MigrationTaskBase):
         self,
         task_configuration: TaskConfiguration,
         library_config: LibraryConfiguration,
+        folio_client
     ):
         csv.register_dialect("tsv", delimiter="\t")
         self.patron_item_combos: set = set()
@@ -62,7 +63,7 @@ class LoansMigrator(MigrationTaskBase):
         self.failed_and_not_dupe: dict = {}
         self.migration_report = MigrationReport()
         self.valid_legacy_loans = []
-        super().__init__(library_config, task_configuration)
+        super().__init__(library_config, task_configuration, folio_client)
         self.circulation_helper = CirculationHelper(
             self.folio_client,
             task_configuration.fallback_service_point_id,
@@ -168,7 +169,7 @@ class LoansMigrator(MigrationTaskBase):
             logging.info("SMTP connection is disabled...")
 
     def do_work(self):
-        with httpx.Client(timeout=None) as self.http_client:
+        with self.folio_client.get_folio_http_client() as self.http_client:
             logging.info("Starting")
             starting_index = (
                 self.task_configuration.starting_row - 1

folio_migration_tools/migration_tasks/manual_fee_fines_transformer.py

@@ -4,18 +4,21 @@ import logging
 import sys
 import time
 import traceback
-import i18n
-from typing import List
-from typing import Optional
+from typing import List, Optional
 
+import i18n
 from folio_uuid.folio_namespaces import FOLIONamespaces
 
-from folio_migration_tools.custom_exceptions import TransformationFieldMappingError
-from folio_migration_tools.custom_exceptions import TransformationProcessError
-from folio_migration_tools.custom_exceptions import TransformationRecordFailedError
+from folio_migration_tools.custom_exceptions import (
+    TransformationFieldMappingError,
+    TransformationProcessError,
+    TransformationRecordFailedError,
+)
 from folio_migration_tools.helper import Helper
-from folio_migration_tools.library_configuration import FileDefinition
-from folio_migration_tools.library_configuration import LibraryConfiguration
+from folio_migration_tools.library_configuration import (
+    FileDefinition,
+    LibraryConfiguration,
+)
 from folio_migration_tools.mapping_file_transformation.manual_fee_fines_mapper import (
     ManualFeeFinesMapper,
 )
@@ -44,11 +47,12 @@ class ManualFeeFinesTransformer(MigrationTaskBase):
         self,
         task_configuration: TaskConfiguration,
         library_config: LibraryConfiguration,
+        folio_client,
         use_logging: bool = True,
     ):
         csv.register_dialect("tsv", delimiter="\t")
 
-        super().__init__(library_config, task_configuration, use_logging)
+        super().__init__(library_config, task_configuration, folio_client, use_logging)
         self.object_type_name = self.get_object_type().name
         self.task_configuration = task_configuration
         self.check_source_files(