view user_photos/models.py @ 955:71a671dab55d

First commit of whitelisting image hosts. This is behind a feature flag courtesy of waffle.
author Brian Neal <bgneal@gmail.com>
date Wed, 03 Jun 2015 21:13:08 -0500
parents b6e98717690b
children
line wrap: on
line source
"""Models for the user_photos application."""

import datetime

from django.db import models
from django.conf import settings
from django.core.urlresolvers import reverse


class Photo(models.Model):
    """This model represents data about a user uploaded photo."""
    user = models.ForeignKey(settings.AUTH_USER_MODEL,
                             related_name='uploaded_photos')
    upload_date = models.DateTimeField()
    url = models.URLField(max_length=200)
    thumb_url = models.URLField(max_length=200, blank=True)
    signature = models.CharField(max_length=32, blank=True, db_index=True)

    def __unicode__(self):
        return u'Photo by {} on {}'.format(self.user.username,
                               self.upload_date.strftime('%Y-%m-%d %H:%M:%S'))

    def get_absolute_url(self):
        return reverse('user_photos-detail', kwargs={'pk': self.pk})

    def save(self, *args, **kwargs):
        if not self.pk and not self.upload_date:
            self.upload_date = datetime.datetime.now()
        super(Photo, self).save(*args, **kwargs)