Register API models in admin (#27)

src/hope_dedup_engine/apps/api/admin.py

@@ -0,0 +1,8 @@
+from django.contrib import admin
+
+from hope_dedup_engine.apps.api.models import DeduplicationSet, Duplicate, HDEToken, Image
+
+admin.site.register(DeduplicationSet)
+admin.site.register(Duplicate)
+admin.site.register(HDEToken)
+admin.site.register(Image)

src/hope_dedup_engine/apps/api/models/__init__.py

@@ -1,2 +1,2 @@
 from hope_dedup_engine.apps.api.models.auth import HDEToken  # noqa: F401
-from hope_dedup_engine.apps.api.models.deduplication import DeduplicationSet  # noqa: F401
+from hope_dedup_engine.apps.api.models.deduplication import DeduplicationSet, Duplicate, Image  # noqa: F401

tests/extras/testutils/factories/__init__.py

@@ -1,3 +1,8 @@
+import importlib
+import pkgutil
+from pathlib import Path
+
+from factory.django import DjangoModelFactory
 from pytest_factoryboy import register
 
 from .base import AutoRegisterModelFactory, TAutoRegisterModelFactory, factories_registry
@@ -6,13 +11,22 @@
 from .user import ExternalSystemFactory, GroupFactory, SuperUserFactory, User, UserFactory  # noqa
 from .userrole import UserRole, UserRoleFactory  # noqa
 
+for _, name, _ in pkgutil.iter_modules([str(Path(__file__).parent)]):
+    importlib.import_module(f".{name}", __package__)
+
+
+django_model_factories = {factory._meta.model: factory for factory in DjangoModelFactory.__subclasses__()}
 
-def get_factory_for_model(_model) -> type[TAutoRegisterModelFactory]:
+
+def get_factory_for_model(_model) -> type[TAutoRegisterModelFactory] | type[DjangoModelFactory]:
     class Meta:
         model = _model
 
     bases = (AutoRegisterModelFactory,)
     if _model in factories_registry:
         return factories_registry[_model]  # noqa
 
+    if _model in django_model_factories:
+        return django_model_factories[_model]
+
     return register(type(f"{_model._meta.model_name}AutoCreatedFactory", bases, {"Meta": Meta}))  # noqa

tests/extras/testutils/factories/api.py

@@ -1,12 +1,14 @@
 from factory import SubFactory, fuzzy
 from factory.django import DjangoModelFactory
-from testutils.factories import ExternalSystemFactory
+from testutils.factories import ExternalSystemFactory, UserFactory
 
 from hope_dedup_engine.apps.api.models import DeduplicationSet, HDEToken
 from hope_dedup_engine.apps.api.models.deduplication import Duplicate, Image
 
 
 class TokenFactory(DjangoModelFactory):
+    user = SubFactory(UserFactory)
+
     class Meta:
         model = HDEToken