model-bakery 1.16.0__tar.gz → 1.17.0__tar.gz

{model_bakery-1.16.0 → model_bakery-1.17.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: model-bakery
-Version: 1.16.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>
@@ -55,7 +55,7 @@ Model Bakery is a rename of the legacy [Model Mommy project](https://pypi.org/pr
 ## Install
 
 ```bash
-pip install model_bakery
+pip install model-bakery
 ```
 
 ## Usage and Info

{model_bakery-1.16.0 → model_bakery-1.17.0}/README.md

@@ -15,7 +15,7 @@ Model Bakery is a rename of the legacy [Model Mommy project](https://pypi.org/pr
 ## Install
 
 ```bash
-pip install model_bakery
+pip install model-bakery
 ```
 
 ## Usage and Info

model_bakery-1.17.0/model_bakery/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.17.0"

{model_bakery-1.16.0 → model_bakery-1.17.0}/model_bakery/baker.py

@@ -498,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.
@@ -609,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-1.16.0 → model_bakery-1.17.0}/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.16.0/model_bakery/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.16.0"