diff options
Diffstat (limited to 'brevisurl/backends/local.py')
| -rw-r--r-- | brevisurl/backends/local.py | 6 |
1 files changed, 5 insertions, 1 deletions
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()})) |