github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-03 08:25:04 +00:00
Code Issues Projects Releases Wiki Activity

perf: 优化标签搜索资产使用的关系

Browse Source
This commit is contained in:
ibuler
2024-02-29 19:01:38 +08:00
committed by Bryan
parent f0ffa2408d
commit 18707d365b
2 changed files with 27 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
apps/labels/mixins.py
View File

@@ -1,4 +1,5 @@
from django.contrib.contenttypes.fields import GenericRelation
from django.contrib.contenttypes.models import ContentType
from django.db import models
from django.db.models import OneToOneField, Count
@@ -38,8 +39,11 @@ class LabeledMixin(models.Model):
self.real.labels.set(value, bulk=False)
@classmethod
def filter_resources_by_labels(cls, resources, label_ids):
return cls._get_filter_res_by_labels_m2m_all(resources, label_ids)
def filter_resources_by_labels(cls, resources, label_ids, rel='all'):
if rel == 'all':
return cls._get_filter_res_by_labels_m2m_all(resources, label_ids)
else:
return cls._get_filter_res_by_labels_m2m_in(resources, label_ids)
@classmethod
def _get_filter_res_by_labels_m2m_in(cls, resources, label_ids):
@@ -60,7 +64,8 @@ class LabeledMixin(models.Model):
@classmethod
def get_labels_filter_attr_q(cls, value, match):
resources = LabeledResource.objects.all()
res_type = ContentType.objects.get_for_model(cls.label_model())
resources = LabeledResource.objects.all().filter(res_type=res_type)
if not value:
return None