From 40d7e0dfc4a02d80d56379d2ee97019a27b0dc96 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bodor=20M=C3=A1t=C3=A9?= <bodor.mate@kszk.bme.hu>
Date: Thu, 3 Jan 2019 21:28:59 +0100
Subject: [PATCH] Bug fix get_queryset, change user to profile
---
src/homework/views.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/homework/views.py b/src/homework/views.py
index 41f1638..d0762cc 100755
--- a/src/homework/views.py
+++ b/src/homework/views.py
@@ -18,10 +18,10 @@ class SolutionsViewSet(viewsets.ModelViewSet):
def get_queryset(self):
user = self.request.user
- queryset = models.Solution.objects.filter(created_by=user)
+ queryset = models.Solution.objects.filter(created_by=user.profile)
if user.has_perm(permissions.IsStaffUser):
queryset = models.Solution.objects.all()
- user_id = self.request.query_params.get('userID', None)
- if user_id is not None:
- queryset = queryset.filter(created_by=user_id)
+ profile_id = self.request.query_params.get('profileID', None)
+ if profile_id is not None:
+ queryset = queryset.filter(created_by=profile_id)
return queryset
--
GitLab