From b39cc1baf5e9a3b87cb0c71ac591e654c80c424f Mon Sep 17 00:00:00 2001
From: Minigrim0 <grimauflorent@gmail.com>
Date: Tue, 13 Apr 2021 17:44:42 +0200
Subject: [PATCH] [Fix] multiple category edit

---
 ctf/views.py | 13 +++++--------
 1 file changed, 5 insertions(+), 8 deletions(-)

diff --git a/ctf/views.py b/ctf/views.py
index 552c94e..fae859c 100644
--- a/ctf/views.py
+++ b/ctf/views.py
@@ -32,18 +32,15 @@ def challenges(request, ctf_id):
 
 @login_required
 def edit_chall(request, ctf_id, chall_id):
+
+    challenge = get_object_or_404(Challenge, id=chall_id)
     if request.method == "POST":
-        form = ChallengeForm(request.POST)
+        form = ChallengeForm(request.POST, instance=challenge)
         if form.is_valid:
-            challenge = Challenge.objects.get(id=chall_id)
-            challenge.name = request.POST['name']
-            challenge.points = request.POST['points']
-            challenge.description = request.POST['description']
-            challenge.category.set(request.POST['category'])
-            challenge.save()
+            challenge = form.save()
         return HttpResponseRedirect(reverse("chal", kwargs={"ctf_id": ctf_id}))
 
-    form = ChallengeForm(instance=Challenge.objects.get(id=chall_id))
+    form = ChallengeForm(instance=challenge)
     context = {
         "EditChallenge": form,
         "ctf_id": ctf_id,