folio-migration-tools 1.10.2__py3-none-any.whl → 1.10.3__py3-none-any.whl

folio_migration_tools/marc_rules_transformation/rules_mapper_base.py

@@ -1,3 +1,10 @@
+"""Base class for MARC rules-based transformations.
+
+Provides the abstract RulesMapperBase class that all MARC rules mappers inherit from.
+Handles loading transformation rules from JSON, applying rules with conditions,
+and managing the transformation workflow for MARC-to-FOLIO conversions.
+"""
+
 import datetime
 import json
 import logging
@@ -41,6 +48,17 @@ class RulesMapperBase(MapperBase):
         conditions=None,
         parent_id_map: dict[str, tuple] = None,
     ):
+        """Initialize base mapper for MARC rules-based transformations.
+
+        Args:
+            folio_client (FolioClient): FOLIO API client.
+            library_configuration (LibraryConfiguration): Library configuration.
+            task_configuration: Task configuration for MARC transformation.
+            statistical_codes_map (Optional[Dict]): Mapping for statistical codes.
+            schema (dict): JSON schema for validation.
+            conditions: Conditions processor for rules evaluation.
+            parent_id_map (dict[str, tuple]): Optional mapping of parent IDs.
+        """
         super().__init__(library_configuration, task_configuration, folio_client, parent_id_map)
         self.parsed_records = 0
         self.id_map: dict[str, tuple] = {}
@@ -538,7 +556,7 @@ class RulesMapperBase(MapperBase):
             )
 
     def remove_from_id_map(self, former_ids: List[str]):
-        """removes the ID from the map in case parsing failed
+        """Removes the ID from the map in case parsing failed.
 
         Args:
             former_ids (_type_): _description_
@@ -739,9 +757,7 @@ class RulesMapperBase(MapperBase):
 
     @staticmethod
     def grouped(marc_field: Field):
-        """Groups the subfields
-        s -> (s0,s1,s2,...sn-1), (sn,sn+1,sn+2,...s2n-1), (s2n,s2n+1,s2n+2,...s3n-1), ...
-
+        """Group subfields into tuples by repeated subfield occurrences.
 
         Args:
             marc_field (Field): _description_
@@ -778,8 +794,7 @@ class RulesMapperBase(MapperBase):
 
     @staticmethod
     def remove_repeated_subfields(marc_field: Field):
-        """Removes repeated subfields
-        s -> (s0,s1,s2,...sn-1), (sn,sn+1,sn+2,...s2n-1), (s2n,s2n+1,s2n+2,...s3n-1), ...
+        """Remove repeated subfields, keeping only the first occurrence of each.
 
         Args:
             marc_field (Field): _description_
@@ -803,16 +818,13 @@ class RulesMapperBase(MapperBase):
         marc_record: Record,
         folio_record,
     ):
-        """Saves the source marc_record to a file to be loaded via Data Import
+        """Save the source MARC record to a file for Data Import loading.
 
         Args:
