AtomicSlugRetryMixin

Complexity

Total Complexity

6

Size/Duplication

Total Lines	33
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes	3
Bugs	0	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
B	save()	0	29	6

from django.db import transaction

The import django.db could not be resolved.

from django.conf import settings

The import django.conf could not be resolved.

from django.db.utils import IntegrityError

The import django.db.utils could not be resolved.

class AtomicSlugRetryMixin:
    """Makes :see:LocalizedUniqueSlugField work by retrying upon
    violation of the UNIQUE constraint."""

    def save(self, *args, **kwargs):
        """Saves this model instance to the database."""

        max_retries = getattr(
            settings,
            'LOCALIZED_FIELDS_MAX_RETRIES',
            100
        )

        if not hasattr(self, 'retries'):
            self.retries = 0

The attribute retries was defined outside __init__.

        with transaction.atomic():
            try:
                return super().save(*args, **kwargs)
            except IntegrityError as ex:
                # this is as retarded as it looks, there's no
                # way we can put the retry logic inside the slug
                # field class... we can also not only catch exceptions
                # that apply to slug fields... so yea.. this is as
                # retarded as it gets... i am sorry :(
                if 'slug' not in str(ex):
                    raise ex

                if self.retries >= max_retries:
                    raise ex

        self.retries += 1
        return self.save()