想从 ModelForm 中排除一些数据

Question

我尝试使用仅链接到当前基础对象的 objectl 来制作表单（从他们的模型中排除其他对象）：

forms.py

class RoomTypeForm(ModelForm):
    class Meta:
        model = RoomType
        fields = {'Name', 'Rooms', 'Beds', 'Facilities', 'Capacity'}
        exclude = ('Building',)

    def __init__(self, *args, **kwargs):
        self.building_id = kwargs.pop('building_id')
        self.fields['Capacity'].queryset = Capacity.objects.filter(
                                           Building=self.building_id
                                           )

views.py

def building_details(request, hotel_id, building_id):
    capacity_form = CapacityForm
    roomtype_form = RoomTypeForm
    args = {}
    args.update(csrf(request))
    args['building'] = Building.objects.get(id=building_id)
    args['capacity'] = Capacity.objects.filter(Building=building_id)
    args['roomtypes'] = RoomType.objects.filter(Building=building_id)
    args['capform'] = capacity_form
    args['rtform'] = roomtype_form(building_id=building_id)
    return render_to_response('building.html', args)

但我有错误：

“RoomTypeForm”对象没有“字段”属性

排队：

self.fields['容量'].queryset = 容量.objects.filter(Building=self.building_id)

如何才能使所有这些工作？请帮忙。

Answer 1

你需要调用__init__中的super()类

def __init__(self, *args, **kwargs):
    self.building_id = kwargs.pop('building_id')
    super(RoomTypeForm, self).__init__(*args, **kwargs)
    self.fields['Capacity'].queryset = Capacity.objects.filter(Building=self.building_id)

或者如果你使用的是 python3，你可以这样做：

super().__init__(*args, **kwargs)

另一个注意事项，在fields 中指定列表或元组是标准做法。你指定了一个集合。