django 非常實用的無限級分類功能

1
2
3
4
5
6
7

     def htmlpath ( self ):
        paths  =  [ ]
         for p  in  self. path. split ( ':' ):
            c  = ArticleCategory. objects. get (id__exact =p )
            url  = reverse ( 'cms.article.list' , kwargs = { 'cid':c. id } )
            paths. append ( '<a href="%s" target="_blank">%s</a>' %  (url , c. name ) )
         return  " > ". join (paths )

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

from django. db. models. signals  import pre_save

class ArticleCategory (models. Model ):
    name  = models. CharField (max_length = 50 )
    parent  = models. ForeignKey ( 'self' , null = True , blank = True , related_name = 'children' )    
    path  = models. CharField (max_length = 255 ,  null = True , blank = True )


     def  __unicode__ ( self ):
         if  self. id  ==  self. path:
             return  self. name
         else:
             return  self. node

    
     def _node ( self ):
        indent_num  =  len ( self. path. split ( ':' ) ) - 1
        indent  =  '....' * indent_num
        node  = u '%s%s' %  (indent ,  self. name )
         return node
    node  =  property (_node )


     class Meta:
        ordering  =  [ 'path' ]


     #設置在model中的用途是，是在所有節點保存時遞歸的循環下去，更新所有的節點的路徑
     def save ( self , * args , ** kwargs ):
         super (ArticleCategory , self ). save (*args , ** kwargs )

         if  self. parent:
             self. path  =  '%s:%s' %  ( self. parent. path ,  self. id )
         else:
             self. path  =  self. id

        childrens  =  self. children. all ( )
         if  len (childrens )  >  0:
             for children  in childrens:
                children. path  =  '%s:%s' %  ( self. path , children. id )
                children. save ( )
         super (ArticleCategory , self ). save (*args , ** kwargs )

#信號觸發，更新
def inital_articlecategory_path (sender , instance ,  **kwargs ):
     if instance. id:
         if instance. parent:
            instance. path  =  '%s:%s' %  (instance. parent. path , instance. id )
         else:
            instance. path  = instance. id
pre_save. connect (inital_articlecategory_path , sender =ArticleCategory )

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

class ArticleCategoryAdmin (admin. ModelAdmin ):
    list_display  =  [ 'treenode' , 'patha' , 'id' ,  ]
    ordering  =  [ 'path' ]

     def patha ( self , obj ):
         if obj. parent:
             return u '%s > %s' %  (obj. parent , obj. name )
         return obj. name

    patha. short_description  =  'path'
    patha. allow_tags  =  True
    

     def treenode ( self , obj ):
        indent_num  =  len (obj. path. split ( ':' ) ) - 1
        p  =  '<div style="text-indent:%spx;">%s</div>' %  (indent_num* 25 , obj. name )
         return p

    treenode. short_description  =  'tree path'
    treenode. allow_tags  =  True
admin. site. register (ArticleCategory , ArticleCategoryAdmin )

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

class ArticleCategory (models. Model ):
    name  = models. CharField (max_length = 50 )
    parent  = models. ForeignKey ( 'self' , null = True , blank = True , related_name = 'children' )    
    path  = models. CharField (max_length = 255 ,  null = True , blank = True )


     def  __unicode__ ( self ):
         if  self. id  ==  self. path:
             return  self. name
         else:
             return  self. node

    
     def _node ( self ):
        indent_num  =  len ( self. path. split ( ':' ) ) - 1
        indent  =  '....' * indent_num
        node  = u '%s%s' %  (indent ,  self. name )
         return node
    node  =  property (_node )


     class Meta:
        ordering  =  [ 'path' ]


     #設置在model中的用途是，是在所有節點保存時遞歸的循環下去，更新所有的節點的路徑
     def save ( self , * args , ** kwargs ):
         #先保存數據,如果是新添加的數據，放在第一行是用來獲得id，因為id是path的重要組成
         super (ArticleCategory , self ). save (*args , ** kwargs )
         if  self. parent:
             self. path  =  '%s:%s' %  ( self. parent. path ,  self. id )
         else:
             self. path  =  self. id

         #更新完當前節點path后，要進行一次保存，否則在編輯類別時，子分類循環保存父類path不是最新的
         super (ArticleCategory , self ). save (*args , ** kwargs )

        childrens  =  self. children. all ( )
         if  len (childrens )  >  0:
             for children  in childrens:
                
                children. path  =  '%s:%s' %  ( self. path , children. id )

                children. save ( )