Django MPTT - 树过滤

时间:2009-03-04 13:00:43

标签: django tree django-templates plpgsql django-mptt

我正在使用MPTT的templatetag来渲染我的流派树。

   {% for genre, structure in genres|tree_info %}
      {% if tree.new_level %}<ul><li>{% else %}</li><li>{% endif %}
         {{ genre.name }}
      {% for level in tree.closed_levels %}</li></ul>{% endfor %}
   {% endfor %}

问题是,我的genre对象具有应该受到尊重的is_visible属性。

    def is_visible(self):  
       if self.is_root_node() or not self.visibility:
           return self.visibility                          
       for parent in self.get_ancestors():
           if not parent.visibility:      
               return False                                    
       return True

实现这一目标的最聪明,最干净的方法是什么?


其他信息

我需要(X)HTML列表嵌套才能正确生成。 我已经为inhertited可见性检查定义了SQL函数。

CREATE OR REPLACE function get_genre_parent_id( _genre_id int )
RETURNS INTEGER AS $$
DECLARE 
    _parent_id INTEGER;
    _genre_id ALIAS FOR $1;    
BEGIN
    SELECT parent_id INTO _parent_id
    FROM product_productgenre
    WHERE id = _genre_id;

    RETURN _parent_id;

END;
$$  LANGUAGE plpgsql;


CREATE OR REPLACE function is_genre_visible( _genre_id int )
RETURNS BOOLEAN AS $$
DECLARE 
    _visible BOOLEAN;
    _genre_id ALIAS FOR $1;    
BEGIN
    SELECT visibility INTO _visible
    FROM product_productgenre
    WHERE id = _genre_id;

    RETURN _visible;

END;
$$  LANGUAGE plpgsql;


CREATE OR REPLACE function is_genre_branch_visible( _genre_id int )
RETURNS BOOLEAN AS $$
DECLARE 
    visible BOOLEAN;
    _genre_id ALIAS FOR $1;
    _temp_genre_id INTEGER;
BEGIN
    visible = true;
    _temp_genre_id := _genre_id;
    -- checking for our own genre
    IF NOT is_genre_visible(_temp_genre_id) THEN
        RETURN false;
    END IF;
    -- iterating through all parent genres
    WHILE get_genre_parent_id(_temp_genre_id) IS NOT NULL LOOP
        _temp_genre_id = get_genre_parent_id(_temp_genre_id);
        IF NOT is_genre_visible(_temp_genre_id) THEN
            RETURN false;
        END IF;
    END LOOP;

    RETURN visible;

END;
$$  LANGUAGE plpgsql;

并尝试覆盖full_tree_for_model标记,以使其使用自定义管理器,只需向QuerySet添加额外的is_genre_branch_visible(genre_id)但是排序出了问题,无法解决这个问题。

Plus 它有效,但我不喜欢这种做法,对我来说感觉很难看。

1 个答案:

答案 0 :(得分:-1)

嗯,简单

{% if genre.is_visible %}

在for循环中应该可以做到这一点:)