Class UpdateFlagV2Serializer

api/features/feature_states/serializers.py:166–248

class UpdateFlagV2Serializer(BaseFeatureUpdateSerializer):
    feature = FeatureIdentifierSerializer(required=True)
    environment_default = EnvironmentDefaultSerializer(required=True)
    segment_overrides = SegmentOverrideSerializer(many=True, required=False)

    def validate_segment_overrides(
        self,
        value: list[dict],  # type: ignore[type-arg]
    ) -> list[dict]:  # type: ignore[type-arg]
        if not value:
            return value

        segment_ids = [override["segment_id"] for override in value]
        if len(segment_ids) != len(set(segment_ids)):
            raise serializers.ValidationError(
                "Duplicate segment_id values are not allowed"
            )

        # TODO: optimise this once out of experimentation
        for segment_id in segment_ids:
            self.validate_segment_id(segment_id)

        return value

    def validate(self, data: dict) -> dict:  # type: ignore[type-arg]
        overrides = data.get("segment_overrides", [])
        if any(o.get("multivariate_feature_state_values") for o in overrides):
            feature = Feature.objects.filter(
                project_id=self.environment.project_id, **data["feature"]
            ).first()
            if feature is not None:
                for override in overrides:
                    validate_multivariate_state_values(
                        feature,
                        override.get("multivariate_feature_state_values", []),
                    )
        return data

    @property
    def change_set_v2(self) -> FlagChangeSetV2:
        validated_data = self.validated_data

        env_default = validated_data["environment_default"]
        env_value_data = env_default["value"]

        segment_overrides_data = validated_data.get("segment_overrides", [])
        segment_overrides = []

        for override_data in segment_overrides_data:
            value_data = override_data["value"]

            multivariate_data = override_data.get("multivariate_feature_state_values")
            segment_override = SegmentOverrideChangeSet(
                segment_id=override_data["segment_id"],
                enabled=override_data["enabled"],
                feature_state_value=value_data["value"],
                type_=value_data["type"],
                priority=override_data.get("priority"),