如何使用django-treebeard實現樹類型存儲與編輯

更新時間：2024年08月05日 09:03:49 作者：程序設計實驗室

這篇文章主要介紹了使用django-treebeard實現樹類型存儲與編輯的宣相關操作代碼,本文給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友參考下吧

前言

其實之前做很多項目都有遇到跟樹相關的功能，以前都是自己實現的，然后前端很多UI組件庫都有Tree組件，套上去就可以用。

不過既然用 Django 了，還是得充分發(fā)揮一下生態(tài)的優(yōu)勢，但是我找了半天，也就這個 treebeard 能用，其他要不停更了要不就功能很拉，沒有可視化編輯樹的功能。

難道Django已經沒落了？

效果

以這個汽車改裝項目為例

實現以下效果，可以拖動節(jié)點進行編輯

安裝

安裝依賴

pdm add django-treebeard

將 treebeard 添加到 INSTALLED_APPS

定義model

繼承 MP_Node 類型就可以

from treebeard.mp_tree import MP_Node
class CaseCategory(MP_Node):
    name = models.CharField('類別名稱', max_length=100)
    node_order_by = ['name']
    def __str__(self):
        return '改裝類別: {}'.format(self.name)
    class Meta:
        db_table = 'car_case_category'
        verbose_name = '改裝類別'
        verbose_name_plural = verbose_name

配置 admin

需要繼承 TreeAdmin 才能實現可視化的樹編輯

from treebeard.admin import TreeAdmin
@admin.register(CaseCategory)
class CaseCategoryAdmin(TreeAdmin):
    form = movenodeform_factory(CaseCategory)
    list_display = ['name', 'depth']
    search_fields = ['name']

初始化數據

可以使用代碼把初始化的Tree數據導入

（事實上是因為不先導入初始化數據，admin界面連個添加按鈕都沒有……估計是bug）

def seed_data_treebeard():
    from apps.car.models import CaseCategory
    get = lambda node_id: CaseCategory.objects.get(pk=node_id)
    root = CaseCategory.add_root(name='車衣')
    node = get(root.pk).add_child(name='亮面/光面')
    node = get(root.pk).add_child(name='磨砂/啞光')
    root = CaseCategory.add_root(name='改色')
    get(root.pk).add_child(name='純色系')
    get(root.pk).add_child(name='漸變色')
    get(root.pk).add_child(name='定制彩繪')
    root = CaseCategory.add_root(name='改裝')
    get(root.pk).add_child(name='輪轂')
    get(root.pk).add_child(name='剎車')
    get(root.pk).add_child(name='避震')
    root = CaseCategory.add_root(name='省心提')

這樣打開admin界面就可以看到了

算是能用吧

寫個接口

然后我再寫個簡單的接口，基于 django-ninja

（實際上這些代碼是 DjangoStarter 自動生成的）

from typing import List
from django.shortcuts import get_object_or_404
from ninja import Router, ModelSchema
from django_starter.http.response import responses
router = Router(tags=['case_category'])
class CaseCategoryOut(ModelSchema):
    class Meta:
        model = CaseCategory
        fields = ['id', 'path', 'depth', 'numchild', 'name', ]
@router.get('/', response=List[CaseCategoryOut], url_name='car/case_category/list')
def list_items(request):
    qs = CaseCategory.objects.all()
    return qs

結果出來的數據是這樣（省略部分數據）

{
  "code": 200,
  "data": [
    {
      "id": 4,
      "path": "0001",
      "depth": 1,
      "numchild": 3,
      "name": "改色"
    },
    {
      "id": 7,
      "path": "00010001",
      "depth": 2,
      "numchild": 0,
      "name": "定制彩繪"
    },
    {
      "id": 6,
      "path": "00010002",
      "depth": 2,
      "numchild": 0,
      "name": "漸變色"
    }
  ]
}