perf: 修改标签的搜索

This commit is contained in:
ibuler
2023-12-20 10:52:02 +08:00
committed by Bryan
parent 9b2b71dddc
commit ac5991fc43
2 changed files with 3 additions and 2 deletions

View File

@@ -5,6 +5,7 @@ class LabeledResourceType(LazyObject):
@staticmethod
def get_res_types():
from rbac.models import ContentType
from .mixins import LabeledMixin
content_types = ContentType.objects.all()
ids = []
for ct in content_types:
@@ -13,7 +14,7 @@ class LabeledResourceType(LazyObject):
continue
if model_cls._meta.parents:
continue
if 'labels' in model_cls._meta._forward_fields_map.keys():
if issubclass(model_cls, LabeledMixin):
ids.append(ct.id)
return ContentType.objects.filter(id__in=ids)