Django:使用原始查询来限制ModelForm中的foreignkey选择域

时间:2012-08-16 14:44:39

标签: django forms foreign-keys choicefield

以经典的foo吧为例:

models.py

Class Foo(models.Model):
    name = models.CharField(max_length= 200)

Class Bar(models.Model):
    name = models.CharField(max_length= 200)
    foo = models.ForeignKey('Foo')

在我的表单中,我尝试使用原始查询将foreignkeyFoo的选项限制为Foo的子集。

forms.py

class BarForm(ModelForm):
    search_field = CharField(max_length=100, required=False)

    def __init__(self,*args,**kwargs):
        search_str = kwargs.pop('search_str', None)
        super(BarForm,self ).__init__(*args,**kwargs)
        self.fields['search_field'].initial = search_str
        self.fields['foo'].queryset = Bar.objects.raw("""
select f.id as id, f.name as name from bar_lookup(%s)""", [search_str])

    class Meta:
        model = Bar
        exclude = ('foo',)

bar_lookup(%s)是一个返回表的DB过程。它搜索多个关系并以优化的方式处理结果的过滤和排序。它可以工作,我宁愿不用再次在Django代码中编写代码。

我收到以下错误:"'RawQuerySet' object has no attribute 'all'"。 如果我使用普通Bar.objects.filter()代替,则表单有效。

我应该如何将RawQuerySet转换为普通QuerySet? 我应该使用self.fields['line_stop'].choice选项吗?

1 个答案:

答案 0 :(得分:1)

我遇到了类似的问题,并提出了一个棘手的解决方案:

class BarForm(ModelForm):
    search_field = CharField(max_length=100, required=False)

    def __init__(self,*args,**kwargs):
        search_str = kwargs.pop('search_str', None)
        super(BarForm,self ).__init__(*args,**kwargs)
        self.fields['search_field'].initial = search_str
        self.foo_choices = Bar.objects.raw("""
              select f.id as id, f.name as name from bar_lookup(%s)""", [search_str])
        self.fields['foo'].choices = [(x.id, x) for x in self.foo_choices]

    def clean_foo(self):
        foo = self.cleaned_data['foo']
        if foo not in self.foo_choices:
             raise forms.ValidationError("Some error!")
        return foo

    class Meta:
        model = Bar

我知道它并不完美,继承ModelChoiceField会更好。