From 5af3b95c8d67ff4be38240527dc3425355752691 Mon Sep 17 00:00:00 2001 From: ThePetrovich Date: Thu, 3 Jul 2025 20:56:22 +0800 Subject: [PATCH] add fields to serializer --- api/serializers.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/api/serializers.py b/api/serializers.py index 7b9688c..ad7ef2f 100644 --- a/api/serializers.py +++ b/api/serializers.py @@ -13,9 +13,9 @@ class PredictionSerializer(serializers.ModelSerializer): PROFILE_STANDARD = "standard_profile" -PROFILE_FLOAT = "float" -PROFILE_REVERSE = "reverse" -PROFILE_CUSTOM = "custom" +PROFILE_FLOAT = "float_profile" +PROFILE_REVERSE = "reverse_profile" +PROFILE_CUSTOM = "custom_profile" LATEST_DATASET_KEYWORD = "latest" SUPPORTED_PROFILES = [PROFILE_STANDARD, PROFILE_FLOAT, PROFILE_REVERSE, PROFILE_CUSTOM] @@ -38,6 +38,15 @@ class PredictionRequestSerializer(serializers.Serializer): ascent_curve = serializers.CharField(required=False) descent_curve = serializers.CharField(required=False) interpolate = serializers.BooleanField(required=False, default=False) + start_point = serializers.PrimaryKeyRelatedField( + queryset=SavedPoint.objects.all(), required=False, allow_null=True + ) + rate_profile = serializers.PrimaryKeyRelatedField( + queryset=SavedRateProfile.objects.all(), required=False, allow_null=True + ) + template = serializers.PrimaryKeyRelatedField( + queryset=PreditctionTemplate.objects.all(), required=False, allow_null=True + ) def validate(self, data): profile = data.get("profile", PROFILE_STANDARD)