afanasyev.aa/api-drf-1
1
Fork 0
Code Issues 3 Pull requests Projects 1 Releases Activity

Views various fixes

Browse source
This commit is contained in:
ThePetrovich 2025-07-03 22:20:18 +08:00
parent 5af3b95c8d
commit 51415765da
4 changed files with 41 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

18
api/serializers.py
View file

@ -79,6 +79,24 @@ class PredictionRequestSerializer(serializers.Serializer):
data['descent_rate'] = rate_clip(data['descent_rate'])
return data
def create(self, validated_data):
if 'ascent_curve' in validated_data:
validated_data['ascent_curve'] = base64_to_curve(validated_data['ascent_curve'])
if 'descent_curve' in validated_data:
validated_data['descent_curve'] = base64_to_curve(validated_data['descent_curve'])
prediction = Prediction(
user=validated_data.get('user'),
request=validated_data.get('request', {}),
result=validated_data.get('result', {}),
start_point=validated_data.get('start_point'),
template=validated_data.get('template'),
rate_profile=validated_data.get('rate_profile')
)
prediction.save()
return prediction