/
tree_filters.py
41 lines (30 loc) · 1.14 KB
/
tree_filters.py
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
"""
This is a collection of tags and filters for models which inherit from the MPTT model
:class:`~integreat_cms.cms.models.abstract_tree_node.AbstractTreeNode`
(:class:`~integreat_cms.cms.models.pages.page.Page` and
:class:`~integreat_cms.cms.models.languages.language_tree_node.LanguageTreeNode`).
"""
from __future__ import annotations
from typing import TYPE_CHECKING
from django import template
if TYPE_CHECKING:
from ..models import LanguageTreeNode, Page
register = template.Library()
@register.filter
def get_descendant_ids(node: LanguageTreeNode | Page) -> list[int]:
"""
This filter returns the ids of all the node's descendants.
:param node: The requested node
:return: The list of all the node's descendants' ids
"""
return [
descendant.id for descendant in node.get_cached_descendants(include_self=True)
]
@register.filter
def get_children_ids(node: Page) -> list[int]:
"""
This filter returns the ids of all the node's direct children.
:param node: The requested node
:return: The list of all the node's children's ids
"""
return [child.id for child in node.cached_children]