mirror of
https://github.com/Hopiu/django-uuslug.git
synced 2026-03-16 20:10:24 +00:00
d1ae1e8b42
* switch to six for django3 compatibility * update travis matrix for django3 * fix excludes * exclude legacy py versions for django3 * python-slugify dependency supports 2.7 and 3.5 and above only
54 lines
2.1 KiB
Python
54 lines
2.1 KiB
Python
from django.db.models.base import ModelBase
|
|
from slugify import slugify as pyslugify
|
|
import six
|
|
if six.PY3:
|
|
from django.utils.encoding import smart_str
|
|
else:
|
|
from django.utils.encoding import smart_unicode as smart_str
|
|
|
|
__all__ = ['slugify', 'uuslug']
|
|
|
|
|
|
def slugify(text, entities=True, decimal=True, hexadecimal=True, max_length=0,
|
|
word_boundary=False, separator='-', save_order=False, stopwords=()):
|
|
"""
|
|
Make a slug from a given text.
|
|
"""
|
|
|
|
return smart_str(pyslugify(text, entities, decimal, hexadecimal, max_length,
|
|
word_boundary, separator, save_order, stopwords))
|
|
|
|
|
|
def uuslug(s, instance, entities=True, decimal=True, hexadecimal=True,
|
|
slug_field='slug', filter_dict=None, start_no=1, max_length=0,
|
|
word_boundary=False, separator='-', save_order=False, stopwords=()):
|
|
|
|
""" This method tries a little harder than django's django.template.defaultfilters.slugify. """
|
|
|
|
if isinstance(instance, ModelBase):
|
|
raise Exception("Error: you must pass an instance to uuslug, not a model.")
|
|
|
|
queryset = instance.__class__.objects.all()
|
|
if filter_dict:
|
|
queryset = queryset.filter(**filter_dict)
|
|
if instance.pk:
|
|
queryset = queryset.exclude(pk=instance.pk)
|
|
|
|
# The slug max_length cannot be bigger than the max length of the field
|
|
slug_field_max_length = instance._meta.get_field(slug_field).max_length
|
|
if not max_length or max_length > slug_field_max_length:
|
|
max_length = slug_field_max_length
|
|
|
|
slug = slugify(s, entities=entities, decimal=decimal, hexadecimal=hexadecimal,
|
|
max_length=max_length, word_boundary=word_boundary, separator=separator,
|
|
save_order=save_order, stopwords=stopwords)
|
|
|
|
new_slug = slug
|
|
counter = start_no
|
|
while queryset.filter(**{slug_field: new_slug}).exists():
|
|
if len(slug) + len(separator) + len(str(counter)) > max_length:
|
|
slug = slug[:max_length - len(slug) - len(separator) - len(str(counter))]
|
|
new_slug = "{}{}{}".format(slug, separator, counter)
|
|
counter += 1
|
|
|
|
return new_slug
|