Django循环中实现按钮样式切换：可用时绿，已选课则红

嗨，我来帮你搞定这个按钮颜色切换的需求～ 先看看你现有代码的小问题：目前只有当学生匹配到classList里week=14的记录时才会显示红色按钮，但没匹配到的情况就不会输出按钮，而且循环classList还可能导致同一个学生出现多个按钮（如果有重复记录的话）。下面给你两种可行的解决方案，优先推荐第一种，因为更符合Django“尽量把逻辑放在视图，模板只负责渲染”的最佳实践。

方案一：视图预处理（推荐） #

我们可以在视图里提前给每个学生标记是否已选14周的课程，这样模板里直接用这个标记判断就行，逻辑更清晰，性能也更好。

步骤1：修改视图代码 #

假设你的视图里原本是获取studentList和classList，现在调整成这样：

# 在你的视图函数中
from django.db.models import Q

# 获取所有学生
studentList = Student.objects.all()
# 筛选出week=14的选课记录（如果classList原本就是week=14的，可以跳过这步）
week14_classes = Class.objects.filter(week=14)
# 提取已选课的学生ID列表
enrolled_student_ids = week14_classes.values_list('student_id', flat=True)

# 给每个学生添加一个标记属性
for student in studentList:
    student.is_enrolled_week14 = student.id in enrolled_student_ids

步骤2：修改模板代码 #

现在模板里直接根据标记切换按钮样式，不用再循环classList了：

{% for student in studentList %}
<tr>
    <td>{{ student.firstname }}</td>
    <td>{{ student.lastname }}</td>
    <td>{{ student.teacher }}</td>
    <td>
        {% if student.is_enrolled_week14 %}
            <a href="#" class="btn btn-danger" role="button">Not Available</a>
        {% else %}
            <a href="#" class="btn btn-success" role="button">Available</a>
        {% endif %}
    </td>
</tr>
{% endfor %}

方案二：纯模板内处理（不推荐，需额外依赖） #

如果不想修改视图，只想在模板里处理，需要用到Django模板的扩展标签（比如{% set %}和{% break %}），不过默认Django没有这些标签，你需要先安装django-template-extensions包。

修改后的模板代码：

{% load template_extras %} {# 加载扩展标签 #}
{% for student in studentList %}
<tr>
    <td>{{ student.firstname }}</td>
    <td>{{ student.lastname }}</td>
    <td>{{ student.teacher }}</td>
    <td>
        {% set is_enrolled = False %}
        {% for class in classList %}
            {% if student.id == class.student_id and class.week == 14 %}
                {% set is_enrolled = True %}
                {% break %} {# 找到匹配项就跳出循环，避免重复判断 #}
            {% endif %}
        {% endfor %}
        {% if is_enrolled %}
            <a href="#" class="btn btn-danger" role="button">Not Available</a>
        {% else %}
            <a href="#" class="btn btn-success" role="button">Available</a>
        {% endif %}
    </td>
</tr>
{% endfor %}

这个方案的缺点是模板逻辑变复杂了，而且需要额外安装依赖，所以还是优先用方案一哦～

内容的提问来源于stack exchange，提问作者user9527415