From 9b48ea9b7233d1c3ec5f253935deaec3b9b84f16 Mon Sep 17 00:00:00 2001
From: SKairinos <stefan.kairinos@ocado.com>
Date: Tue, 3 Dec 2024 15:39:50 +0000
Subject: [PATCH] disable worksheets

---
 .../game/python_den_level_selection.html      | 44 ++++++++++---------
 game/views/level_selection.py                 |  5 +++
 2 files changed, 29 insertions(+), 20 deletions(-)

diff --git a/game/templates/game/python_den_level_selection.html b/game/templates/game/python_den_level_selection.html
index 940d64748..9c6be6710 100644
--- a/game/templates/game/python_den_level_selection.html
+++ b/game/templates/game/python_den_level_selection.html
@@ -149,7 +149,18 @@ <h4>Introduction to Python</h4>
                             {% for level in episode.levels %}
                                 {% for worksheet in episode.worksheets %}
                                     {% if worksheet.before_level == level.id %}
-                                        {% if user|is_logged_in %}
+                                        {% if not user|is_logged_in or user.new_student.class_field in worksheet.locked_classes.all %}
+                                            {% if not user|is_logged_in %}
+                                                <p class="text--warning">In order to access the full content for this course, please log in.</p>
+                                            {% endif %}
+                                            <div class="col-sm-10 python-den-level">
+                                                <a class="disabled">{{episode.name}}</a>
+                                            </div>
+                                            <div class="col-sm-2 python-den-level">
+                                                <a class="button button--level button--icon disabled">
+                                                    Worksheet<span class="iconify" data-icon="mdi:open-in-new"></span></a>
+                                            </div>
+                                        {% else %}
                                             <div class="col-sm-7 python-den-level">
                                                 <a href="https://editor.raspberrypi.org/en/projects/blank-python-starter" target="_blank">{{episode.name}}</a>
                                             </div>
@@ -178,15 +189,6 @@ <h4>Introduction to Python</h4>
                                                         Worksheet<span class="iconify" data-icon="mdi:open-in-new"></span></a>
                                                 </div>
                                             {% endif %}
-                                        {% else %}
-                                            <p class="text--warning">In order to access the full content for this course, please log in.</p>
-                                            <div class="col-sm-10 python-den-level">
-                                                <a class="disabled">{{episode.name}}</a>
-                                            </div>
-                                            <div class="col-sm-2 python-den-level">
-                                                <a class="button button--level button--icon disabled">
-                                                    Worksheet<span class="iconify" data-icon="mdi:open-in-new"></span></a>
-                                            </div>
                                         {% endif %}
                                     {% endif %}
                                 {% endfor %}
@@ -220,7 +222,18 @@ <h4>Introduction to Python</h4>
                             {% endfor %}
                             {% for worksheet in episode.worksheets %}
                                 {% if not worksheet.before_level %}
-                                    {% if user|is_logged_in %}
+                                    {% if not user|is_logged_in or user.new_student.class_field in worksheet.locked_classes.all %}
+                                        {% if not user|is_logged_in %}
+                                            <p class="text--warning">In order to access the full content for this course, please log in.</p>
+                                        {% endif %}
+                                        <div class="col-sm-10 python-den-level">
+                                            <a class="disabled">{{episode.name}}{% if episode.worksheets|length > 1 %} pt. {{ forloop.counter }}{% endif %}</a>
+                                        </div>
+                                        <div class="col-sm-2 python-den-level">
+                                            <a class="button button--level button--icon disabled">
+                                                Worksheet<span class="iconify" data-icon="mdi:open-in-new"></span></a>
+                                        </div>
+                                    {% else %}
                                         <div class="col-sm-7 python-den-level">
                                             <a href="https://editor.raspberrypi.org/en/projects/blank-python-starter" target="_blank">{{episode.name}}{% if episode.worksheets|length > 1 %} pt. {{ forloop.counter }}{% endif %}</a>
                                         </div>
@@ -249,15 +262,6 @@ <h4>Introduction to Python</h4>
                                                     Worksheet<span class="iconify" data-icon="mdi:open-in-new"></span></a>
                                             </div>
                                         {% endif %}
-                                    {% else %}
-                                        <p class="text--warning">In order to access the full content for this course, please log in.</p>
-                                        <div class="col-sm-10 python-den-level">
-                                            <a class="disabled">{{episode.name}}{% if episode.worksheets|length > 1 %} pt. {{ forloop.counter }}{% endif %}</a>
-                                        </div>
-                                        <div class="col-sm-2 python-den-level">
-                                            <a class="button button--level button--icon disabled">
-                                                Worksheet<span class="iconify" data-icon="mdi:open-in-new"></span></a>
-                                        </div>
                                     {% endif %}
                                 {% endif %}
                             {% endfor %}
diff --git a/game/views/level_selection.py b/game/views/level_selection.py
index 5fea8ba32..9779757bd 100644
--- a/game/views/level_selection.py
+++ b/game/views/level_selection.py
@@ -251,6 +251,11 @@ def levels(request, language):
                     id=level["id"]
                 ).locked_for_class
 
+            for worksheet in episode["worksheets"]:
+                worksheet["locked_classes"] = Worksheet.objects.get(
+                    id=worksheet["id"]
+                ).locked_classes
+
         context["pythonEpisodes"] = python_episodes
 
     return context