From 16c53240515f46bfb2d843cbbf9a5a1f3dd2e32b Mon Sep 17 00:00:00 2001
From: Ross <xkjq@ymail.com>
Date: Sat, 11 Dec 2021 23:55:43 +0000
Subject: [PATCH] .

---
 generic/templates/generic/cid_view.html | 18 +++++++++++++-----
 generic/views.py                        |  2 ++
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/generic/templates/generic/cid_view.html b/generic/templates/generic/cid_view.html
index 4aa26f65..0057321a 100644
--- a/generic/templates/generic/cid_view.html
+++ b/generic/templates/generic/cid_view.html
@@ -17,10 +17,9 @@
     {% render_table table %}
 
     <details>
-        <summary>Add new CID users</summary>
-        <button id="add-new-cids">Add New</button>
-        <input type="number" id="add-number" value="number to add">
-
+        <summary><h3>Add/Edit CID users</h3></summary>
+        <button id="add-selected-cids">Add to selected</button>
+        <p>Select exams below to add user to.</p>
         <div class="exam-selectors">
             <span>
                 Physics Exams<br/>
@@ -67,6 +66,10 @@
 
                 </select>
             </span>
+
+        <button id="add-new-cids">Add New</button>
+        <input type="number" id="add-number" value="number to add">
+
         </div>
     </details>
 
@@ -82,7 +85,12 @@
                     url: "{% url 'generic:create_cid_users' %}",
                     data: {
                         csrfmiddlewaretoken: "{{ csrf_token }}",
-                        number: $("#add-number").get(0).value
+                        number: $("#add-number").get(0).value,
+                        physics_exams: $("#physics-exams").val(),
+                        rapid_exams: $("#rapid-exams").val(),
+                        sbas_exams: $("#sbas-exams").val(),
+                        longs_exams: $("#longs-exams").val(),
+                        anatomy_exams: $("#anatomy-exams").val(),
                     },
                     type: "POST",
                     dataType: "json",
diff --git a/generic/views.py b/generic/views.py
index 4d3ee729..2eb87260 100644
--- a/generic/views.py
+++ b/generic/views.py
@@ -954,12 +954,14 @@ class CidUserView(LoginRequiredMixin, SingleTableMixin, FilterView):
 def create_cid_users(request):
     if request.is_ajax() and request.method == "POST":
         number = int(request.POST.get("number"))
+        physics_exams = request.POST.get("physics_exams")
 
         new_cid = max(CidUser.objects.all().order_by("-cid")[0].cid + 1, 50000)
 
         for n in range(number):
             passcode = "".join(random.choices(string.ascii_uppercase, k=4))
             c = CidUser(cid=new_cid, passcode=passcode)
+            c.physics_exams.set(physics_exams)
             c.save()
             new_cid = new_cid + 1