mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-09-04 00:44:48 +00:00
perf: 优化标签绑定,仅绑定到资产上
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
from django.contrib.contenttypes.fields import GenericRelation
|
||||
from django.db import models
|
||||
from django.db.models import OneToOneField
|
||||
|
||||
from .models import LabeledResource
|
||||
|
||||
@@ -12,10 +13,25 @@ class LabeledMixin(models.Model):
|
||||
class Meta:
|
||||
abstract = True
|
||||
|
||||
@classmethod
|
||||
def label_model(cls):
|
||||
pk_field = cls._meta.pk
|
||||
model = cls
|
||||
if isinstance(pk_field, OneToOneField):
|
||||
model = pk_field.related_model
|
||||
return model
|
||||
|
||||
@property
|
||||
def real(self):
|
||||
pk_field = self._meta.pk
|
||||
if isinstance(pk_field, OneToOneField):
|
||||
return getattr(self, pk_field.name)
|
||||
return self
|
||||
|
||||
@property
|
||||
def labels(self):
|
||||
return self._labels
|
||||
return self.real._labels
|
||||
|
||||
@labels.setter
|
||||
def labels(self, value):
|
||||
self._labels.set(value, bulk=False)
|
||||
self.real._labels.set(value, bulk=False)
|
||||
|
Reference in New Issue
Block a user