分别基于 Django_rest_framework 的 APIView、GenericAPIView 写的增删查改接口
为演示方便,增删查改接口仅限于单张book表,其他表方法一样,数据库使用的是自带的 sqlite3。
写接口前的准备
创建模型表
# models.py
from django.db import models
class BaseModel(models.Model):
"""
将图书表、出版社表、作者表都有的三个字段写在一个 BaseModel 类中,直接继承即可
但是本 BaseModel 类也会生成一张表,如果不想要这张表,可以在内部的 Meta 类中定义 abstract=True
"""
is_delete = models.BooleanField(default=False) # 默认是 False,不删除
# 创建时间
create_time = models.DateTimeField(auto_now_add=True)
# auto_now_add=True 只要记录创建,不需要手动插入时间,自动将当前时间插入
# 另一种写法:
# import datetime
# create_time = models.DateTimeField(default=datetime.datetime.now) # now 不能加括号,不然记录的是每次程序启动的时间
# 最后更新时间
last_update_time = models.DateTimeField(auto_now=True)
# auto_now=True 只要更新,就把当前时间插入
class Meta:
# 单个字段,有索引,可唯一
# 多个字段,有联合索引,可联合唯一
abstract = True # 抽象表,不在数据库中建立出表
class Book(BaseModel):
id = models.AutoField(primary_key=True) # id 字段可以不写,不写默认也有。但所有的表最好统一,写就都写
name = models.CharField(max_length=32, help_text="这里填书名")
price = models.DecimalField(max_digits=5, decimal_places=2)
""" 外键 """
publish = models.ForeignKey(to="Publish", on_delete=models.DO_NOTHING, db_constraint=False)
# to_field 参数默认不用写,作用是关联到 Publish 主键
# db_constraint=False 表示这个外键只是逻辑上的关联,实质上没有外键联系。但不会影响增删改和 orm 查询
"""
表断关联:
1、表之间没有外键关联,但是有外键逻辑关联(有充当外键的字段)
2、断关联后不会影响数据库查询效率,但是会极大的提高数据库增删改的效率(不影响增删改查操作)
3、断关联一定要通过逻辑保证表之间数据的安全,不要出现脏数据(一般由程序员在代码层面限制)
4、断关联后的级联关系:
作者删掉,作者详情也跟着删掉:on_delete=models.CASCADE
出版社删掉,图书还在:on_delete=models.DO_NOTHING
部门删掉,员工还在,员工的部门可以为空:null=True,on_delete=models.SET_NULL
部门删掉,员工进入默认部门(默认值):default=0, on_delete=models.SET_DEFAULT
"""
# 多对多的第三张表,只有关联字段时,用自动就够了,有扩展字段,用半自动
# 因为第三张表一共有三张表互相关联,所以不能写 on_delete
authors = models.ManyToManyField(to="Author", db_constraint=False)
def __str__(self):
return self.name
class Meta:
verbose_name_plural = "图书表" # admin 中显示中文表名
@property
def publish_name(self):
return self.publish.name
def author_list(self):
author_list = self.authors.all()
lis = []
for author in author_list:
lis.append({"name": author.name, "sex": author.get_sex_display()})
return lis
# 也可以用列表推导式一行完成
# return [{"name": author.name, "sex": author.get_sex_display()} for author in author_list]
class Publish(BaseModel):
id = models.AutoField(primary_key=True)
name = models.CharField(max_length=32)
addr = models.CharField(max_length=32)
def __str__(self):
return self.name
class Meta:
verbose_name_plural = "出版社表"
class Author(BaseModel):
name = models.CharField(max_length=32)
sex = models.IntegerField(choices=((1, "男"), (2, "女")))
""" 外键 """
# OneToOneField 本质就是 ForeignKey + unique
authordetail = models.OneToOneField(to="AuthorDetail", db_constraint=False, on_delete=models.CASCADE)
def __str__(self):
return self.name
class Meta:
verbose_name_plural = "作者表"
class AuthorDetail(BaseModel):
mobile = models.CharField(max_length=11)
def __str__(self):
return self.name
class Meta:
verbose_name_plural = "作者详情表"
写一个序列化类
# ser.py
from rest_framework import serializers
from api import models
# 写一个类,继承 ListSerializer,重写 update
class BookListSerializer(serializers.ListSerializer):
# 重写 create 方法,加了一个打印 validated_data 功能
# def create(self, validated_data):
# print(validated_data)
# return super().create(validated_data)
def update(self, instance, validated_data):
# instance ==> [book对象1, book对象2 ...]
# validated_data ==> [{name: xx, price: xx}, {name: xx, price: xx} ...]
# self.child 是 BookModelSerializer 对象,
# 想保存数据,可以使用 BookModelSerializer 的 update 方法,
return [
# self.child.update(对象,字典) for attrs in validated_data
self.child.update(instance[i], attrs) for i, attrs in enumerate(validated_data)
# 通过 enumerate 枚举出 validated_data 这个列表的 index(i) 和 字典(attrs)
]
class BookModelSerializer(serializers.ModelSerializer):
# 如何让前端收到的数据显示的是名称而不是id号:
# 方法一:
# publish = serializers.CharField(source="publish.name") # 只序列化可以,反序列化就有问题了
# 方法二,在 models.py 中对应的表里写一个方法 publish_name 就可以直接放在 fields 中
"""
@property
def publish_name(self):
return self.publish.name
"""
class Meta:
list_serializer_class = BookListSerializer
model = models.Book
# fields = "__all__"
# depth = 1 # 展示数据的深度,数字表示展示几层,很少用
fields = ("name", "price", "authors", "publish", "publish_name", "author_list")
# 但前端会收到两个 publish,一个是 id 号,一个是名字。解决方法是定义一个为只写,一个为只读
extra_kwargs = {
"authors": {"write_only": True},
"author_list": {"read_only": True},
"publish": {"write_only": True},
"publish_name": {"read_only": True},
}
自定义异常处理和自定义Response
最好在项目文件夹下新建一个 utils 文件夹。
将自定义的异常处理写在 exceptions.py 中,放在 utils 文件夹内。
将自定义的Response写在 response.py 中,放在 utils 文件夹内。
但本次在视图类中没有使用自定义Response,只使用了自定义异常处理。
自定义异常处理
# exceptions.py
from rest_framework.views import exception_handler
from rest_framework import status
from rest_framework.response import Response
def my_exception_handler(exc, context):
response = exception_handler(exc, context)
if not response: # 如果返回的是 None
# if isinstance(exc, ZeroDivisionError): # 如果视图类中有特定错误就可以做出相应的返回
# return Response(data={"status": 777, "msg": "除以0的错误" + str(exc)}, status=status.HTTP_400_BAD_REQUEST)
return Response(data={"status": 999, "msg": str(exc)}, status=status.HTTP_400_BAD_REQUEST)
else: # 如果返回的是 response 对象
return Response(data={"status": 888, "msg": response.data.get("detail")}, status=status.HTTP_400_BAD_REQUEST)
自定义Response
# response.py
from rest_framework.response import Response
class APIResponse(Response):
def __init__(self, code=100, msg="成功", data=None, status=None, headers=None, **kwargs):
dic = {"code": code, "msg": msg}
if data:
dic = {"code": code, "msg": msg, "data": data}
dic.update(kwargs)
super().__init__(data=dic, status=status, headers=headers)
全局配置自定义的异常处理
# settings.py
REST_FRAMEWORK = {"EXCEPTION_HANDLER": "utils.exceptions.my_exception_handler"}
配置URL
URL的配置使用了路由分发
# 项目下的 urls.py
from django.conf.urls import url, include
from django.contrib import admin
urlpatterns = [
url(r"^admin/", admin.site.urls),
url(r"^api/", include("api.urls"))
]
# 应用下的 urls.py
from django.conf.urls import url
from api import views
urlpatterns = [
url(r"^books/$", views.BookAPIView.as_view()),
url(r"^books/(?P<pk>d+)", views.BookAPIView.as_view()),
url(r"^books2/$", views.Book2APIView.as_view()),
url(r"^books2/(?P<pk>d+)", views.Book2APIView.as_view()),
]
视图类接口的编写
基于 APIView 的视图类接口
from api import models
from rest_framework.views import APIView
from rest_framework.response import Response
from api.ser import BookModelSerializer
class BookAPIView(APIView):
def get(self, request, *args, **kwargs):
"""查询单条数据和查询所有数据合在一个方法中"""
# 查询单条数据
if kwargs.get("pk", None):
book = models.Book.objects.filter(pk=kwargs.get("pk")).first()
book_ser = BookModelSerializer(instance=book)
return Response(book_ser.data)
# 查询所有数据
book_list = models.Book.objects.all().filter(is_delete=False) # 查询所有没有被删的书
book_list_ser = BookModelSerializer(book_list, many=True)
return Response(data=book_list_ser.data)
def post(self, request, *args, **kwargs):
"""同时具备增单条和增多条的功能"""
# 如果是字典,就是单条数据新增
if isinstance(request.data, dict):
book_ser = BookModelSerializer(data=request.data)
book_ser.is_valid(raise_exception=True)
book_ser.save()
return Response(data=book_ser.data)
# 如果是列表,就是多条数据新增,列表内部是多个字典(单条数据)
elif isinstance(request.data, list):
book_ser = BookModelSerializer(data=request.data, many=True) # 增多条
# 有了 many=True 这个参数,book_ser 就是 ListSerializer 的对象了
# print(type(book_ser)) # 输出:<class "rest_framework.serializers.ListSerializer">
book_ser.is_valid(raise_exception=True)
book_ser.save() # ListSerializer 的 create 方法
"""
create 方法内部就是一个for循环,self.child 就是 BookModelSerializer 对象
也就是说,增多条的 ListSerializer 内部其实就是for循环了一个个增单条的 BookModelSerializer
def create(self, validated_data):
return [
self.child.create(attrs) for attrs in validated_data
]
"""
return Response(data=book_ser.data)
def put(self, request, *args, **kwargs):
"""同时具备单条修改和多条修改功能"""
# 单条数据修改
if kwargs.get("pk", None):
book = models.Book.objects.filter(pk=kwargs.get("pk")).first()
book_ser = BookModelSerializer(instance=book, data=request.data, partial=True)
# partial=True:允许局部修改,如只修改书名和价格
# 如果不写这个字段想达到同样的效果,要在 ser.py 的 extra_kwargs 中给每个字段都加上 require=False
book_ser.is_valid(raise_exception=True)
book_ser.save()
return Response(data=book_ser.data)
# 多条数据修改
else:
"""
前端传来的数据格式必须是:[{id: 1, name: xx, price: xx}, {id: 2, name: xx, price: xx}]
要先将数据处理成:对象列表[book对象1, book对象2] 和 要修改的数据列表[{name: xx, price: xx}, {name: xx, price: xx}]
"""
book_list = []
modify_data = []
for item in request.data:
# 前端传来的 request.data 可能不是列表套字典的格式,直接处理可能会报错,所以最好有全局的异常处理进行捕获
pk = item.pop("id")
book = models.Book.objects.get(pk=pk)
book_list.append(book)
modify_data.append(item)
"""
处理好后再修改数据,有两种方案:
第一种方案:
直接在视图里写,简单
for循环 book_ser = BookModelSerializer(instance=对象列表, data=修改的数据列表) 一个一个修改。
"""
# for i, is_data in enumerate(modify_data):
# book_ser = BookModelSerializer(instance=book_list[i], data=is_data)
# book_ser.is_valid(raise_exception=True)
# book_ser.save()
# return Response("成功")
"""
第二种方案:
在序列化类中写,不常用
book_ser = BookModelSerializer(instance=对象列表, data=修改的数据列表, many=True)
但要重写 ListSerializer 的 update 方法,
因为此时 book_ser 是 ListSerializer 的对象,而源码中没写 update 方法,
需要自己写一个类继承 ListSerializer,里面重写 update 方法。
问题在于如何让 BookModelSerializer 和自己写的类建立联系?
查看源码可知,ListSerializer 的父类 BaseSerializer 的 __new__() 中调用了 many_init()
many_init() 内部的如下代码通过反射,在meta中找不到 list_serializer_class 时才会走 ListSerializer
list_serializer_class = getattr(meta, "list_serializer_class", ListSerializer)
因此可以在 ser.py 中自己写的 BookModelSerializer 的 Meta 类中定一个变量叫 list_serializer_class,
把自己写的 BookListSerializer 这个类赋给这个变量,在 BookListSerializer 内重写 update 方法
这样只要修改数据时 BookModelSerializer 传了 many=True,就走的是 BookListSerializer 的 update 方法
同样,如果重写了 create 方法,新增数据时只要传了 many=True,就走的是 BookListSerializer 的 create 方法
"""
book_ser = BookModelSerializer(instance=book_list, data=modify_data, many=True)
book_ser.is_valid(raise_exception=True)
book_ser.save()
return Response(book_ser.data)
def delete(self, request, *args, **kwargs):
"""单个删除和批量删除"""
pk = kwargs.get("pk") # 如果是单条删除,pk=要删除数据的id值,多条删除,pk=None
pks = []
if pk: # 如果有值,说明是单条删除
pks.append(pk) # 即使是单条数据也放在一个列表中,和多条删除用同一个方法处理
else: # 如果没有值,是多条删除
# 如果是多条删除,前端发来的数据格式必须是:{"pks": [要删除的多条数据的id值]},这是下面一行代码规定的
# 获取到要删除的多条id值组成的列表
pks = request.data.get("pks")
# 最后,不论是单条还是多条删除,pks都是一个列表,里面是要删除数据的id值
ret = models.Book.objects.filter(pk__in=pks, is_delete=False).update(is_delete=True)
"""
pk__in:后面必须等于一个列表,是依次拿出列表里的每一项
is_delete=False:确保之前没有被删除过的数据被过滤出来
.update(is_delete=True):用 update 方法加上 is_delete=True 来标识要被删除的数据
ret:返回值是受影响的行数(如果是0,表示要删的数据之前都已被标识为已删除了)
"""
if ret:
return Response(data={"msg": "删除成功!删除了%s条数据" % ret})
return Response(data={"msg": "没有要删除的数据..."})
基于 GenericAPIView 的视图类接口
from api import models
from rest_framework.generics import GenericAPIView
from rest_framework.response import Response
from api.ser import BookModelSerializer
class Book2APIView(GenericAPIView):
queryset = models.Book.objects.all()
serializer_class = BookModelSerializer
def get(self, request, *args, **kwargs):
"""单查和群查"""
if kwargs.get("pk"):
book = self.get_object()
book_ser = self.get_serializer(book)
return Response(book_ser.data)
# 查询所有数据
book_list = self.get_queryset().filter(is_delete=False) # 查询所有没有被删的书
book_list_ser = self.get_serializer(book_list, many=True)
return Response(data=book_list_ser.data)
def post(self, request, *args, **kwargs):
"""单增和群增"""
if isinstance(request.data, dict):
book_ser = self.get_serializer(data=request.data)
book_ser.is_valid(raise_exception=True)
book_ser.save()
return Response(data=book_ser.data)
elif isinstance(request.data, list):
book_ser = self.get_serializer(data=request.data, many=True)
book_ser.is_valid(raise_exception=True)
book_ser.save()
return Response(data=book_ser.data)
def put(self, request, *args, **kwargs):
"""单修和群修"""
if kwargs.get("pk"):
book = self.get_object()
book_ser = self.get_serializer(instance=book, data=request.data, partial=True)
book_ser.is_valid(raise_exception=True)
book_ser.save()
return Response(data=book_ser.data)
else:
book_list = []
modify_data = []
for item in request.data:
pk = item.pop("pk")
book = self.get_queryset().get(pk=pk)
# book = self.get_queryset().filter(pk=pk).first() # 这样写也可以
book_list.append(book)
modify_data.append(item)
book_ser = self.get_serializer(instance=book_list, data=modify_data, many=True, partial=True)
book_ser.is_valid(raise_exception=True)
book_ser.save()
return Response(book_ser.data)
def delete(self, request, *args, **kwargs):
"""单删和多删"""
pk = kwargs.get("pk")
pks = []
if pk:
pks.append(pk)
else:
pks = request.data.get("pks")
ret = self.get_queryset().filter(pk__in=pks, is_delete=False).update(is_delete=True)
if ret:
return Response(data={"msg": "删除成功!删除了%s条数据" % ret})
return Response(data={"msg": "没有可删除的数据..."})
hmoban主题是根据ripro二开的主题,极致后台体验,无插件,集成会员系统
自学咖网 » 分别基于 Django_rest_framework 的 APIView、GenericAPIView 写的增删查改接口
自学咖网 » 分别基于 Django_rest_framework 的 APIView、GenericAPIView 写的增删查改接口