【问题标题】:How to Insert foreign key in other table using Django Rest Framework?如何使用 Django Rest Framework 在其他表中插入外键?
【发布时间】:2019-02-04 07:13:12
【问题描述】:

我正在使用 Django REST 框架,我被困在将外键数据插入另一个表中。 请参考以下场景:
如下代码所示,我想在template 表中插入数据,并使用单个发布请求将外键关系插入template_owners 表中。
我尝试了几个解决方案,但没有按预期工作,非常感谢帮助。

Models.py:

from django.db import models
from datetime import datetime
from django.utils import timezone

# Template model starts here.
class Template(models.Model):

    Yes = "Yes"
    No = 'No'

    STATUS = [
        (Yes, "Yes"),
        (No, "No"),
    ]

    class Meta:
        db_table = "template"
        ordering = ("date",)

    uuid = models.CharField(max_length=64, null=False, primary_key=True)
    name = models.CharField(max_length=255, null=False)
    hypervisor = models.CharField(max_length=32, null=False)
    download_url = models.CharField(max_length=255, null=False)
    description = models.CharField(max_length=255, null=True)
    metadata = models.CharField(max_length=255, null=True)
    user = models.CharField(max_length=64, null=True)
    command = models.CharField(max_length=32, null=True)
    size = models.FloatField(null=False, blank=True, default=0)
    is_default = models.CharField(max_length=3, choices=STATUS, default=Yes)
    date_modified = models.DateTimeField(default=timezone.now)
    date = models.DateTimeField(default=timezone.now)
    owners = models.CharField(max_length=255, null=True)

    def __str__(self):
        return "Template [ uuid : {} ]" . format(self.uuid)

# Template model ends here.

# Template Owners model starts here.
class TemplateOwners(models.Model):

    class Meta:
        db_table = "template_owners"
        ordering = ("date",)

    template_uuid = models.ForeignKey(Template, on_delete=models.CASCADE, db_column="template_uuid")
    gid = models.IntegerField(null=False, default=1)
    uid = models.IntegerField(null=False, default=1)
    date = models.DateTimeField(default=timezone.now)

    # def __init__(self):
    #   self.fields['name'].error_messages = {'required': 'Please let us know what to call you!'}

    def __str__(self):
        return "Template Owner [ template_uuid : {} ]" . format(self.template_uuid)

# Template Owners model ends here.

Serializers.py:

from rest_framework import serializers
from template.models import Template, TemplateOwners

def trigger_validator(self, data):
    if data["name"] == "":
        raise serializers.ValidationError({"name": "Template name is required"})

    return data

class TemplateOwnerSerializer(serializers.ModelSerializer):
    class Meta:
        model = TemplateOwners  
        fields = ("gid", "uid", "date", "template_uuid")

class TemplateSerializer(serializers.ModelSerializer):
    owners = TemplateOwnerSerializer(many=True)

    class Meta:
        model = Template
        fields = ("uuid", "name", "hypervisor", "download_url", "size", "date", "owners")

    def create(self, validated_data):
        owners_data = validated_data.pop('owners')
        template = Template.objects.create(**validated_data)
        for owner_data in owners_data:
            TemplateOwners.objects.create(template=template, **owner_data)
        return template

Views.py:

from django.http import Http404
from rest_framework.response import Response
from rest_framework import status

from template.models import Template
from template.serializers import TemplateSerializer
from template.serializers import TemplateOwnerSerializer
from rest_framework.permissions import IsAuthenticated
from rest_framework import generics

class TemplateList(generics.ListCreateAPIView):

    permission_classes = (IsAuthenticated,)
    queryset = Template.objects.all()
    serializer_class = TemplateSerializer

    def get(self, request, fromat=None):
        templates = Template.objects.all()
        serialized = TemplateSerializer(templates, many=True)
        return Response(serialized.data)

    def post(self, request, fromat=None):
        serializer = TemplateSerializer(data=request.data)
        if serializer.is_valid():
            serializer.save()

            #tempowner = TemplateOwnerSerializer(data=request.data)
            #if tempowner.is_valid():
            #tempowner.save()
            return Response(serializer.data, status=status.HTTP_201_CREATED)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)


class TemplateModify(generics.RetrieveUpdateDestroyAPIView):

    permission_classes = (IsAuthenticated,)
    serializer_class = TemplateSerializer

    def get_object(self, pk):
        try:
                return Template.objects.get(pk=pk)
        except Template.DoesNotExist:
                raise Http404

    def get(self, request, pk, fromat=None):
        template = self.get_object(pk)
        serialized = TemplateSerializer(template)
        return Response(serialized.data)

    def put(self, request, pk, fromat=None):
        template = self.get_object(pk)
        serializer = TemplateSerializer(template, data=request.DATA)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data)

        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

    def delete(self, request, pk, format=None):
        template = self.get_object(pk)
        template.delete()
        return Response(status=status.HTTP_204_NO_CONTENT)

Django Rest Framework中与Foreignkey相关的问题很少,但没有一个关心数据插入他们主要关注数据获取,所以在标记重复之前请先浏览整个场景。

【问题讨论】:

  • 发布堆栈跟踪以及您要插入哪个字段?
  • @Exprator,我想在 template_owners 表中插入 uuid、gid、uid、日期。

标签: django django-rest-framework


【解决方案1】:

您在序列化程序和创建方面几乎没有问题,因此请相应地更改它们

class TemplateOwnerSerializer(serializers.ModelSerializer):
    class Meta:
        model = TemplateOwners  
        fields = ("gid", "uid", "date")

class TemplateSerializer(serializers.ModelSerializer):
    owners_data = TemplateOwnerSerializer(many=True)

    class Meta:
        model = Template
        fields = ("uuid", "name", "hypervisor", "download_url", "size", "date", "owners_data")

    def create(self, validated_data):
        owners_data = validated_data.pop('owners_data')
        template = Template.objects.create(**validated_data)
        for owner_data in owners_data:
            TemplateOwners.objects.create(template_uuid=template, **owner_data)
        return template

【讨论】:

  • 感谢您的回答,它也帮助了我,但有一个问题:我将如何为另外一张桌子做同样的事情。我的意思是我需要将相同的外键和其他一些数据插入到另一个表中。我以同样的方式做到了,但它只在第一个表中插入数据,而不是在第二个表中。等待您的回复。
猜你喜欢
  • 1970-01-01
  • 2018-12-16
  • 2019-07-02
  • 2017-06-23
  • 2019-02-12
  • 2015-03-27
  • 2021-02-26
  • 2018-08-04
  • 2017-08-15
相关资源
最近更新 更多