Class UpdateFlagSerializer

api/features/feature_states/serializers.py:98–126

class UpdateFlagSerializer(BaseFeatureUpdateSerializer):
    feature = FeatureIdentifierSerializer(required=True)
    segment = FeatureUpdateSegmentDataSerializer(required=False)
    enabled = serializers.BooleanField(required=True)
    value = FeatureValueSerializer(required=True)

    def validate_segment(self, value: dict) -> dict:  # type: ignore[type-arg]
        if value and "id" in value:
            self.validate_segment_id(value["id"])
        return value

    @property
    def flag_change_set(self) -> FlagChangeSet:
        validated_data = self.validated_data
        value_data = validated_data["value"]
        segment_data = validated_data.get("segment")

        return FlagChangeSet(
            author=AuthorData.from_request(self.context["request"]),
            enabled=validated_data["enabled"],
            feature_state_value=value_data["value"],
            type_=value_data["type"],
            segment_id=segment_data.get("id") if segment_data else None,
            segment_priority=segment_data.get("priority") if segment_data else None,
        )

    def save(self, **kwargs: object) -> FeatureState:
        feature = self.get_feature()
        return update_flag(self.environment, feature, self.flag_change_set)


class EnvironmentDefaultSerializer(serializers.Serializer):  # type: ignore[type-arg]