-            srs_records_file (_type_): _description_
-            record_type (FOLIONamespaces): _description_
-            folio_client (FolioClient): _description_
-            marc_record (Record): _description_
-            folio_record (_type_): _description_
-            legacy_ids (List[str]): _description_
-            suppress (bool): _description_
+            data_import_marc_file: File handle for writing MARC records.
+            record_type (FOLIONamespaces): Type of record being saved.
+            marc_record (Record): The MARC record to save.
+            folio_record: The corresponding FOLIO record.
         """
         marc_record.add_ordered_field(
             Field(
@@ -838,7 +850,7 @@ class RulesMapperBase(MapperBase):
         file_def: FileDefinition,
         marc_record: Record,
     ):
-        """Map statistical codes to FOLIO instance
+        """Map statistical codes to FOLIO instance.
 
         This method first calls the base class method to map statistical codes
         from the file_def. Then, it checks to see if there are any MARC field
@@ -920,7 +932,7 @@ class RulesMapperBase(MapperBase):
         legacy_ids: List[str],
         suppress: bool,
     ):
-        """Saves the source Marc_record to the Source record Storage module
+        """Saves the source Marc_record to the Source record Storage module.
 
         Args:
             srs_records_file (_type_): _description_
@@ -1075,18 +1087,12 @@ def is_array_of_objects(schema_property):
 
 
 def entity_indicators_match(entity_mapping, marc_field):
-    """
-    Check if the indicators of the entity mapping match the indicators of the MARC field.
-    Entity mappings can limit the fields they are applied to by specifying indicator values that
-    must match the provided MARC field's indicators. If the entity mapping does not specify any
-    indicator values, it is assumed to match all MARC fields. Entity indicator values can be a
-    specific value or a wildcard "*", which matches any value.
-
-    This function compares the indicators of the entity mapping with the indicators of the MARC field.
-    If the entity does not specify any indicator values, the function returns True. If the entity does
-    specify indicator values, the function checks if the MARC field's indicators match the specified
-    values or if the specified values are wildcards. If both indicators match, the function returns True;
-    otherwise, it returns False.
+    """Check if entity mapping indicators match the MARC field indicators.
+
+    Entity mappings can limit the fields they are applied to by specifying indicator
+    values that must match the provided MARC field's indicators. If the entity mapping
+    does not specify any indicator values, it is assumed to match all MARC fields.
+    Entity indicator values can be a specific value or a wildcard "*".
 
     Args:
         entity_mapping (dict): _description_
@@ -1094,7 +1100,7 @@ def entity_indicators_match(entity_mapping, marc_field):
 
     Returns:
         bool: True if the indicators match, False otherwise.
-    """  # noqa: E501
+    """
     if indicator_rule := [x["indicators"] for x in entity_mapping if "indicators" in x]:
         return all(
             [

folio_migration_tools/marc_rules_transformation/rules_mapper_bibs.py

@@ -1,5 +1,4 @@
-"""The default mapper, responsible for parsing MARC21 records acording to the
-FOLIO community specifications"""
+"""MARC21 to FOLIO Instance mapper using community specifications."""
 
 import logging
 import sys
@@ -37,8 +36,7 @@ from folio_migration_tools.migration_tasks.migration_task_base import MarcTaskCo
 
 
 class BibsRulesMapper(RulesMapperBase):
-    """Maps a MARC record to inventory instance format according to
-    the FOLIO community convention"""
+    """Map MARC records to FOLIO inventory instance format."""
 
     def __init__(
         self,
@@ -47,6 +45,14 @@ class BibsRulesMapper(RulesMapperBase):
         task_configuration: MarcTaskConfigurationBase,
         statistical_codes_map: Dict[str, str] = None,
     ):
+        """Initialize mapper for bibliographic record transformations.
+
+        Args:
+            folio_client (FolioClient): FOLIO API client.
+            library_configuration (LibraryConfiguration): Library configuration.
+            task_configuration (MarcTaskConfigurationBase): Bibs transformation configuration.
+            statistical_codes_map (Dict[str, str]): Mapping for statistical codes.
+        """
         super().__init__(
             folio_client,
             library_configuration,
@@ -110,9 +116,9 @@ class BibsRulesMapper(RulesMapperBase):
     def parse_record(
         self, marc_record: Record, file_def: FileDefinition, legacy_ids: List[str]
     ) -> list[dict]:
-        """Parses a bib recod into a FOLIO Inventory instance object
+        """Parse a MARC bib record into a FOLIO Inventory instance object.
+
         Community mapping suggestion: https://bit.ly/2S7Gyp3
-         This is the main function
 
         Args:
             marc_record (Record): _description_
@@ -155,17 +161,16 @@ class BibsRulesMapper(RulesMapperBase):
         ignored_subsequent_fields: set,
         legacy_ids: List[str],
     ):
-        """
-        This method applies a much simplified MARC-to-instance
-        mapping to create a minimal FOLIO Instance record to be
-        used with a Data Import based MARC loading flow, rather
-        than creating SRS records during transformation.
+        """Apply simplified MARC-to-instance mapping for Data Import flow.
+
+        Creates a minimal FOLIO Instance record to be used with a Data Import
+        based MARC loading flow, rather than creating SRS records during transformation.
 
         Args:
-            folio_instance (dict): _description_
-            marc_record (Record): _description_
-            legacy_ids (List[str]): _description_
-            file_def (FileDefinition): _description_
+            folio_instance (dict): The FOLIO instance record to populate.
+            marc_record (Record): The source MARC record.
+            ignored_subsequent_fields (set): Fields to skip during processing.
+            legacy_ids (List[str]): Legacy identifiers for the record.
         """
         main_entry_field_tags = ["100", "110", "111", "130"]
         main_entry_fields = marc_record.get_fields(*main_entry_field_tags)
@@ -200,7 +205,7 @@ class BibsRulesMapper(RulesMapperBase):
         legacy_ids: List[str],
         file_def: FileDefinition,
     ) -> None:
