在ModelChoiceField中分隔选择

时间:2010-07-04 16:53:40

标签: python django django-forms

我是python和django的新手所以这个问题可能很容易解决,但我无法让它工作。 基本上我有一个模型,其中包含两个用户类型的外键。我正在构建一个表单,我想在其中删除基于另一个字段的ModelChoiceField的一个选项。我希望用户无法在表单中选择自己。 我的代码。

Models.py

from django.db import models
from django.contrib.auth.models import User

class Deuda(models.Model):
    ESTADOS = (("S", "Sin pagar"), ("P", "Pagada"), ("PC", "Pagada (Confirmada)"))

    propietario = models.ForeignKey(User, verbose_name="Propietario", related_name="propietario_deuda")
    adeudado = models.ForeignKey(User, verbose_name="Adeudado", related_name="adeudado_deuda")
    cantidad = models.FloatField("Cantidad")
    descripcion = models.TextField("Descripcion")
    estado = models.CharField(max_length=2, choices=ESTADOS)

    def __str__(self):
        return self.descripcion

forms.py

from django import forms
from housemanager.iou.models import Deuda
from django.contrib.auth.models import User

class DeudaForm(forms.ModelForm):

    class Meta:
        model = Deuda
        exclude = ('propietario',)

    def __init__(self, propietario):
        d = Deuda()
        d.propietario = propietario
        forms.ModelForm.__init__(self, instance=d)

    adeudado = forms.ModelChoiceField(queryset=User.objects.exclude(pk=d.propietario.pk))

views.py

的摘录
[...]
form = DeudaForm(request.user)

所以我的想法是将用户传递给表单,以便它可以定义一个不包含它的ModelChoiceField。 我的代码不起作用,因为在设置'adedado'时d不在范围内。我确信有很多方法可以实现这个目标,随时可以改变我的设计。

最终代码:

from django import forms
from housemanager.iou.models import Deuda
from django.contrib.auth.models import User

class DeudaForm(forms.ModelForm):

    class Meta:
        model = Deuda
        exclude = ('propietario',)

    def __init__(self, propietario):
        d = Deuda()
        d.propietario = propietario
        forms.ModelForm.__init__(self, instance=d)
        self.fields['adeudado'].queryset = User.objects.exclude(pk=propietario.pk)

1 个答案:

答案 0 :(得分:1)

尝试将其放在表格__init__

class DeudaForm(forms.ModelForm):

    class Meta:
        model = Deuda
        exclude = ('propietario',)

    def __init__(self, propietario):
        forms.ModelForm.__init__(self)
        self.fields['adeudado'].queryset = User.objects.exclude(pk=d.propietario.pk)