Add Factory.reset_sequence.

1 files changed, 15 insertions, 0 deletions

diff --git a/factory/base.py b/factory/base.py
index 23fdac7..ff5404f 100644
--- a/factory/base.py
+++ b/factory/base.py
@@ -240,6 +240,21 @@ class BaseFactory(object):
     FACTORY_HIDDEN_ARGS = ()
 
     @classmethod
+    def reset_sequence(cls, value=None, force=False):
+        """Reset the sequence counter."""
+        if cls._base_factory:
+            if force:
+                cls._base_factory.reset_sequence(value=value)
+            else:
+                raise ValueError(
+                    "Cannot reset the sequence of a factory subclass. "
+                    "Please call reset_sequence() on the root factory, "
+                    "or call reset_sequence(forward=True)."
+                )
+        else:
+            cls._next_sequence = value
+
+    @classmethod
     def _setup_next_sequence(cls):
         """Set up an initial sequence value for Sequence attributes.