diff --git a/django/library/views.py b/django/library/views.py index 5f7ad37..26757f7 100644 --- a/django/library/views.py +++ b/django/library/views.py @@ -21,11 +21,21 @@ def quiz_names_json(request): form = QuizFilterForm(request.GET) if form.is_valid(): search = form.cleaned_data.get('search') - names = list( - QivipQuiz.objects.all().annotate(max_amout_questions=Count('questions')).filter(max_amout_questions__gte=1, status='öffentlich', name__icontains=search) - .values_list('name', flat=True) - .distinct() -) + + + names = QivipQuiz.objects.all().annotate(max_amout_questions=Count('questions')).filter(max_amout_questions__gte=1, status='öffentlich', name__icontains=search) + + + if request.user.is_authenticated: + names = names | QivipQuiz.objects.filter( + user_id=request.user.id, + status='privat', + name__icontains=search + ) + + names=list(names.values_list('name', flat=True).distinct()) + + return JsonResponse(names, safe=False) def overview_quiz(request):