view user_photos/models.py @ 697:67f8d49a9377

Cleaned up the code a bit. Separated the S3 stuff out into its own class. This class maybe should be in core. Still want to do some kind of context manager around the temporary file we are creating to ensure it gets deleted.
author Brian Neal <bgneal@gmail.com>
date Sun, 08 Sep 2013 21:02:58 -0500
parents b2a8fde3173a
children b6e98717690b
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)

    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)