1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
import hashlib
import logging
from django.core.exceptions import ValidationError
from django.db import models
from django.core.validators import URLValidator
from brevisurl import get_connection
import brevisurl.settings
log = logging.getLogger(__name__)
class ShortUrl(models.Model):
"""Model that represents shortened url."""
original_url = models.URLField(max_length=brevisurl.settings.LOCAL_BACKEND_ORIGINAL_URL_MAX_LENGTH,
null=False, blank=False, db_index=True)
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)
created = models.DateTimeField(auto_now_add=True, db_index=True, null=False, blank=False)
def __unicode__(self):
return self.shortened_url
def get_connection(self, fail_silently=False):
if not hasattr(self, 'brevis_connection'):
if self.pk is not None:
self.brevis_connection = get_connection(backend=self.backend,
fail_silently=fail_silently)
else:
self.brevis_connection = get_connection(fail_silently=fail_silently)
return self.brevis_connection
def clean(self):
url_validator = URLValidator()
try:
url_validator(self.original_url)
except ValidationError:
log.exception('ShortUrl.original_url "%s" is not valid URL', self.original_url)
raise
try:
url_validator(self.shortened_url)
except ValidationError:
log.exception('ShortUrl.shortened_url "%s" is not valid URL', self.shortened_url)
raise
return super(ShortUrl, self).clean()
def save(self, force_insert=False, force_update=False, using=None):
if self.pk is None:
self.original_url_hash = hashlib.sha256(self.original_url).hexdigest()
self.full_clean()
return super(ShortUrl, self).save(force_insert, force_update, using)
class Meta:
unique_together = (('original_url_hash', 'backend'),)
verbose_name = 'Short url'
verbose_name_plural = 'Short urls'
ordering = ['-created']
get_latest_by = 'created'
|
