perf: 优化标签绑定,仅绑定到资产上

This commit is contained in:
ibuler
2024-01-03 16:14:27 +08:00
committed by 老广
parent eb5a53b91b
commit 15ac81a422
7 changed files with 44 additions and 24 deletions

View File

@@ -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)