-        """Do stuff not easily captured by the mapping rules
+        """Do stuff not easily captured by the mapping rules.
 
         Args:
             folio_instance (dict): _description_
@@ -575,7 +580,7 @@ class BibsRulesMapper(RulesMapperBase):
         return languages
 
     def get_languages(self, marc_record: Record, legacy_id: List[str]) -> List[str]:
-        """Get languages and tranforms them to correct codes
+        """Get languages and tranforms them to correct codes.
 
         Args:
             marc_record (Record): A pymarc Record object
@@ -591,7 +596,7 @@ class BibsRulesMapper(RulesMapperBase):
         return list(languages)
 
     def fetch_language_codes(self) -> Generator[str, None, None]:
-        """Loads the  list of standardized language codes from LoC
+        """Loads the  list of standardized language codes from LoC.
 
         Yields:
             Generator[str, None, None]: _description_

folio_migration_tools/marc_rules_transformation/rules_mapper_holdings.py

@@ -1,3 +1,10 @@
+"""MARC holdings records rules-based transformation.
+
+Implements transformation of MARC21 holdings (MFHD) records to FOLIO Holdings using
+rules-based mapping. Handles holdings-specific fields including locations, call numbers,
+holdings statements, and notes.
+"""
+
 import copy
 import json
 import logging
@@ -45,6 +52,17 @@ class RulesMapperHoldings(RulesMapperBase):
         boundwith_relationship_map_rows: List[Dict],
         statistical_codes_map: Optional[Dict] = None,
     ):
+        """Initialize mapper for holdings record transformations.
+
+        Args:
+            folio_client (FolioClient): FOLIO API client.
+            location_map: Mapping of legacy to FOLIO locations.
+            task_configuration: Holdings transformation configuration.
+            library_configuration (LibraryConfiguration): Library configuration.
+            parent_id_map (dict): Mapping of parent instance IDs.
+            boundwith_relationship_map_rows (List[Dict]): Bound-with relationship mappings.
+            statistical_codes_map (Optional[Dict]): Mapping for statistical codes.
+        """
         self.conditions = Conditions(
             folio_client,
             self,
@@ -116,9 +134,9 @@ class RulesMapperHoldings(RulesMapperBase):
     def parse_record(
         self, marc_record: Record, file_def: FileDefinition, legacy_ids: List[str]
     ) -> list[dict]:
-        """Parses a mfhd recod into a FOLIO Inventory holdings object
+        """Parse a MFHD record into a FOLIO Inventory holdings object.
+
         Community mapping suggestion: https://tinyurl.com/3rh52e2x
-         This is the main function
 
         Args:
             marc_record (Record): _description_
@@ -131,7 +149,6 @@ class RulesMapperHoldings(RulesMapperBase):
         Returns:
             dict: _description_
         """
-
         self.print_progress()
         folio_holding = self.perform_initial_preparation(marc_record, legacy_ids)
         self.prep_852_notes(marc_record)
@@ -246,7 +263,7 @@ class RulesMapperHoldings(RulesMapperBase):
         ignored_subsequent_fields: Set,
         index_or_legacy_ids: List[str],
     ):
-        """This overwrites the implementation for Auth and instances
+        """This overwrites the implementation for Auth and instances.
 
         Args:
             folio_holding (dict): _description_
@@ -273,7 +290,7 @@ class RulesMapperHoldings(RulesMapperBase):
         legacy_ids: List[str],
         file_def: FileDefinition,
     ):
-        """_summary_
+        """_summary_.
 
         Args:
             marc_record (Record): _description_
@@ -367,15 +384,15 @@ class RulesMapperHoldings(RulesMapperBase):
         self.collect_mrk_statement_notes(marc_record, folio_holding, legacy_ids)
 
     def collect_mrk_statement_notes(self, marc_record, folio_holding, legacy_ids):
-        """Collects MFHD holdings statements as MARC Maker field strings in a FOLIO holdings note
-        and adds them to the FOLIO holdings record.
+        """Collect MFHD holdings statements as MARC Maker field strings in a note.
 
-        This is done to preserve the information in the MARC record for future reference.
+        Preserves the MARC record information for future reference by adding
+        the statements to the FOLIO holdings record.
 
         Args:
-            marc_record (Record): PyMARC record
-            folio_holding (Dict): FOLIO holdings record
-
+            marc_record (Record): PyMARC record.
+            folio_holding (Dict): FOLIO holdings record.
+            legacy_ids: Legacy identifiers for the record.
         """
         if self.task_configuration.include_mrk_statements:
             mrk_statement_notes = []
@@ -389,13 +406,14 @@ class RulesMapperHoldings(RulesMapperBase):
                 ) + self.add_mrk_statements_note(mrk_statement_notes, legacy_ids)
 
     def add_mrk_statements_note(self, mrk_statement_notes: List[str], legacy_ids) -> List[Dict]:
