github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-07-01 01:02:06 +00:00
Code Issues Projects Releases Wiki Activity

perf: 优化计算数量

Browse Source
This commit is contained in:
ibuler 2021-07-30 11:04:59 +08:00 committed by 老广
parent f031f4d560
commit b56b897260
1 changed files with 11 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
apps/applications/models/application.py
View File

@ -1,3 +1,5 @@
from collections import defaultdict
from django.db import models from django.db import models
from django.db.models import Count from django.db.models import Count
from django.utils.translation import ugettext_lazy as _ from django.utils.translation import ugettext_lazy as _
@ -67,7 +69,7 @@ class ApplicationTreeNodeMixin:
i = root_node.id + '_' + category.value i = root_node.id + '_' + category.value
node = cls.create_choice_node( node = cls.create_choice_node(
category, i, pid=root_node.id, tp='category', category, i, pid=root_node.id, tp='category',
counts=counts, opened=True, show_empty=show_empty, counts=counts, opened=False, show_empty=show_empty,
show_count=show_count show_count=show_count
) )
if not node: if not node:
@ -84,7 +86,7 @@ class ApplicationTreeNodeMixin:
pid = root_node.id + '_' + category.value pid = root_node.id + '_' + category.value
i = root_node.id + '_' + tp.value i = root_node.id + '_' + tp.value
node = cls.create_choice_node( node = cls.create_choice_node(
tp, i, pid, tp='type', counts=counts, tp, i, pid, tp='type', counts=counts, opened=False,
show_empty=show_empty, show_count=show_count show_empty=show_empty, show_count=show_count
) )
if not node: if not node:
@ -94,18 +96,13 @@ class ApplicationTreeNodeMixin:
@staticmethod @staticmethod
def get_tree_node_counts(queryset): def get_tree_node_counts(queryset):
counts = {'applications': queryset.count()} counts = defaultdict(int)
category_counts = queryset.annotate(count=Count('id'))\ values = queryset.values_list('type', 'category')
.values('category', 'count') \ for i in values:
.order_by() tp = i[0]
for item in category_counts: category = i[1]
counts[item['category']] = item['count'] counts[tp] += 1
counts[category] += 1
type_counts = queryset.annotate(count=Count('id')) \
.values('type', 'count') \
.order_by()
for item in type_counts:
counts[item['type']] = item['count']
return counts return counts
@classmethod @classmethod