mirror of
https://github.com/Hopiu/django-uuslug.git
synced 2026-03-16 20:10:24 +00:00
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
from slugify import slugify as pyslugify
|
|
from django.utils 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):
|
|
"""
|
|
Make a slug from a given text.
|
|
"""
|
|
|
|
return smart_str(pyslugify(text, entities, decimal, hexadecimal, max_length,
|
|
word_boundary, separator, save_order))
|
|
|
|
|
|
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):
|
|
|
|
""" This method tries a little harder than django's django.template.defaultfilters.slugify. """
|
|
|
|
if hasattr(instance, 'objects'):
|
|
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)
|
|
|
|
slug = slugify(s, entities=entities, decimal=decimal, hexadecimal=hexadecimal,
|
|
max_length=max_length, word_boundary=word_boundary, separator=separator,
|
|
save_order=save_order)
|
|
|
|
new_slug = slug
|
|
counter = start_no
|
|
while queryset.filter(**{slug_field: new_slug}).exists():
|
|
if max_length > 0:
|
|
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
|