快捷導(dǎo)航

Django表單外鍵選項初始化的問題及解決方法

更新時間：2021年04月29日 11:13:27 作者：BH78

這篇文章主要介紹了Django表單外鍵選項初始化的問題及解決方法,需本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，要的朋友可以參考下

問題描述

先說明一下問題的由來：
Django的模型中經(jīng)常會用ForeignKey來關(guān)聯(lián)其他表格數(shù)據(jù)

class MeasureTask(models.Model):
    taskname = models.CharField(max_length=LEN_FULLNAME, verbose_name="任務(wù)名稱")
    road = models.ForeignKey(Road, on_delete=models.CASCADE, verbose_name="設(shè)計路段")
    # 路面層,附加一個參數(shù) ,指定這個層的厚度,相對于底層的厚度
    # road_level = models.ForeignKey(RoadLevel, on_delete=models.CASCADE, verbose_name="路面層")
    level_thick = models.IntegerField(default=0, verbose_name="層厚(mm)")
    # ...

使用Django的ModelForm轉(zhuǎn)化為表單代碼如下：

class MeasureTaskNewForm(forms.ModelForm):
    class Meta:
        model = MeasureTask
        fields = ('taskname', 'staff', 'start_mileage', 'end_mileage',
                  'road', 'level_thick', 'step', 'equip', 'comment')

如果不做進一步處理，在網(wǎng)頁中使用這個From時，關(guān)聯(lián)字段會自動轉(zhuǎn)化為一個select控件，里面包含了所有選項，如下圖：

實際應(yīng)用時，需要對關(guān)聯(lián)的字段做一些選擇過濾。期望的結(jié)果如下：

解決方式

在From類中設(shè)置一個初始化函數(shù)：

class MeasureTaskNewForm(forms.ModelForm):
    class Meta:
        model = MeasureTask
        fields = ('taskname', 'staff', 'start_mileage', 'end_mileage',
                  'road', 'level_thick', 'step', 'equip', 'comment')

    # 對參數(shù)作初始化設(shè)置，導(dǎo)致返回之后的Form驗證失敗
    def __init__(self, road_choices=None, *args, **kwargs):
        super(MeasureTaskNewForm, self).__init__(*args, **kwargs)
        if road_choices:
            self.fields['road'].choices = road_choices

應(yīng)用這個類的方式如下，注意傳入?yún)?shù)的數(shù)據(jù)類型，

 # 對關(guān)聯(lián)數(shù)據(jù)過濾
        roads = Road.objects.filter(project=p_item) 
        # 生成值，分別對應(yīng)于 html 中 select->option 設(shè)置
        choices = roads.values_list('id', 'name')
        dataform = MeasureTaskNewForm(road_choices=choices)
        # dataform = MeasureTaskNewFormShadow()
        return render(request, "mdata/html/measure_task_add.html", locals())

生成的html代碼：

<select name="road" id="id_road" class="form-control">
  <option value="1">北四環(huán)主線</option>
  <option value="5">匝道A</option>
</select>

到這里生成的表單頁面沒有問題了，但是表單提交返回時如果還是用這個From來接收Request數(shù)據(jù)，則會出現(xiàn)數(shù)據(jù)校驗失敗的問題

if request.method == "POST":
        dataform = MeasureTaskNewForm(request.POST)
        # 這里將出現(xiàn)校驗失敗的問題
        if dataform.is_valid():
            dataform.save()
        return redirect('mdata:measure_task', pid=p_item.id)

為了解決這個問題，另外做了一個沒有初始化函數(shù)的表單類來接收數(shù)據(jù).

# 影子表單模型
class MeasureTaskNewFormShadow(forms.ModelForm):
    class Meta:
        model = MeasureTask
        fields = ('taskname', 'staff', 'start_mileage', 'end_mileage',
                  'road', 'level_thick', 'step', 'equip', 'comment')

if request.method == "POST":
        dataform = MeasureTaskNewFormShadow(request.POST)
        if dataform.is_valid():
            dataform.save()
        return redirect('mdata:measure_task', pid=p_item.id)

感覺這里應(yīng)該有更好的方法，嘗試對Form的初始化函數(shù)做了一些修改，但是沒有成功。