Update populator.py

Adding reset capabilities to allow reusing populator in tests
joke2k · Dec 31, 2016 · 4c63929 · 4c63929
1 parent 71186e5
commit 4c63929
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/django_faker/populator.py b/django_faker/populator.py
@@ -119,7 +119,15 @@ def __init__(self, generator):
         self.entities = {}
         self.quantities = {}
         self.orders = []
-
+
+
+    def reset(self):
+        """
+        :param generator: Generator
+        """
+        self.entities = {}
+        self.quantities = {}
+        self.orders = []
 
     def addEntity(self, model, number, customFieldFormatters=None):
         """