djresttoolkit 0.5.0__py3-none-any.whl → 0.6.1__py3-none-any.whl

{djresttoolkit-0.5.0.dist-info → djresttoolkit-0.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: djresttoolkit
-Version: 0.5.0
+Version: 0.6.1
 Summary: A collection of Django and DRF utilities to simplify API development.
 Project-URL: Homepage, https://github.com/shaileshpandit141/djresttoolkit
 Project-URL: Documentation, https://shaileshpandit141.github.io/djresttoolkit
@@ -46,6 +46,7 @@ Classifier: Topic :: Software Development :: Libraries
 Classifier: Topic :: Utilities
 Classifier: Typing :: Typed
 Requires-Python: >=3.13
+Requires-Dist: faker>=37.5.3
 Requires-Dist: pydantic>=2.11.7
 Provides-Extra: dev
 Requires-Dist: mypy; extra == 'dev'

{djresttoolkit-0.5.0.dist-info → djresttoolkit-0.6.1.dist-info}/RECORD

@@ -5,10 +5,17 @@ src/djresttoolkit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 src/djresttoolkit/admin.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 src/djresttoolkit/apps.py,sha256=nKb5GUIEhAB3IL3lTmEXNc5XuvvaZupH-1CCuYKFrEQ,158
 src/djresttoolkit/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+src/djresttoolkit/dbseed/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+src/djresttoolkit/dbseed/models/__init__.py,sha256=B2WJqw2ncqf5nNhkM3XDFx13oNg2ccsULUQkFAjOyrY,105
+src/djresttoolkit/dbseed/models/_gen.py,sha256=qBPQaLvh1rcEam0YmE4JBJqpa-Vv5IFlIIagkEMHDVw,206
+src/djresttoolkit/dbseed/models/_seed_model.py,sha256=0cmbi0VNKjmJbwhjeCFsvb3iKYjok6TJOk6Y2MF_3N4,2443
 src/djresttoolkit/mail/__init__.py,sha256=tB9SdMlhfWQ640q4aobZ0H1c7fTWalpDL2I-onkr2VI,268
 src/djresttoolkit/mail/_email_sender.py,sha256=bPMqgD5HibJcOZgO6xxHOhdK9HEhnGNC6BoMPpo-h7k,3096
 src/djresttoolkit/mail/_models.py,sha256=of5KsLGvsN2OWgDYgdtLEijulg817TXgsLKuUdsnDQc,1447
 src/djresttoolkit/mail/_types.py,sha256=zf6CcXR1ei_UmZ1nLAJa378OAJ6ftnBICqEOkzXPNw8,646
+src/djresttoolkit/management/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+src/djresttoolkit/management/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+src/djresttoolkit/management/commands/dbseed.py,sha256=vS-s_B8No2rTl_T7HuIBg7f_Yp_6lztvK7FbcaxTZaA,4227
 src/djresttoolkit/middlewares/__init__.py,sha256=GZHU3Yy4xXoEi62tHn0UJNxN6XgGM2_HES8Bt5AS5Lk,100
 src/djresttoolkit/middlewares/_response_time_middleware.py,sha256=1wCwdkW5Ng6HJo8zx0F7ylms84OGP-1K0kbyG6Vacuk,908
 src/djresttoolkit/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -20,8 +27,8 @@ src/djresttoolkit/throttling/_throttle_inspector.py,sha256=Kss6ZxKy-EXq9UGaGprGD
 src/djresttoolkit/views/__init__.py,sha256=XrxBrs6sH4HmUzp41omcmy_y94pSaXAVn01ttQ022-4,76
 src/djresttoolkit/views/_exceptions/__init__.py,sha256=DrCUxuPNyBR4WhzNutn5HDxLa--q51ykIxSG7_bFsOI,83
 src/djresttoolkit/views/_exceptions/_exception_handler.py,sha256=_o7If47bzWLl57LeSXSWsIDsJGo2RIpwYAwNQ-hsHVY,2839
-djresttoolkit-0.5.0.dist-info/METADATA,sha256=_XWebt-T78KglxuWs9jNFymCdcHhmbGQ1bJSlNFfh8A,8472
-djresttoolkit-0.5.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-djresttoolkit-0.5.0.dist-info/entry_points.txt,sha256=YMhfTF-7mYppO8QqqWnvR_hyMWvoYxD6XI94_ViFu3k,60
-djresttoolkit-0.5.0.dist-info/licenses/LICENSE,sha256=8-oZM3yuuTRjySMbVKX9YXYA7Y4M_KhQNBYXPFjeWUo,1074
-djresttoolkit-0.5.0.dist-info/RECORD,,
+djresttoolkit-0.6.1.dist-info/METADATA,sha256=doM3qba2_MGPAyrrfz3g79IjcNGaKyqiXYodOYFotso,8501
+djresttoolkit-0.6.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+djresttoolkit-0.6.1.dist-info/entry_points.txt,sha256=YMhfTF-7mYppO8QqqWnvR_hyMWvoYxD6XI94_ViFu3k,60
+djresttoolkit-0.6.1.dist-info/licenses/LICENSE,sha256=8-oZM3yuuTRjySMbVKX9YXYA7Y4M_KhQNBYXPFjeWUo,1074
+djresttoolkit-0.6.1.dist-info/RECORD,,

src/djresttoolkit/dbseed/models/__init__.py

@@ -0,0 +1,4 @@
+from ._gen import Field, Gen
+from ._seed_model import SeedModel
+
+__all__ = ["SeedModel", "Gen", "Field"]

src/djresttoolkit/dbseed/models/_gen.py

@@ -0,0 +1,11 @@
+from faker import Faker
+from pydantic import Field as PydField
+
+
+class Generator:
+    @classmethod
+    def create_faker(cls) -> Faker:
+        return Faker()
+
+Gen = Generator.create_faker()
+Field = PydField

src/djresttoolkit/dbseed/models/_seed_model.py

@@ -0,0 +1,65 @@
+from __future__ import annotations
+
+from typing import Any, ClassVar
+
+from django.db.models import ForeignKey, ManyToManyField, Model, OneToOneField
+from pydantic import BaseModel
+
+
+class SeedModel(BaseModel):
+    """
+    Base class for all fake data models.
+    Each subclass must define a `__model__` attribute (a Django model).
+    """
+
+    __model__: ClassVar[type[Model] | None] = None
+
+    def __init_subclass__(cls, **kwargs: Any) -> None:
+        super().__init_subclass__(**kwargs)
+        if not hasattr(cls, "__model__"):
+            raise TypeError(
+                f"{cls.__name__} must define a `__model__` attribute "
+                f"pointing to a Django model"
+            )
+
+    @classmethod
+    def get_model(cls) -> type[Model]:
+        """Class-level access to the Django model."""
+        if cls.__model__ is None:
+            raise ValueError(
+                f"{cls.__name__} has no `__model__` defined. "
+                "Please set it to a Django model class."
+            )
+        return cls.__model__
+
+    @classmethod
+    def create_instance(cls) -> tuple[dict[str, Any], list[ManyToManyField[Any, Any]]]:
+        """Handle ForeignKey, OneToOneField and ManyToMany relationship."""
+
+        # dump pydantic model to python dict
+        data = cls().model_dump()
+
+        # Handle ForeignKey and OneToOneField
+        for field in cls.get_model()._meta.get_fields():
+            if isinstance(field, (ForeignKey, OneToOneField)):
+                rel_model = field.remote_field.model
+                if rel_model.objects.exists():
+                    # For OneToOne, must ensure unique (pick unused relation)
+                    if isinstance(field, OneToOneField):
+                        used_ids = cls.get_model().objects.values_list(
+                            field.name, flat=True
+                        )
+                        available = rel_model.objects.exclude(pk__in=used_ids)
+                        if available.exists():
+                            data[field.name] = available.order_by("?").first()
+                    else:  # Normal ForeignKey
+                        data[field.name] = rel_model.objects.order_by("?").first()
+
+        # Collect ManyToMany fields
+        m2m_fields: list[ManyToManyField[Any, Any]] = [
+            field
+            for field in cls.get_model()._meta.get_fields()
+            if isinstance(field, ManyToManyField)
+        ]
+        return data, m2m_fields
+  

src/djresttoolkit/management/commands/dbseed.py

@@ -0,0 +1,115 @@
+import pkgutil
+from importlib import import_module
+from types import ModuleType
+from typing import Any
+
+from django.apps import apps
+from django.core.management.base import BaseCommand, CommandParser
+from django.db import transaction
+from django.db.models import Model, QuerySet
+
+from djresttoolkit.dbseed.models import SeedModel
+
+
+class Command(BaseCommand):
+    help = "Seed the database with fake data for any Django model"
+
+    def add_arguments(self, parser: CommandParser) -> None:
+        """Add command arguments."""
+        parser.add_argument(
+            "--count",
+            type=int,
+            default=5,
+            help="Number of records per model",
+        )
+        parser.add_argument(
+            "--model",
+            type=str,
+            default=None,
+            help="Specific model name to seed (e.g., User, Product)",
+        )
+        parser.add_argument(
+            "--seed",
+            type=int,
+            default=None,
+            help="Optional Faker seed for reproducible data",
+        )
+
+    def handle(self, *args: Any, **options: Any) -> None:
+        """Handle ForeignKey, OneToOneField and ManyToMany relationship."""
+
+        count = options["count"]
+        model_name = options["model"]
+        seed = options["seed"]
+
+        if seed is not None:
+            from faker import Faker
+
+            faker = Faker()
+            faker.seed_instance(seed)
+
+        seed_model_classes: list[type[SeedModel]] = []
+
+        # Discover all dbseed dirs in installed apps
+        for app_config in apps.get_app_configs():
+            try:
+                module: ModuleType = import_module(f"{app_config.name}.dbseed")
+            except ModuleNotFoundError:
+                continue  # app has no dbseed dir
+
+            # Iterate over modules in the dbseed package
+            for _, name, ispkg in pkgutil.iter_modules(module.__path__):
+                if ispkg:
+                    continue
+
+                submodule = import_module(f"{app_config.name}.dbseed.{name}")
+                for attr_name in dir(submodule):
+                    attr = getattr(submodule, attr_name)
+                    if (
+                        isinstance(attr, type)
+                        and issubclass(attr, SeedModel)
+                        and attr is not SeedModel
+                    ):
+                        # Filter by model name if provided
+                        if (
+                            not model_name
+                            or model_name.lower()
+                            == attr.__name__.replace("DBSeedModel", "").lower()
+                        ):
+                            seed_model_classes.append(attr)
+
+        if not seed_model_classes:
+            self.stdout.write(self.style.WARNING("No matching dbseed models found."))
+            return None
+
+        # Generate fake data for each discovered dbseed model
+        for dbseed_cls in seed_model_classes:
+            django_model = dbseed_cls.get_model()
+            created_count: int = 0
+
+            for _ in range(count):
+                try:
+                    with transaction.atomic():
+                        data, m2m_fields = dbseed_cls.create_instance()
+                        obj = django_model.objects.create(**data)
+                        created_count += 1
+
+                        # Assign ManyToMany fields
+                        for m2m_field in m2m_fields:
+                            rel_model = m2m_field.remote_field.model
+                            related_instances: QuerySet[Model] | list[Any] = (
+                                rel_model.objects.order_by("?")[:2]
+                                if rel_model.objects.exists()
+                                else []
+                            )
+                            getattr(obj, m2m_field.name).set(related_instances)
+                except Exception as error:
+                    self.stderr.write(
+                        f"Error creating {django_model.__name__} instance: {error}"
+                    )
+                    continue
+            self.stdout.write(
+                self.style.SUCCESS(
+                    f"{created_count} records inserted for {django_model.__name__}"
+                )
+            )