]> git.sommitrealweird.co.uk Git - sommitrealweird.git/blobdiff - sommitrealweird/photo/views.py
Update to work with newer django and python3
[sommitrealweird.git] / sommitrealweird / photo / views.py
index fea7dd3750785654e2728d88d2a15771d1dadf87..75b1f73c21d91414863af3c8f96b79bbcd3bd54a 100644 (file)
@@ -1,6 +1,6 @@
 from django.views.generic import ListView, DetailView
 from django.http import Http404
-from photo.models import Album, Photo
+from sommitrealweird.photo.models import Album, Photo
 
 class AlbumListView(ListView):
     model = Album
@@ -10,12 +10,14 @@ class AlbumListView(ListView):
 class PhotoListView(ListView):
     model = Photo
     paginate_by = 20
-    order_by = 'order, image'
+
+    def order_by(self):
+        return self.ordering
 
     def get_queryset(self):
         try:
             album = Album.objects.get(slug__exact=self.kwargs['slug'])
-            return Photo.objects.filter(album=album)
+            return Photo.objects.filter(album=album).order_by('order', 'image')
         except:
             raise Http404