feat: adds new fork migration strategy

Author: rpenido

cms/djangoapps/modulestore_migrator/api.py

@@ -11,7 +11,6 @@
 from openedx.core.types.user import AuthUser
 
 from . import tasks
-from .data import RepeatHandlingStrategy
 from .models import ModulestoreSource
 
 __all__ = (
@@ -35,9 +34,6 @@ def start_migration_to_library(
     """
     Import a course or legacy library into a V2 library (or, a collection within a V2 library).
     """
-    # Can raise NotImplementedError for the Fork strategy
-    assert RepeatHandlingStrategy(repeat_handling_strategy).is_implemented()
-
     source, _ = ModulestoreSource.objects.get_or_create(key=source_key)
     target_library = get_library(target_library_key)
     # get_library ensures that the library is connected to a learning package.

cms/djangoapps/modulestore_migrator/data.py

@@ -70,12 +70,3 @@ def default(cls) -> RepeatHandlingStrategy:
         Returns the default repeat handling strategy.
         """
         return cls.Skip
-
-    def is_implemented(self) -> bool:
-        """
-        Returns True if the repeat handling strategy is implemented.
-        """
-        if self == self.Fork:
-            raise NotImplementedError("Forking is not implemented yet.")
-
-        return True

cms/djangoapps/modulestore_migrator/tasks.py

@@ -132,6 +132,13 @@ def should_update_strategy(self) -> bool:
         """
         return self.repeat_handling_strategy is RepeatHandlingStrategy.Update
 
+    @property
+    def should_fork_strategy(self) -> bool:
+        """
+        Determines whether the repeat handling strategy should fork the entity.
+        """
+        return self.repeat_handling_strategy is RepeatHandlingStrategy.Fork
+
 
 @shared_task(base=_MigrationTask, bind=True)
 # Note: The decorator @set_code_owner_attribute cannot be used here because the UserTaskMixin
@@ -601,8 +608,9 @@ def _get_distinct_target_container_key(
     Returns:
         LibraryContainerLocator: The target container key.
     """
-    # Check if we already processed this block
-    if context.is_already_migrated(source_key):
+    # Check if we already processed this block and we are not forking. If we are forking, we will
+    # want a new target key.
+    if context.is_already_migrated(source_key) and not context.should_fork_strategy:
         existing_version = context.get_existing_target(source_key)
 
         return LibraryContainerLocator(
@@ -646,8 +654,9 @@ def _get_distinct_target_usage_key(
     Raises:
         ValueError: If source_key is invalid
     """
-    # Check if we already processed this block
-    if context.is_already_migrated(source_key):
+    # Check if we already processed this block and we are not forking. If we are forking, we will
+    # want a new target key.
+    if context.is_already_migrated(source_key) and not context.should_fork_strategy:
         log.debug(f"Block {source_key} already exists, reusing existing target")
         existing_target = context.get_existing_target(source_key)
         block_id = existing_target.component.local_key