diff options
Diffstat (limited to 'brevisurl/backends')
| -rw-r--r-- | brevisurl/backends/base.py | 1 | ||||
| -rw-r--r-- | brevisurl/backends/local.py | 6 |
2 files changed, 6 insertions, 1 deletions
diff --git a/brevisurl/backends/base.py b/brevisurl/backends/base.py index 3fc053a..f667e79 100644 --- a/brevisurl/backends/base.py +++ b/brevisurl/backends/base.py @@ -10,6 +10,7 @@ class BaseBrevisUrlBackend(object): def __init__(self, fail_silently=False, **kwargs): self.fail_silently = fail_silently self.class_path = '{0}.{1}'.format(self.__module__, self.__class__.__name__) + self.kwargs = kwargs def open(self): """Open a network connection. diff --git a/brevisurl/backends/local.py b/brevisurl/backends/local.py index 9cb37a6..741904d 100644 --- a/brevisurl/backends/local.py +++ b/brevisurl/backends/local.py @@ -36,7 +36,11 @@ class BrevisUrlBackend(BaseBrevisUrlBackend): try: short_url = ShortUrl() - if brevisurl.settings.LOCAL_BACKEND_DOMAIN is not None: + if self.kwargs.get('domain') is not None: + short_url.shortened_url = '{0}{1}'.format(self.kwargs.get('domain').rstrip('/'), + reverse('brevisurl_redirect', + kwargs={'token': self.__generate_token()})) + elif brevisurl.settings.LOCAL_BACKEND_DOMAIN is not None: short_url.shortened_url = '{0}{1}'.format(brevisurl.settings.LOCAL_BACKEND_DOMAIN.rstrip('/'), reverse('brevisurl_redirect', kwargs={'token': self.__generate_token()})) |