Включая подсчет внешнего ключа в полный список django mptt? - PullRequest
Новая
       29

Включая подсчет внешнего ключа в полный список django mptt?

3 голосов
/ 12 июля 2010

Я выкладываю свое дерево категорий следующим образом: 

<div id="categories-tree">
{% load mptt_tags %}
{% full_tree_for_model bugs.Category as cats cumalative count bugs.Bug.categories %}
{% for node, structure in cats|tree_info %}
{% if structure.new_level %}<ul><li>{% else %}</li><li>{% endif %}
    <a href="/categories/{{node.slug}}">{{ node }}</a>
    {% for level in structure.closed_levels %}</li></ul>{% endfor %}
{% endfor %}
</div>

Однако я также хочу сгенерировать <span class="count">13</span> для своих внутренних категорий (возможно, начиная с 3-го уровня?) Суммыошибок, связанных с категорией, так как мои ошибки могут иметь несколько связанных категорий.

Я думаю, мне нужно что-то вроде этого фрагмента внутри моего вложенного цикла for, но я не совсем уверен, как это сделать: 

   {% drilldown_tree_for_node [node] as [varname] count [foreign_key] in [count_attr] %}

Вот мои модели для справки: 

class Bug( models.Model ):
    name = models.CharField( max_length=100 )
    slug = models.SlugField(unique=True)
    summary = models.TextField()
    date_added = models.DateTimeField()
    poster = models.ForeignKey(User)
    categories = models.ManyToManyField('Category')

class Category ( models.Model ):
    name = models.CharField( max_length=100 )
    parent = models.ForeignKey('self', null=True, blank=True, related_name='children')
    slug = models.SlugField(unique=True)

mptt.register(Category)

Таким образом, текущий выход: 

<ul>
    <li><a href="#">CSS</a>
        <ul>
            <li><a href="#">Position</a>
                <ul>
                    <li><a href="#">Absolute</a></li>
                    <li><a href="#">Absolute Fixed</a></li>
                </ul>
            </li>
        </ul>
    </li>
</ul>

И идеально: 

<ul>
    <li><a href="#">CSS</a>
        <ul>
            <li><a href="#">Position</a>
                <ul>
                    <li><a href="#">Absolute</a> <span>13</span></li>
                    <li><a href="#">Absolute Fixed</a> <span>10</span></li>
                </ul>
            </li>
        </ul>
    </li>
</ul>

1 Ответ

1 голос
/ 13 сентября 2013

Начните с добавления метода к модели категории: 

def get_bug_count(self):
    return Bugs.objects.filter(category=self).count()

А затем в шаблоне теоретически вы сможете сделать: 

<span>{{ node.get_bug_count }}</span>
...