model-bakery 1.11.0__py3-none-any.whl → 1.13.0__py3-none-any.whl

model_bakery/__about__.py

@@ -1 +1 @@
-__version__ = "1.11.0"
+__version__ = "1.13.0"

model_bakery/baker.py

@@ -59,6 +59,10 @@ def _valid_quantity(quantity: Optional[Union[str, int]]) -> bool:
     return quantity is not None and (not isinstance(quantity, int) or quantity < 1)
 
 
+def seed(seed: Union[int, float, str, bytes, bytearray, None]) -> None:
+    Baker.seed(seed)
+
+
 @overload
 def make(
     _model: Union[str, Type[M]],
@@ -312,13 +316,22 @@ def _custom_baker_class() -> Optional[Type]:
 
 
 class Baker(Generic[M]):
+    SENTINEL = object()
+
     attr_mapping: Dict[str, Any] = {}
     type_mapping: Dict = {}
 
+    _global_seed: Union[object, int, float, str, bytes, bytearray, None] = SENTINEL
+
     # Note: we're using one finder for all Baker instances to avoid
     # rebuilding the model cache for every make_* or prepare_* call.
     finder = ModelFinder()
 
+    @classmethod
+    def seed(cls, seed: Union[int, float, str, bytes, bytearray, None]) -> None:
+        random_gen.baker_random.seed(seed)
+        cls._global_seed = seed
+
     @classmethod
     def create(
         cls,

model_bakery/random_gen.py

@@ -14,7 +14,7 @@ import warnings
 from datetime import date, datetime, time, timedelta
 from decimal import Decimal
 from os.path import abspath, dirname, join
-from random import choice, randint, random, uniform
+from random import Random
 from typing import Any, Callable, List, Optional, Tuple, Union
 from uuid import UUID
 
@@ -27,6 +27,8 @@ MAX_LENGTH = 300
 # Postgres database.
 MAX_INT = 100000000000
 
+baker_random = Random()
+
 
 def get_content_file(content: bytes, name: str) -> ContentFile:
     return ContentFile(content, name=name)
@@ -57,7 +59,7 @@ def gen_from_list(a_list: Union[List[str], range]) -> Callable:
         >>>     attr_mapping = {'some_field': gen_from_list(['A', 'B', 'C'])}
 
     """
-    return lambda: choice(list(a_list))
+    return lambda: baker_random.choice(list(a_list))
 
 
 # -- DEFAULT GENERATORS --
@@ -75,16 +77,16 @@ def gen_from_choices(choices: List) -> Callable:
 
 
 def gen_integer(min_int: int = -MAX_INT, max_int: int = MAX_INT) -> int:
-    return randint(min_int, max_int)
+    return baker_random.randint(min_int, max_int)
 
 
 def gen_float() -> float:
-    return random() * gen_integer()
+    return baker_random.random() * gen_integer()
 
 
 def gen_decimal(max_digits: int, decimal_places: int) -> Decimal:
     def num_as_str(x: int) -> str:
-        return "".join([str(randint(0, 9)) for _ in range(x)])
+        return "".join([str(baker_random.randint(0, 9)) for _ in range(x)])
 
     if decimal_places:
         return Decimal(
@@ -110,15 +112,24 @@ def gen_time() -> time:
 
 
 def gen_string(max_length: int) -> str:
-    return str("".join(choice(string.ascii_letters) for _ in range(max_length)))
+    return str(
+        "".join(baker_random.choice(string.ascii_letters) for _ in range(max_length))
+    )
+
+
+def _gen_string_get_max_length(field: Field) -> Tuple[str, int]:
+    max_length = getattr(field, "max_length", None)
+    if max_length is None:
+        max_length = MAX_LENGTH
+    return "max_length", max_length
 
 
-gen_string.required = ["max_length"]  # type: ignore[attr-defined]
+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(choice(valid_chars) for _ in range(max_length)))
+    return str("".join(baker_random.choice(valid_chars) for _ in range(max_length)))
 
 
 gen_slug.required = ["max_length"]  # type: ignore[attr-defined]
@@ -129,11 +140,11 @@ def gen_text() -> str:
 
 
 def gen_boolean() -> bool:
-    return choice((True, False))
+    return baker_random.choice((True, False))
 
 
 def gen_null_boolean():
-    return choice((True, False, None))
+    return baker_random.choice((True, False, None))
 
 
 def gen_url() -> str:
@@ -145,15 +156,15 @@ def gen_email() -> str:
 
 
 def gen_ipv6() -> str:
-    return ":".join(format(randint(1, 65535), "x") for _ in range(8))
+    return ":".join(format(baker_random.randint(1, 65535), "x") for _ in range(8))
 
 
 def gen_ipv4() -> str:
-    return ".".join(str(randint(1, 255)) for _ in range(4))
+    return ".".join(str(baker_random.randint(1, 255)) for _ in range(4))
 
 
 def gen_ipv46() -> str:
-    ip_gen = choice([gen_ipv4, gen_ipv6])
+    ip_gen = baker_random.choice([gen_ipv4, gen_ipv6])
     return ip_gen()
 
 
@@ -190,7 +201,7 @@ gen_ip.required = ["protocol", "default_validators"]  # type: ignore[attr-define
 
 
 def gen_byte_string(max_length: int = 16) -> bytes:
-    generator = (randint(0, 255) for x in range(max_length))
+    generator = (baker_random.randint(0, 255) for x in range(max_length))
     return bytes(generator)
 
 
@@ -205,7 +216,7 @@ def gen_content_type():
     from django.contrib.contenttypes.models import ContentType
 
     try:
-        return ContentType.objects.get_for_model(choice(apps.get_models()))
+        return ContentType.objects.get_for_model(baker_random.choice(apps.get_models()))
     except (AssertionError, RuntimeError):
         # AssertionError is raised by Django's test framework when db access is not available:
         # https://github.com/django/django/blob/stable/4.0.x/django/test/testcases.py#L150
@@ -269,7 +280,7 @@ gen_m2m.required = [_fk_model, "_using"]  # type: ignore[attr-defined]
 
 
 def gen_coord() -> float:
-    return uniform(0, 1)
+    return baker_random.uniform(0, 1)
 
 
 def gen_coords() -> str:

model_bakery/utils.py

@@ -94,7 +94,9 @@ def seq(value, increment_by=1, start=None, suffix=None):
             else:
                 yield series_date
     else:
-        for n in itertools.count(start or increment_by, increment_by):
+        for n in itertools.count(
+            increment_by if start is None else start, increment_by
+        ):
             if suffix:
                 yield value + str(n) + suffix
             else:

{model_bakery-1.11.0.dist-info → model_bakery-1.13.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: model-bakery
-Version: 1.11.0
+Version: 1.13.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>
@@ -12,6 +12,7 @@ Classifier: Framework :: Django
 Classifier: Framework :: Django :: 3.2
 Classifier: Framework :: Django :: 4.0
 Classifier: Framework :: Django :: 4.1
+Classifier: Framework :: Django :: 4.2
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent

{model_bakery-1.11.0.dist-info → model_bakery-1.13.0.dist-info}/RECORD

@@ -1,18 +1,18 @@
-model_bakery/__about__.py,sha256=uTVPjS5n_41_n-e8UW1W9n8Wt-3YLQI1WS0r4xGphOk,23
+model_bakery/__about__.py,sha256=qyRV1Az4m5yhA_Nx1r_oZmIwZAZJpLTJdDw9oDJaqr8,23
 model_bakery/__init__.py,sha256=yY0xZUrr_mskBm07iF-MFBEiXzOuKrlMwpKWkXZs8v0,31
 model_bakery/_types.py,sha256=P0iKC5-Cnh3lyzyZs1mlCfOW31zbEZpxqWCuYonucKo,130
-model_bakery/baker.py,sha256=D1M-5lP38Ypp4qHWTos7_0uslR2eYiq9lI50p_an4BI,26753
+model_bakery/baker.py,sha256=uEAudFQYa9o6CjaZGvy0LXMWYzrxthvZ1VKTxzWrses,27132
 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=sQk-xhUhVzYT8YhEINs5ZO1s_GQzOfUWGVfHdb0IDoE,9196
+model_bakery/random_gen.py,sha256=T_SQ7ouKufWFq7QMV1os-3TUgPynycOhYM_zpA1cbME,9619
 model_bakery/recipe.py,sha256=3BU-UgaofBJhYnHjH0yTMK7e7IFI3Jv_pp-XZlvF6KE,8292
 model_bakery/timezone.py,sha256=rd-KGVHqJoHdHKBgENPMBTRqqxgKE91CM9C0f9nS90M,325
-model_bakery/utils.py,sha256=zL6f9yu50X-yyCxo2hTSp7NT8vSEv-LdysfUNV7sY1E,4509
-model_bakery-1.11.0.dist-info/METADATA,sha256=blHEEP1VLsQtU8JbN2V4R2_bwq625MlJ2mAmwDOV8xE,3641
-model_bakery-1.11.0.dist-info/WHEEL,sha256=EI2JsGydwUL5GP9t6kzZv7G3HDPi7FuZDDf9In6amRM,87
-model_bakery-1.11.0.dist-info/licenses/LICENSE,sha256=HIJ64rfqx0cBo7VpRa9QZwqpHMyPQRQYhvWVq6enYXc,582
-model_bakery-1.11.0.dist-info/RECORD,,
+model_bakery/utils.py,sha256=dsbUlzroP_NMtS7SJ5koxfAwO0F4Y_yzt5FlOAfksRA,4550
+model_bakery-1.13.0.dist-info/METADATA,sha256=ksssXGIZ-gVhGu98VSwHc9SICy83oYwPNyb-z1K1erQ,3680
+model_bakery-1.13.0.dist-info/WHEEL,sha256=9QBuHhg6FNW7lppboF2vKVbCGTVzsFykgRQjjlajrhA,87
+model_bakery-1.13.0.dist-info/licenses/LICENSE,sha256=HIJ64rfqx0cBo7VpRa9QZwqpHMyPQRQYhvWVq6enYXc,582
+model_bakery-1.13.0.dist-info/RECORD,,

{model_bakery-1.11.0.dist-info → model_bakery-1.13.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.14.0
+Generator: hatchling 1.18.0
 Root-Is-Purelib: true
 Tag: py3-none-any