Django 表單模型選擇框如何使用分組

更新時間：2019年05月16日 08:21:28 作者：棲遲於一丘

這篇文章主要介紹了Django 表單模型選擇框如何使用分組,小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

起步

Django 表單中有兩種字段類型可以使用選擇框: ChoiceField 和 ModelChoiceField 。

對于 ChoiceField 的基本使用是:

class ExpenseForm(forms.Form):
  CHOICES = (
    (11, 'Credit Card'),
    (12, 'Student Loans'),
    (13, 'Taxes'),
    (21, 'Books'),
    (22, 'Games'),
    (31, 'Groceries'),
    (32, 'Restaurants'),
  )
  date = forms.DateField()
  category = forms.ChoiceField(choices=CHOICES)

它能渲染出:

使用分組下拉框

還可以使用如下方式生成 <optgourp> 標簽：

class ExpenseForm(forms.Form):
  CHOICES = (
    ('Debt', (
      (11, 'Credit Card'),
      (12, 'Student Loans'),
      (13, 'Taxes'),
    )),
    ('Entertainment', (
      (21, 'Books'),
      (22, 'Games'),
    )),
    ('Everyday', (
      (31, 'Groceries'),
      (32, 'Restaurants'),
    )),
  )
  date = forms.DateField()
  category = forms.ChoiceField(choices=CHOICES)

能夠渲染為：

分組模型下拉框

如果使用的是 ModelChoiceField ，那抱歉，Django本身沒有提供解決方案。

在 https://code.djangoproject.com/ticket/27331 中提供了一個很好的解決方案。

首先為需要分類的類型創(chuàng)建模型，在另一個模型中用外鍵關(guān)聯(lián)它：

from django.db import models

class Category(models.Model):
  name = models.CharField(max_length=30)
  parent = models.ForeignKey('Category', on_delete=models.CASCADE, null=True)

  def __str__(self):
    return self.name

class Expense(models.Model):
  amount = models.DecimalField(max_digits=10, decimal_places=2)
  date = models.DateField()
  category = models.ForeignKey(Category, on_delete=models.CASCADE)

  def __str__(self):
    return self.amount

其次，創(chuàng)建一個新的表單 Field 類型：

from functools import partial
from itertools import groupby
from operator import attrgetter

from django.forms.models import ModelChoiceIterator, ModelChoiceField

class GroupedModelChoiceIterator(ModelChoiceIterator):
  def __init__(self, field, groupby):
    self.groupby = groupby
    super().__init__(field)

  def __iter__(self):
    if self.field.empty_label is not None:
      yield ("", self.field.empty_label)
    queryset = self.queryset
    # Can't use iterator() when queryset uses prefetch_related()
    if not queryset._prefetch_related_lookups:
      queryset = queryset.iterator()
    for group, objs in groupby(queryset, self.groupby):
      yield (group, [self.choice(obj) for obj in objs])

class GroupedModelChoiceField(ModelChoiceField):
  def __init__(self, *args, choices_groupby, **kwargs):
    if isinstance(choices_groupby, str):
      choices_groupby = attrgetter(choices_groupby)
    elif not callable(choices_groupby):
      raise TypeError('choices_groupby must either be a str or a callable accepting a single argument')
    self.iterator = partial(GroupedModelChoiceIterator, groupby=choices_groupby)
    super().__init__(*args, **kwargs)

最后，在表單中可以如下進行使用：

from django import forms
from .fields import GroupedModelChoiceField
from .models import Category, Expense

class ExpenseForm(forms.ModelForm):
  category = GroupedModelChoiceField(
    queryset=Category.objects.exclude(parent=None), 
    choices_groupby='parent'
  )

  class Meta:
    model = Expense
    fields = ('amount', 'date', 'category')

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

pandas中merge()函數(shù)的用法解讀
這篇文章主要介紹了pandas中merge()函數(shù)的用法解讀，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2022-11-11
Python聊天室程序（基礎(chǔ)版）
這篇文章主要為大家詳細介紹了Python聊天室程序的基礎(chǔ)版，包含客戶端和服務器端兩部分，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2018-04-04
$Django {{ MEDIA_URL }}無法顯示圖片的解決方式$
Django {{ MEDIA_URL }}無法顯示圖片的解決方式
這篇文章主要介紹了Django {{ MEDIA_URL }}無法顯示圖片的解決方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-04-04
Python讀取YAML文件過程詳解
這篇文章主要介紹了Python讀取YAML文件過程詳解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
2019-12-12
python實現(xiàn)希爾密碼加密的示例代碼
本文主要介紹了python實現(xiàn)希爾密碼加密，文中通過示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2021-08-08
下面小編就為大家?guī)硪黄猵ython 隨機數(shù)使用方法,推導以及字符串,雙色球小程序?qū)嵗?。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2017-09-09