original url length from settings

2 files changed, 5 insertions, 2 deletions

diff --git a/brevisurl/models.py b/brevisurl/models.py
index 9ade639..417fa20 100644
--- a/brevisurl/models.py
+++ b/brevisurl/models.py
@@ -2,6 +2,7 @@ import hashlib
 import logging
 from django.core.exceptions import ValidationError
 
+from django.conf import settings
 from django.db import models
 from django.core.validators import URLValidator
 
@@ -13,7 +14,7 @@ log = logging.getLogger(__name__)
 
 class ShortUrl(models.Model):
     """Model that represents shortened url."""
-    original_url = models.URLField(max_length=200, null=False, blank=False)
+    original_url = models.URLField(max_length=getattr(settings, 'LOCAL_BACKEND_ORIGINAL_URL_MAX_LENGTH', 200), null=False, blank=False)
     original_url_hash = models.CharField(max_length=64, null=False, blank=False)
     shortened_url = models.URLField(max_length=200, null=False, blank=False, unique=True)
     backend = models.CharField(max_length=200, null=False, blank=False)
diff --git a/brevisurl/settings.py b/brevisurl/settings.py
index b7c7efa..a1cb0a8 100644
--- a/brevisurl/settings.py
+++ b/brevisurl/settings.py
@@ -15,8 +15,10 @@ LOCAL_BACKEND_TOKEN_CHARS = getattr(settings, 'BREVISURL_LOCAL_BACKEND_TOKEN_CHA
 # Settings for token length.
 LOCAL_BACKEND_TOKEN_LENGTH = getattr(settings, 'BREVISURL_LOCAL_BACKEND_TOKEN_LENGTH', 5)
 
+LOCAL_BACKEND_ORIGINAL_URL_MAX_LENGTH = 200
+
 # Settings for url pattern.
 LOCAL_BACKEND_URL_PATTERN = getattr(settings, 'BREVISURL_LOCAL_BACKEND_URL_PATTERN',
                                     r'^(?P<token>[a-zA-Z0-9]{' + str(LOCAL_BACKEND_TOKEN_LENGTH) + r'})$')
 # Protocol for local backend.
-LOCAL_BACKEND_DOMAIN_PROTOCOL = getattr(settings, 'BREVISURL_LOCAL_BACKEND_DOMAIN_PROTOCOL', 'http')
\ No newline at end of file
+LOCAL_BACKEND_DOMAIN_PROTOCOL = getattr(settings, 'BREVISURL_LOCAL_BACKEND_DOMAIN_PROTOCOL', 'http')