（轉）一個簡單的rest_framework demo

轉發：http://www.javashuo.com/article/p-sarssnwq-eq.html

 

 

models.py

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15

from  django.db  import  models   class  UserInfo(models.Model):      username  =  models.CharField(max_length = 32 )      password  =  models.CharField(max_length = 32 )      role_choice  =  (( 1 , '管理員' ),( 2 , '普通用戶' ))      role  =  models.IntegerField(choices = role_choice)   class  BWH(models.Model):      '''三圍信息'''      bust  =  models.IntegerField(verbose_name = '胸圍' )      waist  =  models.IntegerField(verbose_name = '腰圍' )      hips  =  models.IntegerField(verbose_name = '臀圍' )        user  =  models.OneToOneField(to = UserInfo)

　　

serializers.py

1 2 3 4 5 6 7 8 9 10 11 12 13

from  rest_framework  import  serializers from  app01  import  models   class  BWHModelSerializer(serializers.ModelSerializer):      class  Meta:          model  =  models.BWH          fields  =  '__all__'     class  UserModelSerializer(serializers.ModelSerializer):      class  Meta:          model  =  models.UserInfo          fields  =  '__all__'

　　

permission.py

1 2 3 4 5 6 7 8 9 10 11 12 13 14

from  rest_framework.permissions  import  BasePermission from  app01  import  models   class  AdminPermission(BasePermission):      '''管理員權限認證'''      message  =  '您沒有權限訪問！'      def  has_permission( self , request, view):          user_id  =  request.GET.get( 'uid' )          user  =  models.UserInfo.objects. filter (pk = user_id).first()            if  not  user  or  user.role  = =  2 :              return  False          else :              return  True

　　

auth.py

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18

from  rest_framework.authentication  import  BaseAuthentication from  rest_framework.exceptions  import  AuthenticationFailed from  app01  import  models   class  BlackNameAuth(BaseAuthentication):      '''黑名單認證'''        def  authenticate( self , request):          BLACK_NAME_LIST  =  [ '小花' ,  '小翠' ]            # 經過從url獲取user_id的方式模擬用戶登陸          user_id  =  request.GET.get( 'uid' )          user  =  models.UserInfo.objects. filter (pk = user_id).first()            if  not  user  or  user.username  in  BLACK_NAME_LIST:              raise  AuthenticationFailed( '您沒有登陸或者被關小黑屋啦' )          else :              return  user.username,user_id

　　

urls.py

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15

from  django.conf.urls  import  url,include from  django.contrib  import  admin from  rest_framework  import  routers from  app01  import  views   # 實例化一個routers對象 routers  =  routers.DefaultRouter() # 往對象裏註冊（添加）url routers.register( 'bwh' ,views.BWHViewSet) routers.register( 'user' ,views.UserViewSet)   urlpatterns  =  [      url(r '^admin/' , admin.site.urls),      url(r'',include(routers.urls)) ]

　　

view.py

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35

from  app01.serializers  import  BWHModelSerializer from  app01.serializers  import  UserModelSerializer from  rest_framework.viewsets  import  ModelViewSet from  app01  import  models from  app01.auth  import  BlackNameAuth from  app01.permission  import  AdminPermission from  rest_framework.pagination  import  PageNumberPagination   class  MyPagination(PageNumberPagination):      '''分頁，自定義每頁顯示2條'''      page_size  =  2       class  BWHViewSet(ModelViewSet):      '''在黑名單中的用戶禁止查看三圍信息'''      authentication_classes  =  [BlackNameAuth]        # 分頁      pagination_class  =  MyPagination        serializer_class  =  BWHModelSerializer      queryset  =  models.BWH.objects. all ()       class  UserViewSet(ModelViewSet):      '''非管理員禁止查看用戶信息'''      permission_classes  =  [AdminPermission]        # 分頁      pagination_class  =  MyPagination        serializer_class  =  UserModelSerializer      queryset  =  models.UserInfo.objects. all ()