Как добавить данные в дочернюю модель вместе с родительским режимом? - PullRequest
0 голосов
/ 14 марта 2019

Я работаю с Django REST framework. Я хочу добавить данные в дочернюю модель вместе с родительской моделью. Основное создание двух записей одновременно.

models.py

class Visit(models.Model):
   name = models.CharField(max_length=200)
   gender = models.CharField(choices=GENDER_CHOICE, max_length=1)
   mobile = models.CharField(max_length=18,default="")
   email = models.CharField(max_length=256, null=True, blank=True)
   address = models.TextField(null=True, blank=True)
   visit_type = models.IntegerField(choices=VISIT_TYPE)
   visit_purpose = models.CharField(max_length=250)
   visitor_photo = models.FileField(upload_to="visitor/",null=True, blank=True)
   id_photo = models.FileField(upload_to="id_card/",null=True, blank=True)
   date_created = models.DateTimeField(default=timezone.now, editable=False)

class Status(models.Model):
   visit = models.ForeignKey(Visit,on_delete=models.CASCADE)
   description = models.CharField(max_length=200)
   from_time = models.DateField()
   to_time = models.DateTimeField(null=True, blank=True)
   aproved = models.BooleanField(default=False)
   visit_complete = models.BooleanField(default=False)
   exit_time = models.DateTimeField(null=True, blank=True)
   date_created = models.DateTimeField(default=timezone.now, editable=False)

serializers.py

class StatusSerializers(serializers.ModelSerializer):
       class Meta:
         model = Status
         fields = "__all__"
class VisitSerializers(serializers.ModelSerializer):
   visit = StatusSerializers(many=True)
   class Meta:
       model = Visit
       fields = "__all__"

   def create(self, validated_data):
      print(validated_data)
      model_b = Visit.objects.create(**validated_data)
      # print(self.context.request.data.get('description'))
      child_model_data = {
         'description': self.context.request.data.get('description'),
      }
      child_model_serializer = StatusSerializers(data=child_model_data)
      child_model_serializer.is_valid(raise_exception=True)
      child_model_serializer.save(status=model_b)

      return model_b

views.py

  @api_view(['GET', 'POST'])
  def create_visitor(request):
    if request.method == 'GET':
       visitor = Visit.objects.all()
       serializer = VisitSerializers(visitor,context={'request': request},many=True)
       return Response(serializer.data)
    elif request.method == 'POST':
       serializer = VisitSerializers(data=request.data,context={'request': request})
    if serializer.is_valid():
        serializer.save()
        return Response(serializer.data, status=status.HTTP_201_CREATED)
    return Response(serializer.errors,

состояние = status.HTTP_400_BAD_REQUEST) В приведенном выше коде я могу создать дочернюю запись, то есть запись модели состояния, но не знаю, как добавить в нее другие поля во время ее создания.

Ответы [ 2 ]

2 голосов
/ 14 марта 2019

В вашей serializer данных дочерней модели можно найти в self.context.request.data, и вы можете просто использовать дочерний сериализатор для проверки и сохранения в db.

class VisitSerializers(serializers.ModelSerializer):
   class Meta:
       model = Visit
       fields = "__all__"

   def create(self, validated_data):
      print(validated_data)
      model_b = Visit.objects.create(**validated_data)
      child_model_data = {
         'field': self.context['request'].data.get('your_field');
         .....
      }
      child_model_serializer = ChildSerializer(data=child_model_data)
      child_model_serializer.is_valid(raise_exception=True)
      child_model_serializer.save()

      return model_b

Обновление нужно менять в нескольких местах

Сначала, на ваш взгляд, при инициализации сериализатора метода post вам нужно передать объект контекста.

  @api_view(['GET', 'POST'])
  def create_visitor(request):
     if request.method == 'GET':
        visitor = Visit.objects.all()
        serializer = VisitSerializers(visitor,context={'request': request},many=True)
        return Response(serializer.data)
    elif request.method == 'POST':
        serializer = VisitSerializers(data=request.data, context={'request': request})
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data, status=status.HTTP_201_CREATED)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

А потом в вашем VisitSerializer

class VisitSerializers(serializers.ModelSerializer):
   visit = StatusSerializers(many=True)  # not sure why you did this, this should be removed i guess
   class Meta:
       model = Visit
       fields = "__all__"

   def create(self, validated_data):
      print(validated_data)
      model_b = Visit.objects.create(**validated_data)
      # print(self.context.request.data.get('description'))
      child_model_data = {
         'description': self.context.request.data.get('description'),
          'status': model_b
          # and make sure all other necessary fields.
      }
      child_model_serializer = StatusSerializers(data=child_model_data)
      child_model_serializer.is_valid(raise_exception=True)
      child_model_serializer.save()

      return model_b
0 голосов
/ 14 марта 2019

Вы можете иметь дополнительные поля в вашем сериализаторе наряду с полями по умолчанию.Для обратных отношений вы можете добавить поле вроде этого

class VisitSerializers(serializers.ModelSerializer):
   status = StatusSerializer(many=True)
   class Meta:
       model = Visit

Вы можете передавать данные в VisitSerializer следующим образом.{, статус: [,]}

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...