Add likes, update templates

gallery/models/album.py

@@ -1,23 +1,35 @@
 from datetime import datetime
 
 from django.db import models
+from django.templatetags.static import static
 from django.urls import reverse
 from django.utils.text import slugify
 
 from gallery.models.location import Location
 
+#from gallery.models.photo import Photo
+
 
 class Album(models.Model):
     name = models.CharField(max_length=150, unique=True, verbose_name="Album")
     slug = models.SlugField(max_length=150, unique=True, verbose_name="Slug")
     location = models.ForeignKey(Location, blank=True, null=True, on_delete=models.SET_NULL, related_name='albums', verbose_name="Location")
     album_date = models.DateField(default=datetime.now, verbose_name="Album Date")
-    cover = models.ImageField(upload_to="covers/", blank=True, null=True, verbose_name="Album Cover")
+    cover = models.ForeignKey("Photo", blank=True, null=True, on_delete=models.SET_NULL, related_name='cover_to', verbose_name="Album cover")
     is_public = models.BooleanField(default=False, verbose_name="Published")
 
     @property
     def photos_in_album(self):
         return self.photos.count()
+    
+    @property
+    def cover_url(self):
+        if self.cover:
+            return self.cover.photo.url
+        random_cover = self.photos.order_by('-width').first()
+        if random_cover:
+            return random_cover.photo.url
+        return static('img/placeholder.png')
 
     def save(self, *args, **kwargs):
         if not self.slug:

gallery/models/photo.py

@@ -43,6 +43,13 @@ class Photo(models.Model):
         self.views += 1
         self.save()
 
+    def get_next(self):
+        return self.__class__.objects.filter(taken_at__gt=self.taken_at, album=self.album.id).order_by('taken_at').first()
+
+    def get_prev(self):
+        return self.__class__.objects.filter(taken_at__lt=self.taken_at, album=self.album.id).order_by('-taken_at').first()
+
+
     def save(self, *args, **kwargs):
         self.exif_data = Exif(self.photo.path)
         datetime_taken = getattr(self.exif_data, 'datetimeoriginal', datetime.now)()