Source code for django_thumbor

# -*- coding: utf-8 -*-

from libthumbor import CryptoURL
from django_thumbor import conf
from django.conf import settings
import logging
logger = logging.getLogger(__name__)


crypto = CryptoURL(key=conf.THUMBOR_SECURITY_KEY)


def _remove_prefix(url, prefix):
    if url.startswith(prefix):
        return url[len(prefix):]
    return url


def _remove_schema(url):
    return _remove_prefix(url, 'http://')


def _prepend_media_url(url):
    if settings.MEDIA_URL and url.startswith(settings.MEDIA_URL):
        url = _remove_prefix(url, settings.MEDIA_URL)
        url.lstrip('/')
        return '%s/%s' % (conf.THUMBOR_MEDIA_URL, url)
    return url


def _prepend_static_url(url):
    if conf.THUMBOR_STATIC_ENABLED and url.startswith(settings.STATIC_URL):
        url = _remove_prefix(url, settings.STATIC_URL)
        url.lstrip('/')
        return '%s/%s' % (conf.THUMBOR_STATIC_URL, url)
    return url


# Deny empty or none url
def _handle_empty(url):
    if not url:
        logger.error("Empty URL. Skipping.")
        return ""
    return url

# Accept string url and ImageField or similars classes
# with "url" attr as param
def _handle_url_field(url):
    if hasattr(url, "url"):
        return getattr(url, "url", "")
    return url

def generate_url(image_url, alias=None, **kwargs):
    image_url = _handle_empty(image_url)
    image_url = _handle_url_field(image_url)
    image_url = _prepend_media_url(image_url)
    image_url = _prepend_static_url(image_url)
    image_url = _remove_schema(image_url)

    if alias:
        if alias not in conf.THUMBOR_ALIASES:
            raise RuntimeError(
                'Alias "{}" not found in alias map THUMBOR_ALIASES. '
                'Only found these: {}'
                .format(alias, ", ".join(conf.THUMBOR_ALIASES.keys())))
        alias_args = conf.THUMBOR_ALIASES[alias]
    else:
        alias_args = {}

    final_args = dict(conf.THUMBOR_ARGUMENTS)
    final_args.update(alias_args)
    final_args.update(kwargs)

    thumbor_server = final_args.pop(
        'thumbor_server', conf.THUMBOR_SERVER).rstrip('/')

    encrypted_url = crypto.generate(
        image_url=image_url,
        **final_args).strip('/')

    return '%s/%s' % (thumbor_server, encrypted_url)