mirror of
https://github.com/fergalmoran/dss.api.git
synced 2025-12-22 09:18:13 +00:00
106 lines
3.5 KiB
Python
106 lines
3.5 KiB
Python
import re
|
|
import urllib
|
|
|
|
from django.contrib.sites.models import Site
|
|
from django.template.defaultfilters import slugify
|
|
|
|
__author__ = 'fergalm'
|
|
|
|
|
|
def url_path_join(*parts):
|
|
"""Join and normalize url path parts with a slash."""
|
|
schemes, netlocs, paths, queries, fragments = list(zip(*(urllib.parse.urlsplit(part) for part in parts)))
|
|
# Use the first value for everything but path. Join the path on '/'
|
|
scheme = next((x for x in schemes if x), '')
|
|
netloc = next((x for x in netlocs if x), '')
|
|
path = '/'.join(x.strip('/') for x in paths if x)
|
|
query = next((x for x in queries if x), '')
|
|
fragment = next((x for x in fragments if x), '')
|
|
return urllib.parse.urlunsplit((scheme, netloc, path, query, fragment))
|
|
|
|
|
|
def urlclean(url):
|
|
# remove double slashes
|
|
ret = urllib.parse.urljoin(url, urllib.parse.urlparse(url).path.replace('//', '/'))
|
|
return ret
|
|
|
|
|
|
def unique_slugify(instance, value, slug_field_name='slug', queryset=None, slug_separator='-'):
|
|
"""
|
|
Calculates and stores a unique slug of ``value`` for an instance.
|
|
|
|
``slug_field_name`` should be a string matching the name of the field to
|
|
store the slug in (and the field to check against for uniqueness).
|
|
|
|
``queryset`` usually doesn't need to be explicitly provided - it'll default
|
|
to using the ``.all()`` queryset from the model's default manager.
|
|
"""
|
|
slug_field = instance._meta.get_field(slug_field_name)
|
|
|
|
slug = getattr(instance, slug_field.attname)
|
|
slug_len = slug_field.max_length
|
|
|
|
# Sort out the initial slug, limiting its length if necessary.
|
|
slug = slugify(value)
|
|
if slug_len:
|
|
slug = slug[:slug_len]
|
|
slug = _slug_strip(slug, slug_separator)
|
|
original_slug = slug
|
|
|
|
# Create the queryset if one wasn't explicitly provided and exclude the
|
|
# current instance from the queryset.
|
|
if queryset is None:
|
|
queryset = instance.__class__._default_manager.all()
|
|
if instance.pk:
|
|
queryset = queryset.exclude(pk=instance.pk)
|
|
|
|
# Find a unique slug. If one matches, at '-2' to the end and try again
|
|
# (then '-3', etc).
|
|
next = 2
|
|
while not slug or queryset.filter(**{slug_field_name: slug}):
|
|
slug = original_slug
|
|
end = '%s%s' % (slug_separator, next)
|
|
if slug_len and len(slug) + len(end) > slug_len:
|
|
slug = slug[:slug_len - len(end)]
|
|
slug = _slug_strip(slug, slug_separator)
|
|
slug = '%s%s' % (slug, end)
|
|
next += 1
|
|
|
|
return slug
|
|
|
|
|
|
def _slug_strip(value, separator='-'):
|
|
"""
|
|
Cleans up a slug by removing slug separator characters that occur at the
|
|
beginning or end of a slug.
|
|
|
|
If an alternate separator is used, it will also replace any instances of
|
|
the default '-' separator with the new separator.
|
|
"""
|
|
separator = separator or ''
|
|
if separator == '-' or not separator:
|
|
re_sep = '-'
|
|
else:
|
|
re_sep = '(?:-|%s)' % re.escape(separator)
|
|
# Remove multiple instances and if an alternate separator is provided,
|
|
# replace the default '-' separator.
|
|
if separator != re_sep:
|
|
value = re.sub('%s+' % re_sep, separator, value)
|
|
# Remove separator from the beginning and end of the slug.
|
|
if separator:
|
|
if separator != '-':
|
|
re_sep = re.escape(separator)
|
|
value = re.sub(r'^%s+|%s+$' % (re_sep, re_sep), '', value)
|
|
return value
|
|
|
|
|
|
def is_absolute(url):
|
|
return bool(urllib.parse.urlparse(url).scheme)
|
|
|
|
|
|
def wrap_full(url):
|
|
if not is_absolute(url):
|
|
url = "http://%s%s" % (Site.objects.get_current().domain, url)
|
|
|
|
return url
|