[FEATURE] Generate same data for object in every run.

Author: krukas

README.md

@@ -52,6 +52,24 @@ class YourModelAnonymizer(Anonymizer):
         return super().get_queryset().filter(is_staff=True)
 ```
 
+## Faker seed
+
+Default it will use the object id as seed, to generate the same data for every run.
+You can disable this by overriding the `get_object_seed` and return falsy value.
+
+```python
+from django_anonymous import Anonymizer, Faker, register
+from .model import YourModel
+
+
+@register(YourModel)
+class YourModelAnonymizer(Anonymizer):
+    email = Faker("email", unique=True)
+
+    def get_object_seed(self, obj):
+        return None
+```
+
 ## Settings for Anonymizer
 
 Per Anonymizer you can set the select chunk size and update batch size.

django_anonymous/anonymizer.py

@@ -38,10 +38,13 @@ def get_batches(self):
         if batch:
             yield batch
 
+    def get_object_seed(self, obj):
+        return obj.id
+
     def anonymize_object(self, obj):
         for field, value in self._fields.items():
             if callable(value):
-                new_value = value()
+                new_value = value(obj, self.get_object_seed(obj))
             else:
                 new_value = value
 

django_anonymous/faker.py

@@ -14,13 +14,15 @@ def __init__(self, provider, **kwargs):
         self.unique = kwargs.pop("unique", False)
         self.kwargs = kwargs
 
-    def __call__(self):
+    def __call__(self, obj, seed=None):
         local_faker = self._get_faker()
+        if seed:
+            local_faker.seed_instance(seed)
         if self.unique:
             local_faker = local_faker.unique
         return local_faker.format(self.provider, **self.kwargs)
 
-    def _get_faker(self):
+    def _get_faker(self) -> faker.Faker:
         if self.locale not in self._FAKER_REGISTRY:
             self._FAKER_REGISTRY[self.locale] = faker.Faker(locale=self.locale)
         return self._FAKER_REGISTRY[self.locale]

test_project/test_anonymizer.py

@@ -2,6 +2,7 @@
 from django.core.management import call_command
 
 from django_anonymous.register import load_anonymizer, register, registered_models
+from test_project.anon import ItemAnonymizer
 from test_project.models import Item, Order
 
 
@@ -48,3 +49,27 @@ def test_anonymizer():
     assert item.order_id == new_item.order_id
     assert item.description != new_item.description
     assert 25 == new_item.amount
+
+
+@pytest.mark.django_db
+def test_anonymizer_seed():
+    item = Item.objects.create(
+        order=Order.objects.create(first_name="Master", last_name="Chief"),
+        description="test",
+        amount=12.50,
+    )
+    anon = ItemAnonymizer(Item)
+
+    anon.anonymize_object(item)
+    assert item.description != "test"
+
+    # Should generate same description
+    seed_description = item.description
+    item.description = "test"
+    anon.anonymize_object(item)
+    assert item.description == seed_description
+
+    # Should generate different description
+    item.id = item.id + 1
+    anon.anonymize_object(item)
+    assert item.description != seed_description