Split models

gallery/models/album.py

@@ -0,0 +1,30 @@
+from datetime import datetime
+
+from django.db import models
+from django.urls import reverse
+from django.utils.text import slugify
+
+from gallery.models.location import Location
+
+
+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")
+    is_public = models.BooleanField(default=False, verbose_name="Published")
+
+    def save(self, *args, **kwargs):
+        if not self.slug:
+            self.slug = slugify(self.name)
+        super(Album, self).save(*args, **kwargs)
+
+    def get_absolute_url(self):
+        return reverse('gallery:album_url', kwargs={'album_slug': self.slug})
+
+    class Meta:
+        verbose_name_plural = "Albums"
+
+    def __str__(self):
+        return '{}'.format(self.name)