Django MPTT - фильтрация деревьев - PullRequest
Новая
       20

Django MPTT - фильтрация деревьев

1 голос
/ 04 марта 2009

Я использую шаблонный тег MPTT для рендеринга своего жанрового дерева. 

   {% 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 создавалась правильно. Я определил функции 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, чтобы он использовал собственный менеджер, который просто добавляет дополнительный is_genre_branch_visible(genre_id) к QuerySet. Но с заказом что-то не так, на самом деле не могу понять это.

Плюс Это работает, но мне не нравится подход, мне некрасиво.

1 Ответ

0 голосов
/ 04 марта 2009

Ну, просто 

{% if genre.is_visible %}

только внутри цикла for сделайте трюк :)

...