refactor: register unit models in app setup

openedx · Nov 5, 2024 · d963f58 · d963f58
1 parent 23af217
commit d963f58
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/openedx_learning/apps/authoring/containers/apps.py b/openedx_learning/apps/authoring/containers/apps.py
@@ -18,7 +18,7 @@ class ContainersConfig(AppConfig):
 
     def ready(self):
         """
-        Register Component and ComponentVersion.
+        Register ContainerEntity and ContainerEntityVersion.
         """
         from ..publishing.api import register_content_models  # pylint: disable=import-outside-toplevel
         from .models import ContainerEntity, ContainerEntityVersion  # pylint: disable=import-outside-toplevel

diff --git a/openedx_learning/apps/authoring/units/apps.py b/openedx_learning/apps/authoring/units/apps.py
@@ -14,3 +14,12 @@ class UnitsConfig(AppConfig):
     verbose_name = "Learning Core > Authoring > Units"
     default_auto_field = "django.db.models.BigAutoField"
     label = "oel_units"
+
+    def ready(self):
+        """
+        Register Unit and UnitVersion.
+        """
+        from ..publishing.api import register_content_models  # pylint: disable=import-outside-toplevel
+        from .models import Unit, UnitVersion  # pylint: disable=import-outside-toplevel
+
+        register_content_models(Unit, UnitVersion)