From 29c95ebb89985417162c60a4f6fda2fe26cb8d79 Mon Sep 17 00:00:00 2001
From: Yoan Le Clanche <yoan@ellington.pilotsystems.net>
Date: Mon, 14 Oct 2019 18:08:14 +0200
Subject: [PATCH] trainings are now filtered on platform_only field on register
 page : https://trackers.pilotsystems.net/probarreau/0552

---
 teleforma/forms.py                            |  2 +-
 ...period_date_inscription_start__add_fie.py} |  0
 .../registration/registration_form.html       | 40 +++++++------------
 teleforma/views/crfpa.py                      |  4 +-
 4 files changed, 19 insertions(+), 27 deletions(-)
 rename teleforma/migrations/{0116_auto__add_parameters__add_field_period_date_inscription_start__add_fie.py => 0117_auto__add_parameters__add_field_period_date_inscription_start__add_fie.py} (100%)

diff --git a/teleforma/forms.py b/teleforma/forms.py
index b006f816..7ce4fb48 100644
--- a/teleforma/forms.py
+++ b/teleforma/forms.py
@@ -162,7 +162,7 @@ class UserForm(ModelForm):
                           training=data.get('training')
                           )
         student.save()
-        # student.trainings.add(*data.get('trainings', []))
+        student.trainings.add(data.get('training', None))
         return user
 
 
diff --git a/teleforma/migrations/0116_auto__add_parameters__add_field_period_date_inscription_start__add_fie.py b/teleforma/migrations/0117_auto__add_parameters__add_field_period_date_inscription_start__add_fie.py
similarity index 100%
rename from teleforma/migrations/0116_auto__add_parameters__add_field_period_date_inscription_start__add_fie.py
rename to teleforma/migrations/0117_auto__add_parameters__add_field_period_date_inscription_start__add_fie.py
diff --git a/teleforma/templates/registration/registration_form.html b/teleforma/templates/registration/registration_form.html
index 9a5b152d..0bd5d49c 100644
--- a/teleforma/templates/registration/registration_form.html
+++ b/teleforma/templates/registration/registration_form.html
@@ -74,42 +74,32 @@
         </form>
     </center>
     <script>
-        $(document).ready(function () {
-            var id = $("#id_period").val();
-            var training_id = $("#id_training").val();
 
-            if (!id) {
+        var trainingId;
+
+        function updateTrainings(){
+            // filter trainings according to period
+            var periodId = $("#id_period").val();
+            var platformOnly = $("#id_platform_only").val();
+            if (!periodId || !platformOnly) {
                 $('#id_training').html('<option value="" selected="selected">---------</option>');
             } else {
                 $.ajax({
-                    url: "/update-training/" + id + "/",
-                    data: {"period_id": id, "training_id": training_id},
+                    url: "/update-training/" + periodId + "/",
+                    data: {"period_id": periodId, "training_id": trainingId, "platform_only":platformOnly},
                     dataType: "html",
                     type: "post",
                     success: function (data) {
-                        $('#id_student').html(data);
+                        $('#id_training').html(data);
                     }
                 });
             }
+        }
+        $(document).ready(function () {
+            trainingId =  $("#id_training").val();
+            updateTrainings();
+            $("#id_period,  #id_platform_only").change(updateTrainings);
 
-            $("#id_period").change(function () {
-                period_id = this.value
-                if (!period_id) {
-                    $('#id_training').html('<option value="" selected="selected">---------</option>')
-                } else {
-                    $.ajax({
-                        url: "/update-training/" + period_id + "/",
-                        data: {"period_id": period_id},
-                        dataType: "html",
-                        type: "post",
-                        success: function (data) {
-                            $('#id_training').html(data);
-                        }
-                    });
-                }
-            });
-
-            // filter trainings according to
         });
     </script>
     <style>
diff --git a/teleforma/views/crfpa.py b/teleforma/views/crfpa.py
index 593a8d2e..d23d4044 100644
--- a/teleforma/views/crfpa.py
+++ b/teleforma/views/crfpa.py
@@ -569,8 +569,10 @@ class RegistrationPDFViewDownload(RegistrationPDFView):
 
 @csrf_exempt
 def update_training(request, id):
-    trainings = Training.objects.filter(period__id=id)
+    platform_only = request.POST.get('platform_only', "") == 'True' and True or False
+    trainings = Training.objects.filter(period__id=id, platform_only=platform_only)
     training_id = request.POST.get("training_id", "")
+
     html = '<option value="" selected="selected">---------</option>'
     for training in trainings:
         if training_id == str(training.pk):
-- 
2.39.5