diff --git a/django_web_galleries/web_galleries/templates/global.html b/django_web_galleries/web_galleries/templates/global.html
index e96ef43..aa6fbb5 100644
--- a/django_web_galleries/web_galleries/templates/global.html
+++ b/django_web_galleries/web_galleries/templates/global.html
@@ -24,6 +24,11 @@
Upload image
+
+
+ My images
+
+
My galleries
diff --git a/django_web_galleries/web_galleries/urls.py b/django_web_galleries/web_galleries/urls.py
index c252d23..5f4897c 100644
--- a/django_web_galleries/web_galleries/urls.py
+++ b/django_web_galleries/web_galleries/urls.py
@@ -2,10 +2,12 @@ from django.urls import path
from .views import (
ImageUploadView,
- PublicImageListView
+ PublicImageListView,
+ MyImagesListView
)
urlpatterns = [
path('', PublicImageListView.as_view(), name='home'),
- path('upload/', ImageUploadView.as_view(), name='upload_image')
+ path('upload/', ImageUploadView.as_view(), name='upload_image'),
+ path('my_images/', MyImagesListView.as_view(), name='my_images')
]
diff --git a/django_web_galleries/web_galleries/views.py b/django_web_galleries/web_galleries/views.py
index 1df280c..f18792f 100644
--- a/django_web_galleries/web_galleries/views.py
+++ b/django_web_galleries/web_galleries/views.py
@@ -15,7 +15,7 @@ class VisitorSessionMixin(View):
def get_visitor(self):
if self.request.session:
if 'visitor_session' in self.request.session:
- if Visitor.objects.get(session_id=self.request.session['visitor_session']).exists():
+ if Visitor.objects.filter(session_id=self.request.session['visitor_session']).exists():
return Visitor.objects.get(session_id=self.request.session['visitor_session'])
else:
visitor = Visitor.objects.create()
@@ -72,3 +72,13 @@ class PublicImageListView(ListView):
def get_queryset(self):
return Image.objects.all().filter(private=False)
+
+class MyImagesListView(VisitorSessionMixin, ListView):
+ model = Image
+ paginated_by = 20
+ context_object_name = 'images'
+ template_name = 'list_images/image_list.html'
+
+ def get_queryset(self):
+ return Image.objects.all().filter(uploaded_by=self.get_visitor())
+
\ No newline at end of file