-        """Creates a note from the MRK statements
+        """Create a note from the MRK statements.
 
         Args:
-            mrk_statement_notes (List[str]): A list of MFHD holdings statements as MRK strings
+            mrk_statement_notes (List[str]): A list of MFHD holdings statements as MRK strings.
+            legacy_ids: Legacy identifiers for error reporting.
 
         Returns:
-            List: A list containing the FOLIO holdings note object (Dict)
+            List: A list containing the FOLIO holdings note object (Dict).
         """
         holdings_note_type_tuple = self.conditions.get_ref_data_tuple_by_name(
             self.folio.holding_note_types,
@@ -442,13 +460,14 @@ class RulesMapperHoldings(RulesMapperBase):
     def add_mfhd_as_mrk_note(
         self, marc_record: Record, folio_holding: Dict, legacy_ids: List[str]
     ):
-        """Adds the MFHD as a note to the holdings record
+        """Add the MFHD as a note to the holdings record.
 
-        This is done to preserve the information in the MARC record for future reference.
+        Preserves the MARC record information for future reference.
 
         Args:
-            marc_record (Record): PyMARC record
-            folio_holding (Dict): FOLIO holdings record
+            marc_record (Record): PyMARC record.
+            folio_holding (Dict): FOLIO holdings record.
+            legacy_ids (List[str]): Legacy identifiers for error reporting.
         """
         if self.task_configuration.include_mfhd_mrk_as_note:
             holdings_note_type_tuple = self.conditions.get_ref_data_tuple_by_name(
@@ -498,13 +517,14 @@ class RulesMapperHoldings(RulesMapperBase):
     def add_mfhd_as_mrc_note(
         self, marc_record: Record, folio_holding: Dict, legacy_ids: List[str]
     ):
-        """Adds the MFHD as a note to the holdings record
+        """Add the MFHD as a note to the holdings record.
 
-        This is done to preserve the information in the MARC record for future reference.
+        Preserves the MARC record information for future reference.
 
         Args:
-            marc_record (Record): PyMARC record
-            folio_holding (Dict): FOLIO holdings record
+            marc_record (Record): PyMARC record.
+            folio_holding (Dict): FOLIO holdings record.
+            legacy_ids (List[str]): Legacy identifiers for error reporting.
         """
         if self.task_configuration.include_mfhd_mrc_as_note:
             holdings_note_type_tuple = self.conditions.get_ref_data_tuple_by_name(
@@ -660,17 +680,16 @@ class RulesMapperHoldings(RulesMapperBase):
             )
 
     def setup_boundwith_relationship_map(self, boundwith_relationship_map_list: List[Dict]):
-        """
-        Creates a map of MFHD_ID to BIB_ID for boundwith relationships.
+        """Create a map of MFHD_ID to BIB_ID for boundwith relationships.
 
-        Arguments:
-            boundwith_relationship_map: A list of dictionaries containing the MFHD_ID and BIB_ID.
+        Args:
+            boundwith_relationship_map_list: A list of dicts containing MFHD_ID and BIB_ID.
 
         Returns:
             A dictionary mapping MFHD_ID to a list of BIB_IDs.
 
         Raises:
-            TransformationProcessError: If MFHD_ID or BIB_ID is missing from the entry or if the instance_uuid is not in the parent_id_map.
+            TransformationProcessError: If MFHD_ID or BIB_ID is missing from the entry or instance not in parent_id_map.
             TransformationRecordFailedError: If BIB_ID is not in the instance id map.
         """  # noqa: E501
         new_map = {}

folio_migration_tools/migration_report.py

@@ -1,3 +1,10 @@
+"""Migration reporting and statistics tracking.
+
+Provides the MigrationReport class for tracking migration statistics, errors,
+and warnings during transformation and loading tasks. Generates markdown and
+JSON formatted reports with categorized statistics.
+"""
+
 import logging
 import json
 import i18n
@@ -8,9 +15,10 @@ from folio_migration_tools.i18n_cache import i18n_t
 
 
 class MigrationReport:
-    """Class responsible for handling the migration report"""
+    """Class responsible for handling the migration report."""
 
     def __init__(self):
+        """Initialize a new migration report for tracking statistics and issues."""
         self.report = {}
         self.stats = {}
 
@@ -31,19 +39,19 @@ class MigrationReport:
                 self.report[blurb_id][measure_to_add] = number
 
     def set(self, blurb_id, measure_to_add: str, number: int):
-        """Set a section value  to a specific number
+        """Set a section value to a specific number.
 
         Args:
-            blurb (_type_): _description_
-            measure_to_add (str): _description_
-            number (int): _description_
+            blurb_id: The report section identifier.
+            measure_to_add (str): The measure name to set.
+            number (int): The value to set.
         """
         if blurb_id not in self.report:
             self.report[blurb_id] = {}
         self.report[blurb_id][measure_to_add] = number
 
     def add_general_statistics(self, measure_to_add: str):
-        """Shortcut for adding to the first breakdown
+        """Shortcut for adding to the first breakdown.
 
         Args:
             measure_to_add (str): _description_

folio_migration_tools/migration_tasks/__init__.py

@@ -1,3 +1,5 @@
+"""Migration task implementations for various FOLIO data types."""
+
 from os.path import dirname, basename, isfile, join
 import glob
 

folio_migration_tools/migration_tasks/batch_poster.py

@@ -1,3 +1,13 @@
+"""Legacy BatchPoster task for posting transformed records to FOLIO.
+
+This module provides the BatchPoster migration task which posts transformed FOLIO
+records via batch API endpoints. It supports batch processing, automatic retry of
+failed records, and version management for upsert operations.
+
+Note: This is the legacy implementation. For new migrations, consider using
+InventoryBatchPoster or other specialized batch posters instead.
+"""
+
 import asyncio
 import copy
 import json
@@ -37,7 +47,7 @@ def write_failed_batch_to_file(batch, file):
 
 
 class BatchPoster(MigrationTaskBase):
-    """BatchPoster
+    """BatchPoster.
 
     Parents:
         MigrationTaskBase (_type_): _description_
@@ -55,6 +65,8 @@ class BatchPoster(MigrationTaskBase):
     """
 
     class TaskConfiguration(AbstractTaskConfiguration):
+        """Task configuration for BatchPoster."""
+
         name: Annotated[
             str,
             Field(
@@ -220,6 +232,14 @@ class BatchPoster(MigrationTaskBase):
         folio_client,
         use_logging: bool = True,
     ):
+        """Initialize BatchPoster for posting transformed records to FOLIO.
+
+        Args:
+            task_config (TaskConfiguration): Batch posting configuration.
+            library_config (LibraryConfiguration): Library configuration.
+            folio_client: FOLIO API client.
+            use_logging (bool): Whether to set up task logging.
+        """
         super().__init__(library_config, task_config, folio_client, use_logging)
         self.migration_report = MigrationReport()
         self.performing_rerun = False
@@ -335,9 +355,7 @@ class BatchPoster(MigrationTaskBase):
             json_rec["source"] = "CONSORTIUM-FOLIO"
 
     def set_version(self, batch, query_api, object_type) -> None:
-        """
-        Synchronous wrapper for set_version_async
-        """
+        """Synchronous wrapper for set_version_async."""
         try:
             loop = asyncio.get_running_loop()
         except RuntimeError:
@@ -349,8 +367,7 @@ class BatchPoster(MigrationTaskBase):
             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:
-        """
-        Fetches the current version of the records in the batch if the record exists in FOLIO
+        """Fetches the current version of the records in the batch if the record exists in FOLIO.
 
         Args:
             batch (list): List of records to fetch versions for
@@ -388,8 +405,7 @@ class BatchPoster(MigrationTaskBase):
                 self.prepare_record_for_upsert(record, existing_records[record["id"]])
 
     def patch_record(self, new_record: dict, existing_record: dict, patch_paths: List[str]):
-        """
-        Updates new_record with values from existing_record according to patch_paths.
+        """Updates new_record with values from existing_record according to patch_paths.
 
         Args:
             new_record (dict): The new record to be updated.
@@ -420,8 +436,7 @@ class BatchPoster(MigrationTaskBase):
     def collect_existing_records_for_upsert(
         object_type: str, response_json: dict, existing_records: dict
     ):
-        """
-        Collects existing records from API response into existing_records dict.
+        """Collects existing records from API response into existing_records dict.
 
         Args:
             object_type: The key in response containing the records array
@@ -508,8 +523,7 @@ class BatchPoster(MigrationTaskBase):
             new_record.update(updates)
 
     async def get_with_retry(self, url: str, params=None):
-        """
-        Wrapper around folio_get_async with selective retry logic.
+        """Wrapper around folio_get_async with selective retry logic.
 
         Retries on:
         - Connection errors (FolioConnectionError): Always retry
@@ -723,6 +737,7 @@ class BatchPoster(MigrationTaskBase):
                 len(batch),
                 get_req_size(response),
             )
+            self.num_posted += len(batch)
         elif response.status_code == 200:
             json_report = json.loads(response.text)
             self.users_created += json_report.get("createdRecords", 0)
@@ -926,7 +941,10 @@ class BatchPoster(MigrationTaskBase):
                 temp_start = self.start_datetime
                 self.task_configuration.rerun_failed_records = False
                 self.__init__(
-                    self.task_configuration, self.library_configuration, self.folio_client
+                    self.task_configuration,
+                    self.library_configuration,
+                    self.folio_client,
+                    use_logging=False,
                 )
                 self.performing_rerun = True
                 self.migration_report = temp_report
@@ -1062,9 +1080,7 @@ def get_req_size(response: "Response"):
 
 
 def parse_path(path):
-    """
-    Parses a path like 'foo.bar[0].baz' into ['foo', 'bar', 0, 'baz']
-    """
+    """Parses a path like 'foo.bar[0].baz' into ['foo', 'bar', 0, 'baz']."""
     tokens = []
     # Split by dot, then extract indices
     for part in path.split("."):
@@ -1118,8 +1134,8 @@ def extract_paths(data, paths):
 
 
 def deep_update(target, patch):
-    """
-    Recursively update target dict/list with values from patch dict/list.
+    """Recursively update target dict/list with values from patch.
+
     For lists, only non-None values in patch are merged into target.
     """
     if isinstance(patch, dict):

folio_migration_tools/migration_tasks/bibs_transformer.py

@@ -1,3 +1,9 @@
+"""Bibliographic records transformation task.
+
+Transforms MARC21 bibliographic records to FOLIO Inventory Instances using
+rules-based mapping. Supports various ILS flavors and custom field mappings.
+"""
+
 import logging
 from typing import Annotated, List
 
@@ -24,6 +30,8 @@ from folio_migration_tools.migration_tasks.migration_task_base import (
 
 class BibsTransformer(MigrationTaskBase):
     class TaskConfiguration(MarcTaskConfigurationBase):
+        """Task configuration for BibsTransformer."""
+
         ils_flavour: Annotated[
             IlsFlavour,
             Field(
@@ -116,6 +124,14 @@ class BibsTransformer(MigrationTaskBase):
         folio_client,
         use_logging: bool = True,
     ):
+        """Initialize BibsTransformer for transforming bibliographic records.
+
+        Args:
+            task_config (TaskConfiguration): MARC transformation configuration.
+            library_config (LibraryConfiguration): Library configuration.
+            folio_client: FOLIO API client.
+            use_logging (bool): Whether to set up task logging.
+        """
         super().__init__(library_config, task_config, folio_client, use_logging)
         self.task_config = task_config
         self.task_configuration = self.task_config

folio_migration_tools/migration_tasks/courses_migrator.py

@@ -1,3 +1,9 @@
+"""Course records migration task.
+
+Migrates course information from CSV files to FOLIO Course Reserves module.
+Transforms and validates course data including departments and terms.
+"""
+
 import csv
 import json
 import logging
@@ -30,6 +36,8 @@ from folio_migration_tools.task_configuration import AbstractTaskConfiguration
 
 class CoursesMigrator(MigrationTaskBase):
     class TaskConfiguration(AbstractTaskConfiguration):
+        """Task configuration for CoursesMigrator."""
+
         name: Annotated[
             str,
             Field(
@@ -92,6 +100,13 @@ class CoursesMigrator(MigrationTaskBase):
         library_config: LibraryConfiguration,
         folio_client,
     ):
+        """Initialize CoursesMigrator for migrating course reserves.
+
+        Args:
+            task_configuration (TaskConfiguration): Courses migration configuration.
+            library_config (LibraryConfiguration): Library configuration.
+            folio_client: FOLIO API client.
+        """
         csv.register_dialect("tsv", delimiter="\t")
         self.task_configuration = task_configuration
         super().__init__(library_config, task_configuration, folio_client)