sommitrealweird/photo/models.py

   1 from django.db import models
   2 from django.conf import settings
   3
   4 class Album(models.Model):
   5     name = models.CharField(max_length=120)
   6     caption = models.TextField(blank=True, null=True)
   7     slug = models.SlugField()
   8
   9     def get_main_image(self):
  10         try:
  11             main_photo = Photo.objects.filter(album=self).order_by('order','image')[0]
  12             return main_photo.image
  13         except:
  14             return None
  15
  16     def __unicode__(self):
  17         return "%s" %(self.name,)
  18
  19     def __str__(self):
  20         return "%s" %(self.name,)
  21
  22 def get_upload_path(photo, filename):
  23     if photo.album:
  24         return "photos/%s/%s" %(photo.album.slug, filename)
  25     else:
  26         return "photos/%s" %(filename)
  27
  28 class Photo(models.Model):
  29     image = models.ImageField(upload_to=get_upload_path, null=True)
  30     caption = models.TextField(blank=True, null=True)
  31     order = models.IntegerField(blank=True, null=True)
  32     album = models.ForeignKey('Album', on_delete=models.CASCADE)
  33
  34     def __unicode__(self):
  35         return "%s" %(self.image)
  36
  37     def __str__(self):
  38         return "%s" %(self.image)