Django - 一对一模型管理

Question

我对 django 相当熟练，并尝试在 Intranet 项目中使用模型表单。

本质上，我有一个“资源”模型，由另一个团队填充。第二种模型是“Intake”，用户提交资源请求。它与资源具有一对一的映射。

目标是每次摄入仅允许 1 次资源分配。

现在，摄入模型表单显示了该表单，其中有一个资源下拉字段，但需要注意的是，它显示了所有资源，无论以前是否分配。

前任。如果资源被摄入占用，保存按钮检测到不允许保存。这是预期的，但下拉列表首先不应显示该资源。

我该如何做到这一点，即不显示已分配的资源？

class Resource(models.Model):
    label = models.CharField(max_length=50, primary_key=True)
    created = models.DateTimeField(auto_now_add=True)
    created_by = models.ForeignKey('auth.User', default=1)

    class Meta:
        verbose_name_plural = "Resource Pool"

    def __str__(self):
        return self.label

class Intake(models.Model):
    order_id = models.AutoField(primary_key=True)
    requestor = models.ForeignKey('auth.User', default=1)
    resource = models.OneToOneField(Resource, verbose_name="Allocation")
    project = models.CharField(max_length=50)

    class Meta:
        verbose_name_plural = "Environment Request"

    def __str__(self):
        print("self called")
        return self.project

Answer 1

您可以在管理员中创建自定义表单并更改字段的查询集值resource。像这样的东西：

管理员.py

from django import forms
from django.db.models import Q

from .models import Intake

class IntakeForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super(IntakeForm, self).__init__(*args, **kwargs)
        self.fields['resource'].queryset = Resource.objects.filter(
            Q(intake__isnull=True) | Q(intake=self.instance)
        )

class IntakeAdmin(admin.ModelAdmin):
    form = IntakeForm

admin.site.register(Intake, IntakeAdmin)