diff --git a/model_clone/mixins/clone.py b/model_clone/mixins/clone.py index 121373c1..302b78bb 100644 --- a/model_clone/mixins/clone.py +++ b/model_clone/mixins/clone.py @@ -37,6 +37,7 @@ class TestModel(CloneMixin, models.Model): _clonable_one_to_one_fields = [] UNIQUE_DUPLICATE_SUFFIX = 'copy' + USE_UNIQUE_DUPLICATE_SUFFIX = True @property @abc.abstractmethod @@ -75,8 +76,9 @@ def _create_copy_of_instance(cls, instance): .filter(**{'{}__startswith'.format(f.attname): value}) .count() ) - if not str(value).isdigit(): - value += ' {} {}'.format(cls.UNIQUE_DUPLICATE_SUFFIX, count) + if cls.USE_UNIQUE_DUPLICATE_SUFFIX: + if not str(value).isdigit(): + value += ' {} {}'.format(cls.UNIQUE_DUPLICATE_SUFFIX, count) if isinstance(f, SlugField): value = slugify(value) defaults[f.attname] = value