當(dāng)前位置：首頁(yè) > 编程资源 > 综合教程 >内容正文

综合教程

（转）一个简单的rest_framework demo

發(fā)布時(shí)間：2023/10/11 综合教程 67 老码农

生活随笔收集整理的這篇文章主要介紹了（转）一个简单的rest_framework demo 小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.

轉(zhuǎn)發(fā)：https://www.cnblogs.com/fu-yong/p/9100559.html

models.py

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

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

from rest_framework.permissions import BasePermission

from app01 import models

class AdminPermission(BasePermission):

'''管理員權(quán)限認(rèn)證'''

message = '您沒(méi)有權(quán)限訪問(wèn)！'

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

from rest_framework.authentication import BaseAuthentication

from rest_framework.exceptions import AuthenticationFailed

from app01 import models

class BlackNameAuth(BaseAuthentication):

'''黑名單認(rèn)證'''

def authenticate(self, request):

BLACK_NAME_LIST = ['小花', '小翠']

# 通過(guò)從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('您沒(méi)有登錄或者被關(guān)小黑屋啦')

else:

return user.username,user_id

urls.py

from django.conf.urls import url,include

from django.contrib import admin

from rest_framework import routers

from app01 import views

# 實(shí)例化一個(gè)routers對(duì)象

routers = routers.DefaultRouter()

# 往對(duì)象里注冊(cè)（添加）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

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):

'''分頁(yè)，自定義每頁(yè)顯示2條'''

page_size = 2

class BWHViewSet(ModelViewSet):

'''在黑名單中的用戶禁止查看三圍信息'''

authentication_classes = [BlackNameAuth]

# 分頁(yè)

pagination_class = MyPagination

serializer_class = BWHModelSerializer

queryset = models.BWH.objects.all()

class UserViewSet(ModelViewSet):

'''非管理員禁止查看用戶信息'''

permission_classes = [AdminPermission]

# 分頁(yè)

pagination_class = MyPagination

serializer_class = UserModelSerializer

queryset = models.UserInfo.objects.all()

總結(jié)

以上是生活随笔為你收集整理的（转）一个简单的rest_framework demo的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。

上一篇： redirect和forward
下一篇： Linux脚本程序