rest_framework之视图

写一个出版社的增删改查restful接口

models

from django.db import models

# Create your models here.


from django.db import models


# Create your models here.


class Book(models.Model):
    title = models.CharField(max_length=32)
    price = models.IntegerField()
    pub_date = models.DateField()
    publish = models.ForeignKey("Publish")
    authors = models.ManyToManyField("Author")

    def __str__(self):
        return self.title

    def test(self):
        return '555'


class Publish(models.Model):
    name = models.CharField(max_length=32)
    email = models.EmailField()
    # def __str__(self):
    #     return self.name


class Author(models.Model):
    name = models.CharField(max_length=32)
    age = models.IntegerField()

    def __str__(self):
        return self.name

视图

from django.shortcuts import render, HttpResponse
# Book相关操作
# Create your views here.
from api import models
from rest_framework.response import Response
from rest_framework.views import APIView
from rest_framework import serializers


class BookSer(serializers.ModelSerializer):
    class Meta:
        model = models.Book
        fields = "__all__"


class List():
    def list(self):
        ret = self.ret.objects.all()
        ser = self.book_ser(instance=ret, many=True)
        return Response(ser.data)


class Create():
    def create(self, request):
        back_msg = {'status': 0, 'data': None, 'msg': 'error happened'}
        ser = BookSer(data=request.data)
        if ser.is_valid():
            ser.save()
            back_msg['data'] = ser.data
            back_msg['status'] = 1
            back_msg['msg'] = 'create success'
        return Response(back_msg)


class Book(APIView, List, Create):
    ret = models.Book
    book_ser = BookSer

    def get(self, request):
        return self.list()

    def post(self, request):
        return self.create(request)


class BookDetail(APIView):
    def get(self, request, pk):
        ret = models.Book.objects.filter(pk=pk).first()
        ser = BookSer(instance=ret, many=False)
        return Response(ser.data)

    def put(self, request, pk):
        ret = models.Book.objects.filter(pk=pk).first()
        ser = BookSer(instance=ret, data=request.data)
        if ser.is_valid():
            ser.save()
            return Response(ser.data)

    def delete(self, request, pk):
        models.Book.objects.filter(pk=pk).delete()
        return HttpResponse('already deleted')

mixin类和generice类编写视图

from rest_framework.mixins import CreateModelMixin,RetrieveModelMixin,ListModelMixin,UpdateModelMixin,DestroyModelMixin
from rest_framework.generics import GenericAPIView
class PublishView(ListModelMixin,CreateModelMixin,GenericAPIView):
    queryset=models.Publish.objects.all()
    serializer_class=PublishSerializers

    def get(self, request):
        return self.list(request)

    def post(self, request):
        return self.create(request)

class PublishDetailView(RetrieveModelMixin,UpdateModelMixin,DestroyModelMixin,GenericAPIView):
    queryset=models.Publish.objects.all()
    serializer_class=PublishSerializers
    def get(self,request,*args,**kwargs):
        return self.retrieve(request,*args,**kwargs)
    def put(self,request,*args,**kwargs):
        return self.update(request,*args,**kwargs)
    def delete(self,request,*args,**kwargs):
        return self.destroy(request,*args,**kwargs)

使用generics 下ListCreateAPIView,RetrieveUpdateDestroyAPIView

from rest_framework.generics import ListCreateAPIView,RetrieveUpdateDestroyAPIView
class PublishView(ListCreateAPIView):
    queryset=models.Publish.objects.all()
    serializer_class=PublishSerializers

class PublishDetailView(RetrieveUpdateDestroyAPIView):
    queryset=models.Publish.objects.all()
    serializer_class=PublishSerializers

使用ModelViewSet

路由

 url(r'^publish/$', views.PublishView.as_view({'get':'list','post':'create'})),
    url(r'^publish/(?P<pk>d+)/$', views.PublishView.as_view({'get':'retrieve','put':'update','delete':'destroy'})),

视图

from rest_framework.viewsets import ModelViewSet
class PublishView(ModelViewSet):
    queryset=models.Publish.objects.all()
    serializer_class=PublishSerializers