model-bakery 1.15.0__py3-none-any.whl → 1.17.0__py3-none-any.whl

model_bakery/__about__.py

@@ -1 +1 @@
-__version__ = "1.15.0"
+__version__ = "1.17.0"

model_bakery/baker.py

@@ -8,6 +8,7 @@ from typing import (
     Iterator,
     List,
     Optional,
+    Set,
     Type,
     Union,
     cast,
@@ -412,12 +413,10 @@ class Baker(Generic[M]):
         params.update(attrs)
         return self._make(**params)
 
-    def get_fields(self) -> List[Any]:
-        return [
-            f
-            for f in self.model._meta.get_fields()
-            if f not in self.model._meta.related_objects
-        ]
+    def get_fields(self) -> Set[Any]:
+        return set(self.model._meta.get_fields()) - set(
+            self.model._meta.related_objects
+        )
 
     def _make(  # noqa: C901
         self,
@@ -499,17 +498,30 @@ class Baker(Generic[M]):
         self, attrs: Dict[str, Any], _commit, _save_kwargs, _from_manager
     ) -> M:
         one_to_many_keys = {}
+        auto_now_keys = {}
+
         for k in tuple(attrs.keys()):
             field = getattr(self.model, k, None)
+
+            if not field:
+                continue
+
             if isinstance(field, ForeignRelatedObjectsDescriptor):
                 one_to_many_keys[k] = attrs.pop(k)
 
+            if hasattr(field, "field") and (
+                getattr(field.field, "auto_now_add", False)
+                or getattr(field.field, "auto_now", False)
+            ):
+                auto_now_keys[k] = attrs[k]
+
         instance = self.model(**attrs)
         # m2m only works for persisted instances
         if _commit:
             instance.save(**_save_kwargs)
             self._handle_one_to_many(instance, one_to_many_keys)
             self._handle_m2m(instance)
+            self._handle_auto_now(instance, auto_now_keys)
 
             if _from_manager:
                 # Fetch the instance using the given Manager, e.g.
@@ -610,6 +622,12 @@ class Baker(Generic[M]):
 
         return False
 
+    def _handle_auto_now(self, instance: Model, attrs: Dict[str, Any]):
+        if not attrs:
+            return
+
+        instance.__class__.objects.filter(pk=instance.pk).update(**attrs)
+
     def _handle_one_to_many(self, instance: Model, attrs: Dict[str, Any]):
         for key, values in attrs.items():
             manager = getattr(instance, key)

model_bakery/random_gen.py

@@ -112,9 +112,7 @@ def gen_time() -> time:
 
 
 def gen_string(max_length: int) -> str:
-    return str(
-        "".join(baker_random.choice(string.ascii_letters) for _ in range(max_length))
-    )
+    return "".join(baker_random.choice(string.ascii_letters) for _ in range(max_length))
 
 
 def _gen_string_get_max_length(field: Field) -> Tuple[str, int]:
@@ -129,7 +127,7 @@ gen_string.required = [_gen_string_get_max_length]  # type: ignore[attr-defined]
 
 def gen_slug(max_length: int) -> str:
     valid_chars = string.ascii_letters + string.digits + "_-"
-    return str("".join(baker_random.choice(valid_chars) for _ in range(max_length)))
+    return "".join(baker_random.choice(valid_chars) for _ in range(max_length))
 
 
 gen_slug.required = ["max_length"]  # type: ignore[attr-defined]

{model_bakery-1.15.0.dist-info → model_bakery-1.17.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: model-bakery
-Version: 1.15.0
+Version: 1.17.0
 Summary: Smart object creation facility for Django.
 Project-URL: Homepage, https://github.com/model-bakers/model_bakery
 Author-email: berin <bernardoxhc@gmail.com>, amureki <amureki@hey.com>
@@ -25,8 +25,12 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Software Development
 Requires-Python: >=3.8
 Requires-Dist: django>=3.2
+Provides-Extra: docs
+Requires-Dist: sphinx; extra == 'docs'
+Requires-Dist: sphinx-rtd-theme; extra == 'docs'
 Provides-Extra: test
 Requires-Dist: black; extra == 'test'
+Requires-Dist: coverage; extra == 'test'
 Requires-Dist: mypy; extra == 'test'
 Requires-Dist: pillow; extra == 'test'
 Requires-Dist: pytest; extra == 'test'
@@ -36,6 +40,11 @@ Description-Content-Type: text/markdown
 
 # Model Bakery: Smart fixtures for better tests
 
+[![Build](https://img.shields.io/github/actions/workflow/status/model-bakers/model_bakery/ci.yml?branch=main)](https://github.com/model-bakers/model_bakery/actions?workflow=Tests)
+[![Coverage](https://img.shields.io/badge/Coverage-97%25-success)](https://github.com/model-bakers/model_bakery/actions?workflow=Tests)
+[![Latest PyPI version](https://img.shields.io/pypi/v/model_bakery.svg)](https://pypi.python.org/pypi/model_bakery/)
+[![Documentation Status](https://readthedocs.org/projects/model-bakery/badge/?version=latest)](https://model-bakery.readthedocs.io/en/latest/?badge=latest)
+
 *Model Bakery* offers you a smart way to create fixtures for testing in
 Django.
 With a simple and powerful API you can create many objects with a single
@@ -43,14 +52,10 @@ line of code.
 
 Model Bakery is a rename of the legacy [Model Mommy project](https://pypi.org/project/model_mommy/).
 
-![Tests](https://github.com/model-bakers/model_bakery/workflows/Tests/badge.svg)
-[![Latest PyPI version](https://img.shields.io/pypi/v/model_bakery.svg)](https://pypi.python.org/pypi/model_bakery/)
-[![Documentation Status](https://readthedocs.org/projects/model-bakery/badge/?version=latest)](https://model-bakery.readthedocs.io/en/latest/?badge=latest)
-
 ## Install
 
 ```bash
-pip install model_bakery
+pip install model-bakery
 ```
 
 ## Usage and Info
@@ -61,6 +66,8 @@ pip install model_bakery
 
 # models.py
 
+from django.db import models
+
 class Customer(models.Model):
     enjoy_jards_macale = models.BooleanField()
     name = models.CharField(max_length=30)

{model_bakery-1.15.0.dist-info → model_bakery-1.17.0.dist-info}/RECORD

@@ -1,18 +1,18 @@
-model_bakery/__about__.py,sha256=m3KwqTPSyglvdkL8ZVtCe-iTXn6P8Bz06hxbY0iqJkk,23
+model_bakery/__about__.py,sha256=BVr0oLjmHWJS0DsUE-HOjf8hMNndB1Nh8X6y77zSq8s,23
 model_bakery/__init__.py,sha256=yY0xZUrr_mskBm07iF-MFBEiXzOuKrlMwpKWkXZs8v0,31
 model_bakery/_types.py,sha256=P0iKC5-Cnh3lyzyZs1mlCfOW31zbEZpxqWCuYonucKo,130
-model_bakery/baker.py,sha256=cCtnH3cg_veS4VXaPeYvShBk8VpZsGltCMV4n3Bwl7E,27511
+model_bakery/baker.py,sha256=mZBIalLwIVIkgUOBftFENz7_SCujhCaN3y_jbaPHSq8,28035
 model_bakery/exceptions.py,sha256=q1oBZvfxL7HAD-F9aWgiYs4P4rz-hJ5TjeX7gcb3Q0I,333
 model_bakery/generators.py,sha256=IkYFBTwSVwa_wwPnZRiVR1vHqUfjWqmyPGCl10RElB0,6467
 model_bakery/gis.py,sha256=eTNOfT98_ncLAb4KmOYJhE-3ALU2_pQbUpWPTolFZJI,1028
 model_bakery/mock_file.txt,sha256=CO5zPS6Uydk48ZRnGcRKVEYKd2YjaUMa9QOgF5gGMCo,10
 model_bakery/mock_img.jpeg,sha256=iF2ybgAVtDvl48-8VKdv_QFHKfnqeOaeGwR6ZLexJkY,1916
 model_bakery/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-model_bakery/random_gen.py,sha256=ZDtUwEEd735ot2V0HeVb8oX3_0QOwdmKAdBpl0bt7oI,9797
+model_bakery/random_gen.py,sha256=SexyzpuAe5FKXIofXloCmJ3tL4S5WJ9cVowB8sMHXP8,9773
 model_bakery/recipe.py,sha256=miruAnulPVL4N7Q3tXRfRV95MUvbpZZ3A5M7afv-2O8,8411
 model_bakery/timezone.py,sha256=rd-KGVHqJoHdHKBgENPMBTRqqxgKE91CM9C0f9nS90M,325
 model_bakery/utils.py,sha256=fw4ewtwkJ6wZ9PbDzBGvODX7CV81ozy2_6JAQmITmX4,4581
-model_bakery-1.15.0.dist-info/METADATA,sha256=6t1Rx86N3V4MxZ97k5pr7Ub-hqZvwWtMan94V7T_5tM,3896
-model_bakery-1.15.0.dist-info/WHEEL,sha256=9QBuHhg6FNW7lppboF2vKVbCGTVzsFykgRQjjlajrhA,87
-model_bakery-1.15.0.dist-info/licenses/LICENSE,sha256=kCwHls7z8Y7NyZCmdnV1qLSdLrQ8bHrXLji5HOasgUc,611
-model_bakery-1.15.0.dist-info/RECORD,,
+model_bakery-1.17.0.dist-info/METADATA,sha256=q3pWgLVXrodr1osb_HUIJSoBTuDsQHSDQ4CzTQrwmpE,4311
+model_bakery-1.17.0.dist-info/WHEEL,sha256=9QBuHhg6FNW7lppboF2vKVbCGTVzsFykgRQjjlajrhA,87
+model_bakery-1.17.0.dist-info/licenses/LICENSE,sha256=kCwHls7z8Y7NyZCmdnV1qLSdLrQ8bHrXLji5HOasgUc,611
+model_bakery-1.17.0.dist-info/RECORD,,