X-Git-Url: https://git.sommitrealweird.co.uk/sommitrealweird.git/blobdiff_plain/6746b20632b76e4a04be07428ab2581483869d40..77c6bd606ee5003500ed80ea1e3561993f038e05:/sommitrealweird/photo/models.py?ds=sidebyside

diff --git a/sommitrealweird/photo/models.py b/sommitrealweird/photo/models.py
index 3a5d123..7bf1137 100644
--- a/sommitrealweird/photo/models.py
+++ b/sommitrealweird/photo/models.py
@@ -16,6 +16,9 @@ class Album(models.Model):
     def __unicode__(self):
         return "%s" %(self.name,)
 
+    def __str__(self):
+        return "%s" %(self.name,)
+
 def get_upload_path(photo, filename):
     if photo.album:
         return "photos/%s/%s" %(photo.album.slug, filename)
@@ -26,7 +29,10 @@ class Photo(models.Model):
     image = models.ImageField(upload_to=get_upload_path, null=True)
     caption = models.TextField(blank=True, null=True)
     order = models.IntegerField(blank=True, null=True)
-    album = models.ForeignKey('Album')
+    album = models.ForeignKey('Album', on_delete=models.CASCADE)
 
     def __unicode__(self):
         return "%s" %(self.image)
+
+    def __str__(self):
+        return "%s" %(self.image)