56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
|
from rest_framework import serializers
|
||
|
|
|
||
|
|
from apps.labels.models import AssociatedLabel, LabelKeyCache, LabelValueCache
|
||
|
|
from apps.labels.utils import is_labels_feature_enabled
|
||
|
|
|
||
|
|
|
||
|
|
class LabelKeySerializer(serializers.ModelSerializer):
|
||
|
|
id = serializers.CharField()
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = LabelKeyCache
|
||
|
|
fields = (
|
||
|
|
"id",
|
||
|
|
"name",
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class LabelValueSerializer(serializers.ModelSerializer):
|
||
|
|
id = serializers.CharField()
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = LabelValueCache
|
||
|
|
fields = (
|
||
|
|
"id",
|
||
|
|
"name",
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class LabelSerializer(serializers.Serializer):
|
||
|
|
key = LabelKeySerializer()
|
||
|
|
value = LabelValueSerializer()
|
||
|
|
|
||
|
|
|
||
|
|
class LabelKeyValuesSerializer(serializers.Serializer):
|
||
|
|
key = LabelKeySerializer()
|
||
|
|
values = LabelValueSerializer(many=True)
|
||
|
|
|
||
|
|
|
||
|
|
class LabelReprSerializer(serializers.Serializer):
|
||
|
|
name = serializers.CharField()
|
||
|
|
|
||
|
|
|
||
|
|
class LabelsSerializerMixin(serializers.Serializer):
|
||
|
|
labels = LabelSerializer(many=True, required=False)
|
||
|
|
|
||
|
|
def validate_labels(self, labels):
|
||
|
|
if labels:
|
||
|
|
keys = {label["key"]["id"] for label in labels}
|
||
|
|
if len(keys) != len(labels):
|
||
|
|
raise serializers.ValidationError(detail="Duplicate label key")
|
||
|
|
return labels
|
||
|
|
|
||
|
|
def update_labels_association_if_needed(self, labels, instance, organization):
|
||
|
|
if labels is not None and is_labels_feature_enabled(organization):
|
||
|
|
AssociatedLabel.update_association(labels, instance, organization)
|