我的目标是创建一个modelformset,其字段是另一个模型的外键。我想根据登录的用户限制外键字段中的下拉列表。
models.py
class WeeklyWinner(models.Model):
league = models.ForeignKey('League')
week = models.IntegerField()
manager = models.ForeignKey('Managers')
class League(models.Model):
league_name = models.CharField(max_length=30)
number_players = models.IntegerField()
league_fee = models.IntegerField()
commissioner = models.ForeignKey(User)
deadline_date = models.DateField(help_text='Date league fees are due')
fine = models.IntegerField(help_text = 'Amount you will charge fantasy managers that pay late', )
draft_date = models.DateField(blank=True, null=True, help_text='Not Required')
transaction_amount = models.IntegerField(help_text='Amount your league will charge for each transaction at the end of the season', )
payment_info = models.TextField(blank=True, help_text='Instruct your league managers on how to pay up. You can place in a paypal link, venmo or tell them to pay you straight cash.')
notes = models.TextField(blank=True, help_text='This information will be included in the email that will be sent to league managers. Provide any extra information on your league that you want them to know.')
number_weeks = models.IntegerField(blank=True, null=True, help_text='Number of weeks your league contains. This is used to determine the number of weekly winners.')
def __unicode__(self):
return self.league_name
class Managers(models.Model):
league = models.ForeignKey('League')
team_name = models.CharField(max_length=30)
first_name_1 = models.CharField(max_length=30)
last_name_1 = models.CharField(max_length=30)
email_1 = models.EmailField()
first_name_2 = models.CharField(max_length=30, null=True, blank=True)
last_name_2 = models.CharField(max_length=30, null=True, blank=True)
email_2 = models.EmailField(null=True, blank=True)
def __unicode__(self):
return self.team_name
forms.py
class WeeklyWinnerForm(ModelForm):
def __init__(self, *args, **kwargs):
self.user = kwargs.pop('user', None)
super(WeeklyWinnerForm, self).__init__(*args, **kwargs)
self.fields['manager'].queryset = Managers.objects.filter(league_id = 2)
print self.user
class Meta:
model = WeeklyWinner
fields = ['week', 'manager']
views.py
def weeklywinner(request):
league = get_object_or_404(League, commissioner=request.user)
weeklywinner = WeeklyWinner.objects.filter(league_id=league)
weeklywinnerformset = modelformset_factory(WeeklyWinner, form=WeeklyWinnerForm, max_num=10)
form_data = request.POST if request.method == 'POST' else None
form = weeklywinnerformset(data=form_data)
if form.is_valid():
form.save()
return redirect('/dashboard')
return render(request, 'settings.html', {'form': form})
在我的forms.py中,如果我将league_id设置为硬编码的数字,我会获得理想的结果。为了解决这个问题,我需要将request.user传递给我的表单/ formset但是尝试传递用户变量:
form = weeklywinnerformset(data=form_data, user=request.user)
我收到以下错误代码:
Environment:
Request Method: GET
Request URL: http://127.0.0.1:8000/weeklywinner/
Django Version: 1.5.8
Python Version: 2.7.6
Installed Applications:
('django.contrib.admin',
'django.contrib.auth',
'django.contrib.contenttypes',
'django.contrib.sessions',
'django.contrib.messages',
'django.contrib.staticfiles',
'django.contrib.sites',
'ledger',
'allauth',
'allauth.account',
'allauth.socialaccount',
'after_response')
Installed Middleware:
('django.contrib.sessions.middleware.SessionMiddleware',
'django.middleware.common.CommonMiddleware',
'django.middleware.csrf.CsrfViewMiddleware',
'django.contrib.auth.middleware.AuthenticationMiddleware',
'django.contrib.messages.middleware.MessageMiddleware',
'django.middleware.clickjacking.XFrameOptionsMiddleware')
Traceback:
File "/Library/Python/2.7/site-packages/django/core/handlers/base.py" in get_response
113. response = callback(request, *callback_args, **callback_kwargs)
File "/Users/tightflks/Documents/EAV/clients/fantasy/fantasy_development/ledger/views.py" in weeklywinner
379. form = weeklywinnerformset(data=form_data, user=request.user)
File "/Library/Python/2.7/site-packages/django/forms/models.py" in __init__
441. super(BaseModelFormSet, self).__init__(**defaults)
Exception Type: TypeError at /weeklywinner/
Exception Value: __init__() got an unexpected keyword argument 'user'
答案 0 :(得分:0)
Django内部在modelformset_factory中使用了modelform_factory函数,该函数不考虑任何覆盖方法或自定义字段,它只使用给定的Form类的Meta属性(在您的情况下为#34; WeeklyWinnerForm"),如字段,排除等,因此永远不会调用WeeklyWinnerForm的 init ,为此,您必须实现自定义的modelformset_factory。