model-bakery 1.12.0__tar.gz → 1.13.0__tar.gz

{model_bakery-1.12.0 → model_bakery-1.13.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: model-bakery
-Version: 1.12.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>

model_bakery-1.13.0/model_bakery/__about__.py

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

{model_bakery-1.12.0 → model_bakery-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-1.12.0 → model_bakery-1.13.0}/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,7 +112,9 @@ 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]:
@@ -125,7 +129,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(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]
@@ -136,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:
@@ -152,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()
 
 
@@ -197,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)
 
 
@@ -212,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
@@ -276,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-1.12.0/model_bakery/__about__.py

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