django rest framework 向数据库中插入数据时处理外键的方法

时间:2023-03-08 23:56:31
django rest framework 向数据库中插入数据时处理外键的方法

一.models.py中

from django.db import models

class UserModel(models.Model)
user_name = models.CharField() class MyModel(models.Model)
author = models.Foreignkey(user)
age = models.CharField()

二. 序列化文件 serializers.py 中创建序列化类

from rest_framework.serializers import Serializer
from models import MyModel class MySerializer( Serializer.ModelSerializers): # 自定义字段
user_name = serializers.SerializerMethodField() class Meta:
model = MyModel
fields = ("user_name", "age") # 处理自定义的字段返回用户名, 通过外键获取UserModel的数据
def get_user_name(self, obj):
return obj.author.user_name def create(self, validated_data)
# 处理外键字段
  return MyModel.objects.create(author=self.context["author"], **validated_data)

三.视图文件views.py中定义视图

# 视图函数中
from rest_framework.views import APIView # 类视图中使用,集成自该类
from rest_framework.decorators import api_view # 方法视图中使用,是一个装饰器,直接装饰方法视图 from rest_framework import status
from rest_framework.response import Response # 直接可以将字典数据转换成json数据
from models import MyModel
from serializers import MySerializer
from rest_framework.request import Request
from rest_framework import exceptions @api_view(['GET', 'POST', 'PUT', 'DELETE']) # 指明允许那些请求方式进行请求
def api_list(request): if request.method == 'GET':
# 向数据库查询数据,得到查询集
try:
query_set = MyModel.objects.all()
except Snippet.DoesNotExist:
return Response(status=status.HTTP_417_EXPECTATION_FAILED)
# 调用序列化类对象,返回进行了序列化的字段集合,使用serializers.data方式 获取数据
serializers = MySerializer(query_set, many=True)
# 取出数据,向客户端进行返回,Response()会将数据转化为json数据.
return Response(serializers.data, status=status.HTTP_200_OK) # 前端提交数据,调用模型,将其保存到数据库中
elif request.method == 'POST':
     user = request.user
#将外键数据插入 这里插入一个user对象,就是外键所对应的主表的一个对象
serializer = MySerializer(data= request.data, context={"author":user})
# 如果反序列化的对象存在,就说明数据有效,将数据保存到数据库中
if serializer.is_valid():
# 调用save(), 从而调用序列化对象的create()方法,创建一条数据
serializer.save()
return Response(serializer.data, status=status.HTTP_200_OK)
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)