Невозможно правильно сохранить в базу данных с отношением OneToOne
Поток того, что я хочу сделать, следующий:
У меня есть две таблицы, а именно Event и Result. Таблица Event связана ForeignKey с пользователем, а таблица Result связана OneToOne отношениями с Event. В моей views.py у меня есть одна POST и одна GET. В POST я беру некоторые данные и сохраняю их в таблице Event. В GET я решаю задачу оптимизации и хочу сохранить решение в Result.
Следует models.py следующим образом:
from django.db import models
from django.contrib.postgres.fields import ArrayField
from django.contrib.auth import get_user_model
CustomUser = get_user_model()
class Event(models.Model):
user_id_event = models.ForeignKey(CustomUser, on_delete=models.CASCADE, null=True)
dr_notice_period = models.IntegerField(blank=True, null=True)
dr_duration = models.IntegerField(blank=True, null=True)
dr_request = models.FloatField(blank=True, null=True)
class Result(models.Model):
event_id_result = models.OneToOneField(Event, on_delete=models.CASCADE, null=True)
HVAC_flex = ArrayField(models.FloatField(blank=True, null=True))
DHW_flex = ArrayField(models.FloatField(blank=True, null=True))
lights_flex = ArrayField(models.FloatField(blank=True, null=True))
Следует serializers.py следующим образом:
from rest_framework import serializers
from vpp_optimization.models import Event, Result
class EventSerializer(serializers.ModelSerializer):
class Meta:
model = Event
fields = ('__all__')
class ResultSerializer(serializers.ModelSerializer):
HVAC_flex = serializers.ListField(child=serializers.FloatField())
DHW_flex = serializers.ListField(child=serializers.FloatField())
lights_flex = serializers.ListField(child=serializers.FloatField())
class Meta:
model = Result
fields = ('__all__')
Следует views.py следующим образом:
from rest_framework.response import Response
from rest_framework.decorators import api_view, permission_classes
from rest_framework.permissions import IsAuthenticated
from rest_framework import status
from vpp_optimization.importer import DR_event_import_data
from vpp_optimization.utils import optimization_solution, convert_dict_for_serializer
from vpp_optimization.serializers import EventSerializer, ResultSerializer
from vpp_optimization.models import Event, Result
@api_view(['POST'])
@permission_classes([IsAuthenticated,])
def event(request):
serializer = EventSerializer(data=request.data)
if serializer.is_valid():
serializer.save(user_id_event=request.user)
return Response({"status": "success", "data": serializer.data}, status=status.HTTP_200_OK)
else:
return Response({"status": "error", "data": serializer.errors}, status=status.HTTP_400_BAD_REQUEST)
@api_view(['GET'])
def optimization(request):
last_event = Event.objects.last()
if not last_event:
return Response({"res": "Object Event does not exists"}, status=status.HTTP_400_BAD_REQUEST)
serializer = EventSerializer(last_event)
response_dict = {}
DR_event_data = DR_event_import_data(serializer.data)
response_dict = optimization_solution(DR_event_data)
response_dict_final = convert_dict_for_serializer(response_dict)
serializer_result = ResultSerializer(data=response_dict_final)
if serializer_result.is_valid():
serializer_result.save(event_id_result=last_event.id)
return Response(serializer_result.data, status=status.HTTP_200_OK)
Однако, я получаю следующую ошибку:
ValueError: Cannot assign "2": "Result.event_id_result" must be a "Event" instance.
Ошибка связана с тем, как я пытаюсь сохранить с помощью отношения OneToOne, а именно в этой строке views.py:
event_id_result=last_event.id
Очевидно, что это не правильный способ соединения двух баз данных. Может ли кто-нибудь помочь мне найти правильный способ?
Изменение:
event_id_result=last_event.id
To:
event_id_result=last_event