如何使用django-treebeard實(shí)現(xiàn)樹類型存儲(chǔ)與編輯

更新時(shí)間：2024年08月05日 09:03:49 作者：程序設(shè)計(jì)實(shí)驗(yàn)室

這篇文章主要介紹了使用django-treebeard實(shí)現(xiàn)樹類型存儲(chǔ)與編輯的宣相關(guān)操作代碼,本文給大家介紹的非常詳細(xì),對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友參考下吧

前言

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

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

難道Django已經(jīng)沒落了？

效果

以這個(gè)汽車改裝項(xiàng)目為例

實(shí)現(xiàn)以下效果，可以拖動(dòng)節(jié)點(diǎn)進(jìn)行編輯

安裝

安裝依賴

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 才能實(shí)現(xiàn)可視化的樹編輯

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

初始化數(shù)據(jù)

可以使用代碼把初始化的Tree數(shù)據(jù)導(dǎo)入

（事實(shí)上是因?yàn)椴幌葘?dǎo)入初始化數(shù)據(jù)，admin界面連個(gè)添加按鈕都沒有……估計(jì)是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界面就可以看到了

算是能用吧

寫個(gè)接口

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

（實(shí)際上這些代碼是 DjangoStarter 自動(dòng)生成的）

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

結(jié)果出來的數(shù)據(jù)是這樣（省略部分?jǐn)?shù)據(jù)）

{
  "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": "漸變色"
    }
  ]
}

小結(jié)

還是自己實(shí)現(xiàn)的舒服。

不過這個(gè)也算是開箱即用了，小項(xiàng)目的話隨便搞搞還是可以的。

參考資料

https://django-treebeard.readthedocs.io/

到此這篇關(guān)于使用django-treebeard實(shí)現(xiàn)樹類型存儲(chǔ)與編輯的文章就介紹到這了,更多相關(guān)django-treebeard樹類型存儲(chǔ)與編輯內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: