djresttoolkit 0.6.0__tar.gz → 0.6.1__tar.gz

{djresttoolkit-0.6.0 → djresttoolkit-0.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: djresttoolkit
-Version: 0.6.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

{djresttoolkit-0.6.0 → djresttoolkit-0.6.1}/pyproject.toml

@@ -4,7 +4,7 @@
 
 [project]
 name = "djresttoolkit"
-version = "0.6.0"
+version = "0.6.1"
 description = "A collection of Django and DRF utilities to simplify API development."
 readme = { file = "README.md", content-type = "text/markdown" }
 license = { file = "LICENSE" }

djresttoolkit-0.6.1/src/djresttoolkit/dbseed/models/__init__.py

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

djresttoolkit-0.6.0/src/djresttoolkit/dbseed/models/_base_seed_model.py → djresttoolkit-0.6.1/src/djresttoolkit/dbseed/models/_seed_model.py

@@ -1,31 +1,36 @@
 from __future__ import annotations
 
-from typing import Any
+from typing import Any, ClassVar
 
 from django.db.models import ForeignKey, ManyToManyField, Model, OneToOneField
 from pydantic import BaseModel
 
 
-class BaseSeedModel(BaseModel):
+class SeedModel(BaseModel):
     """
     Base class for all fake data models.
-    Each subclass must define a `model` attribute.
+    Each subclass must define a `__model__` attribute (a Django model).
     """
 
-    class Meta:
-        model: type[Model]
+    __model__: ClassVar[type[Model] | None] = None
 
     def __init_subclass__(cls, **kwargs: Any) -> None:
         super().__init_subclass__(**kwargs)
-        if not hasattr(cls, "Meta") or not hasattr(cls.Meta, "model"):
+        if not hasattr(cls, "__model__"):
             raise TypeError(
-                f"{cls.__name__} must define a Meta class with a Django model"
+                f"{cls.__name__} must define a `__model__` attribute "
+                f"pointing to a Django model"
             )
 
     @classmethod
-    def get_meta(cls) -> type[Meta]:
-        """Class-level access."""
-        return cls.Meta
+    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]]]:
@@ -35,13 +40,13 @@ class BaseSeedModel(BaseModel):
         data = cls().model_dump()
 
         # Handle ForeignKey and OneToOneField
-        for field in cls.get_meta().model._meta.get_fields():
+        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_meta().model.objects.values_list(
+                        used_ids = cls.get_model().objects.values_list(
                             field.name, flat=True
                         )
                         available = rel_model.objects.exclude(pk__in=used_ids)
@@ -53,7 +58,8 @@ class BaseSeedModel(BaseModel):
         # Collect ManyToMany fields
         m2m_fields: list[ManyToManyField[Any, Any]] = [
             field
-            for field in cls.get_meta().model._meta.get_fields()
+            for field in cls.get_model()._meta.get_fields()
             if isinstance(field, ManyToManyField)
         ]
         return data, m2m_fields
+  

{djresttoolkit-0.6.0 → djresttoolkit-0.6.1}/src/djresttoolkit/management/commands/dbseed.py

@@ -8,7 +8,7 @@ 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 BaseSeedModel
+from djresttoolkit.dbseed.models import SeedModel
 
 
 class Command(BaseCommand):
@@ -48,7 +48,7 @@ class Command(BaseCommand):
             faker = Faker()
             faker.seed_instance(seed)
 
-        seed_model_classes: list[type[BaseSeedModel]] = []
+        seed_model_classes: list[type[SeedModel]] = []
 
         # Discover all dbseed dirs in installed apps
         for app_config in apps.get_app_configs():
@@ -67,8 +67,8 @@ class Command(BaseCommand):
                     attr = getattr(submodule, attr_name)
                     if (
                         isinstance(attr, type)
-                        and issubclass(attr, BaseSeedModel)
-                        and attr is not BaseSeedModel
+                        and issubclass(attr, SeedModel)
+                        and attr is not SeedModel
                     ):
                         # Filter by model name if provided
                         if (
@@ -84,7 +84,7 @@ class Command(BaseCommand):
 
         # Generate fake data for each discovered dbseed model
         for dbseed_cls in seed_model_classes:
-            django_model = dbseed_cls.get_meta().model
+            django_model = dbseed_cls.get_model()
             created_count: int = 0
 
             for _ in range(count):

djresttoolkit-0.6.0/src/djresttoolkit/dbseed/models/__init__.py

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