2016-03-02 101 views
0

我在我的django應用程序中具有以下表單。我重寫了構造函數,以便限制顯示在「類」字段中的選項,以便只顯示學生註冊的類。無法指定 - 必須是實例

#forms.py 
class TestForm(forms.ModelForm): 

    def __init__(self, *args, **kwargs): 
     self.user = kwargs.pop('user', None) 
     super(TestForm, self).__init__(*args, **kwargs) 
     classes = [Classes.objects.filter(students__username=self.user.username)] 
     class_list = list() 
     for c in classes: 
      for single_class in c: 
       cl = Classes.objects.get(id=single_class.id) 
       class_list.append([cl, single_class]) 
     self.fields['classes'] = forms.ChoiceField(choices=class_list) 

    class Meta: 
     model = Test 
     fields = ('classes', 'name') 

但是,這樣做給了我一個無法分配「Class1234:測試類」 - 「Test.classes」必須是一個「類」的實例。當我打印class_list,我得到這個:

[[<Classes: Class1234: Test class>, <Classes: Class1234: Test class>],[<Classes: Testing: 2nd class>, <Classes: Testing: 2nd class>],[<Classes: 3rd: Class>, <Classes: 3rd: Class>]] 

對我來說,這看起來像實際的類的對象,所以我不明白爲什麼這是行不通的。作爲參考,這裏是我的模型和視圖:

#models.py 
class UserProfile(models.Model): 
    user = models.OneToOneField(User) 
    teacher = models.BooleanField(default=False) 

    def __str__(self): 
     return "{0} {1}".format(self.user.first_name, self.user.last_name) 


class Classes(models.Model): 
    classcode = models.CharField(max_length=15) 
    classname = models.TextField() 
    students = models.ManyToManyField(User) 

    def __str__(self): 
     return "{0}: {1}".format(self.classcode, self.classname) 


class Test(models.Model): 
    classes = models.ForeignKey(Classes, on_delete=models.CASCADE) 
    name = models.TextField(max_length=100) 
    points = models.ManyToManyField(User, default=0) 

    def __str__(self): 
     return "{0}".format(self.name) 

#views.py 
def add_test(request): 
    added = False 
    form = TestForm(user=request.user) 
    if request.method == "POST": 
     form = TestForm(request.POST, user=request.user) 
     if form.is_valid(): 
      form.save() 
      added = True 
    return render(request, 'site/addtest.html', {'form': form, 'added': added}) 

回答

1
[[<Classes: Class1234: Test class>, <Classes: Class1234: Test class>],[<Classes: Testing: 2nd class>, <Classes: Testing: 2nd class>],[<Classes: 3rd: Class>, <Classes: 3rd: Class>]] 

這是一個列表的列表,這可能是失敗的原因。但是你的問題無論如何都有一個更加優雅的解決方案。

不需要在__init__那樣做。只需在表單中覆蓋classes字段並定義查詢集。閱讀更多關於它here

如果這是一個應該應用於數據庫級別的限制,請考慮使用limit_choices_to。閱讀關於它here

+0

感謝您的幫助 - 正是我一直在尋找的東西 – user2151557

相關問題