假设我在我的models.py中有以下内容:

class Company(models.Model):
   name = ...

class Rate(models.Model):
   company = models.ForeignKey(Company)
   name = ...

class Client(models.Model):
   name = ...
   company = models.ForeignKey(Company)
   base_rate = models.ForeignKey(Rate)

也就是说,有多个公司,每个公司都有一系列的费率和客户。每个客户都应该有一个从其母公司的利率中选择的基础利率,而不是其他公司的利率。

当创建一个添加客户端表单时,我想删除公司选项(因为已经通过公司页面上的“添加客户端”按钮进行了选择),并将费率选项限制为该公司。

在Django 1.0中如何做到这一点?

我目前的forms.py文件只是一个样板文件:

from models import *
from django.forms import ModelForm

class ClientForm(ModelForm):
    class Meta:
        model = Client

views.py也是基本的:

from django.shortcuts import render_to_response, get_object_or_404
from models import *
from forms import *

def addclient(request, company_id):
    the_company = get_object_or_404(Company, id=company_id)

    if request.POST:
        form = ClientForm(request.POST)
        if form.is_valid():
            form.save()
            return HttpResponseRedirect(the_company.get_clients_url())
    else:
        form = ClientForm()

    return render_to_response('addclient.html', {'form': form, 'the_company':the_company})

在Django 0.96中,我能够通过在渲染模板之前做如下的事情来破解这个:

manipulator.fields[0].choices = [(r.id,r.name) for r in Rate.objects.filter(company_id=the_company.id)]

ForeignKey。看起来很有前途,但我不知道如何在公司里过关。id和我不清楚,如果这将工作以外的管理界面无论如何。

谢谢。(这似乎是一个非常基本的要求,但如果我应该重新设计一些东西,我愿意听取建议。)


当前回答

ForeignKey由django.forms表示。ModelChoiceField是一个ChoiceField,它的选项是一个模型QuerySet。请参阅ModelChoiceField的参考。

因此,为字段的QuerySet属性提供一个QuerySet。这取决于你的表单是如何构建的。如果构建显式表单,则会直接为字段命名。

form.rate.queryset = Rate.objects.filter(company_id=the_company.id)

如果您使用默认的ModelForm对象,form.fields["rate"]。Queryset =…

这是在视图中显式完成的。不要到处乱黑。

其他回答

除了S.Lott的回答和评论中提到的becomingGuru,还可以通过覆盖ModelForm来添加查询集过滤器。__init__函数。(这可以很容易地应用于常规表单)它可以帮助重用并保持视图函数的整洁。

class ClientForm(forms.ModelForm):
    def __init__(self,company,*args,**kwargs):
        super (ClientForm,self ).__init__(*args,**kwargs) # populates the post
        self.fields['rate'].queryset = Rate.objects.filter(company=company)
        self.fields['client'].queryset = Client.objects.filter(company=company)

    class Meta:
        model = Client

def addclient(request, company_id):
        the_company = get_object_or_404(Company, id=company_id)

        if request.POST:
            form = ClientForm(the_company,request.POST)  #<-- Note the extra arg
            if form.is_valid():
                form.save()
                return HttpResponseRedirect(the_company.get_clients_url())
        else:
            form = ClientForm(the_company)

        return render_to_response('addclient.html', 
                                  {'form': form, 'the_company':the_company})

这对于重用很有用,比如如果在许多模型上需要通用的过滤器(通常我声明一个抽象的Form类)。如。

class UberClientForm(ClientForm):
    class Meta:
        model = UberClient

def view(request):
    ...
    form = UberClientForm(company)
    ...

#or even extend the existing custom init
class PITAClient(ClientForm):
    def __init__(company, *args, **args):
        super (PITAClient,self ).__init__(company,*args,**kwargs)
        self.fields['support_staff'].queryset = User.objects.exclude(user='michael')

除此之外,我只是在重申Django博客材料,那里有很多好的材料。

ForeignKey由django.forms表示。ModelChoiceField是一个ChoiceField,它的选项是一个模型QuerySet。请参阅ModelChoiceField的参考。

因此,为字段的QuerySet属性提供一个QuerySet。这取决于你的表单是如何构建的。如果构建显式表单,则会直接为字段命名。

form.rate.queryset = Rate.objects.filter(company_id=the_company.id)

如果您使用默认的ModelForm对象,form.fields["rate"]。Queryset =…

这是在视图中显式完成的。不要到处乱黑。

在运行时(例如在CreateView中)限制ModelForm的ForeignKey字段的选择的一个好方法是通过覆盖视图中的get_form_class()来为base_fields['field_name']设置limit_choices_to。

例如,在创建客户端时,将费率的选择限制为URL中标识的公司:

class ClientCreateView(LoginRequired, CreateView):
    model = Client
    fields = '__all__'
    
    def get_form_class(self):
        modelform = super().get_form_class()
        modelform.base_fields['rate'].limit_choices_to = {'company': self.kwargs['company']}
        return modelform

如果你还没有创建表单,想要更改查询集,你可以这样做:

formmodel.base_fields['myfield'].queryset = MyModel.objects.filter(...)

这在使用通用视图时非常有用!

要做到这一点,用一个通用视图,如CreateView…

class AddPhotoToProject(CreateView):
    """
    a view where a user can associate a photo with a project
    """
    model = Connection
    form_class = CreateConnectionForm


    def get_context_data(self, **kwargs):
        context = super(AddPhotoToProject, self).get_context_data(**kwargs)
        context['photo'] = self.kwargs['pk']
        context['form'].fields['project'].queryset = Project.objects.for_user(self.request.user)
        return context
    def form_valid(self, form):
        pobj = Photo.objects.get(pk=self.kwargs['pk'])
        obj = form.save(commit=False)
        obj.photo = pobj
        obj.save()

        return_json = {'success': True}

        if self.request.is_ajax():

            final_response = json.dumps(return_json)
            return HttpResponse(final_response)

        else:

            messages.success(self.request, 'photo was added to project!')
            return HttpResponseRedirect(reverse('MyPhotos'))

最重要的是…

    context['form'].fields['project'].queryset = Project.objects.for_user(self.request.user)

,在这里阅读我的文章