ITPub博客

首页 > IT职业 > IT职场 > 使用 django-mptt 实现树形结构

使用 django-mptt 实现树形结构

IT职场 作者:lslzf 时间:2021-09-09 09:28:27 0 删除 编辑

在 Web 开发中,省/市/区(县)是经常需要用到的数据,而省/市/区(县)是一种层级关系。在使用 Django 做 Web 开发时,此时 Django-mptt 是一个很有用的模块。

环境版本:

  • python 2.7.10

  • Django 1.8.5

  • django-mptt   0.8.7

安装 django-mptt
pip install django-mptt
新建 area 应用
startapp area

在 models.py 里添加如下代码:

#coding:utf-8from django.db import modelsfrom mptt.models import MPTTModelclass Area(MPTTModel):
    name = models.CharField('名称', max_length=50, unique=True)
    parent = models.ForeignKey('self', verbose_name='上级区域', null=True, blank=True, related_name='children')    class Meta:
        db_table = 'area'
        verbose_name = verbose_name_plural = '省/市/地区(县)'

    def __unicode__(self):
        return self.name

在这里只需让 Area 继承 MPTTModel。
需要说明的是,实际上 MPTTModel 隐藏了四个变量:level,lft,rght 和 tree_id。大多数时候我们是用不到这几个变量的。另外,如果你的 Model 中 parent 变量名字不是 "parent" 时,应当在 Model 中MPTT 元类中指明:

#coding:utf-8from django.db import modelsfrom mptt.models import MPTTModelclass Area(MPTTModel):
    name = models.CharField('名称', max_length=50, unique=True)
    parent_area = models.ForeignKey('self', verbose_name='上级区域', null=True, blank=True, related_name='children')    class Meta:
        db_table = 'area'
        verbose_name = verbose_name_plural = '省/市/地区(县)'

    class MPTTMeta:
        parent_attr = 'parent_area'

    def __unicode__(self):
        return self.name

在 admin.py 里添加如下代码:

from django.contrib import adminfrom .models import Areaclass AreaAdmin(admin.ModelAdmin):
    list_display = ('id', 'name', 'parent', 'level')

admin.site.register(Area, AreaAdmin)

在项目的 settings.py 里 INSTALLED_APPS 里添加 mptt 和 area,之后执行数据库操作。之后在 admin 后台中按照层级关系添加省/市/区(县)数据。

图片描述


在 views.py 里添加如下代码:

from django.shortcuts import render_to_responsefrom .models import Areadef tree(request):
    nodes = Area.objects.all()    return render_to_response('tree.html', {'nodes': nodes})

在 templates 文件夹下添加模版文件 tree.html:

{% load staticfiles %}
{% load mptt_tags %}html>
    
    Django 树形结构
    
    
    
    
    
        
                
  •                              
  •             
                      {% recursetree nodes %}                    
    •                         {% if node.is_leaf_node %}                            
    •                         {% else %}                            
    •                             
                                        
      •                             
                              {% endif %}                                     {% endrecursetree %}            
            
    

别忘了在 settings.py 里添加如下代码:

STATIC_ROOT = os.path.join(BASE_DIR, 'static')
STATICFILES_DIRS = [
    ('css', os.path.join(STATIC_ROOT, 'css')),
    ('js', os.path.join(STATIC_ROOT, 'js')),
    ('img', os.path.join(STATIC_ROOT, 'img')),
]

否则获取不到静态资源!
最后在项目的 urls.py 里添加如下代码:

from area.views import tree

urlpatterns = [
    ...
    url(r'^tree/', tree),
    ...
]

这样,就可以通过访问 http://127.0.0.1:8000/tree/ 看到树形结构了。

图片描述

具体代码可以参看我在 github 上的 tree 仓库。



作者:与蟒唯舞
链接:https://www.jianshu.com/p/cd5986ccba6b


来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/1070/viewspace-2803406/ ,如需转载,请注明出处,否则将追究法律责任。

请登录后发表评论 登录
全部评论

注册时间:2019-04-15

  • 博文量
    102
  • 访问量
    876