mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

12552 fix cumulative count

Browse Source
This commit is contained in:
Arthur
2023-06-27 13:26:14 -07:00
parent 0261fca097
commit 19bd16ca1f
1 changed files with 3 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
netbox/utilities/tree_queries.py
View File

@ -68,33 +68,11 @@ class TreeManager(Manager.from_queryset(TreeQuerySet), TreeManager_):
""" """
if cumulative: if cumulative:
SQL = """ SQL = """
WITH recursive __tree ( "tree_depth", "tree_path", "tree_pk" ) AS
(
SELECT 0 AS tree_depth,
array[T1.id] AS tree_path,
T1."id"
FROM "{table1}" T0
INNER JOIN "{table2}" T1
ON (T0."{rel_field}_id" = T1."id")
WHERE T1."parent_id" IS NULL
UNION ALL
SELECT __tree.tree_depth + 1 AS tree_depth,
__tree.tree_path
|| T1.id,
T1."id"
FROM "{table1}" T0
INNER JOIN "{table2}" T1
ON (T0."{rel_field}_id" = T1."id")
JOIN __tree
ON T1."parent_id" = __tree.tree_pk
)
SELECT count(*) SELECT count(*)
FROM ( FROM (
SELECT T0."id" SELECT (__tree.tree_path) AS "tree_path"
FROM "{table1}" T0 FROM "{table1}" T3 JOIN __tree ON T3."{rel_field}_id" = __tree.tree_pk
INNER JOIN "{table2}" T1 WHERE ("{table2}"."id" = ANY(tree_path))
ON (T0."{rel_field}_id" = T1."id")
WHERE (T1.id = ANY(__tree.tree_path))
) _count ) _count
""" """