对于您的用例,admin action 可能更合适。
用户将在更改列表页面上选择他们想要修改的行,然后选择要对这些行执行的操作。举例说明:
假设您希望能够在 django 管理员中选择员工并将他们标记为“休假”或“休假”。假期模式可能如下所示:
# models.py
class Employee(models.Model):
name = models.CharField(max_length=255)
age = models.IntegerField()
def is_on_vacation(self):
""" Returns true if the employee is on vacation, otherwise false. """
return self.vacation_set.filter(end_date__isnull=True).exists()
class Vacation(models.Model):
employee = models.ForeignKey("Employee")
start_date = models.DateTimeField(auto_now_add=True)
end_date = models.DateTimeField(blank=True, null=True)
在admin.py 中,我们将为员工设置一个开始和结束假期的操作:
# admin.py
from django.utils import timezone
class EmployeeAdmin(admin.ModelAdmin):
# Adding is_on_vacation here will show a true/false column in the change list
# that displays the output of the Employee.is_on_vacation method.
list_display = ("name", "age", "is_on_vacation")
def start_vacation(self, request, queryset):
for employee in queryset:
Vacation.objects.create(employee=employee)
self.message_user(
request,
f"Successfully started vacations for {len(queryset)} employees",
)
start_vacation.short_description = "Start vacations for selected employees"
def end_vacation(self, request, queryset):
for employee in queryset:
vacation = employee.vacation_set.filter(end_date__isnull=True).first()
vacation.end_date = timezone.now()
vacation.save()
self.message_user(
request,
f"Successfully ended vacations for {len(queryset)} employees",
)
end_vacation.short_description = "End vacations for selected employees"
现在在 django admin 中,当您选择员工并选择开始/结束休假操作时,上述方法将更新数据库中的对象,并且更改列表视图将显示 is_on_vacation 列的更新值。
注意:上面的示例代码不执行任何错误检查,应该进行优化以进行批量查询。