You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.6 KiB
58 lines
1.6 KiB
2 years ago
|
from django.db import models
|
||
|
from django.utils.crypto import get_random_string
|
||
|
|
||
|
try:
|
||
|
from django_scopes import scopes_disabled
|
||
|
except ImportError:
|
||
|
from xenua.decorators import no_op as scopes_disabled
|
||
|
|
||
|
|
||
|
class RandomSlugMixin(models.Model):
|
||
|
"""
|
||
|
some alternatives for slug_chars:
|
||
|
abcdeghjkpqrsxyz23456789 # 24 chars, max clarity while being all lowercase
|
||
|
ABCDGHKLMNQRTabcdeghjr23456789 # 30 chars, max clarity mixed case
|
||
|
abcdefghijkmnpqrstuwxyz123467890 # 32 chars, all lowercase
|
||
|
"""
|
||
|
slug_chars = 'abcdefghijkmnpqrstuwxyz123467890'
|
||
|
|
||
|
class Meta:
|
||
|
abstract = True
|
||
|
|
||
|
slug = models.CharField(max_length=6, unique=True)
|
||
|
|
||
|
def save(self, **kwargs):
|
||
|
self.generate_slug()
|
||
|
super().save(**kwargs)
|
||
|
|
||
|
def generate_slug(self):
|
||
|
if not self.slug:
|
||
|
self.slug = self.generate_new_slug()
|
||
|
|
||
|
@scopes_disabled()
|
||
|
def generate_new_slug(self):
|
||
|
slug = get_random_string(self.slug_length, self.slug_chars)
|
||
|
if self.__class__.objects.filter(slug=slug).exists():
|
||
|
return self.generate_new_slug()
|
||
|
return slug
|
||
|
|
||
|
@scopes_disabled()
|
||
|
@property
|
||
|
def slugs_available(self):
|
||
|
return self.slug_count_total - self.__class__.objects.count()
|
||
|
|
||
|
@property
|
||
|
def slug_count_total(self):
|
||
|
return len(self.slug_chars) ** self.slug_length
|
||
|
|
||
|
@property
|
||
|
def slug_length(self):
|
||
|
return self.__class__._meta.get_field('slug').max_length
|
||
|
|
||
|
|
||
|
class RandomSlugPKMixin(RandomSlugMixin):
|
||
|
slug = models.CharField(max_length=6, unique=True, primary_key=True)
|
||
|
|
||
|
class Meta:
|
||
|
abstract = True
|