當前位置：首頁 > 编程资源 > 综合教程 >内容正文

综合教程

（转）一个简单的rest_framework demo

發布時間：2023/10/11 综合教程 68 老码农

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

轉發：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):

'''管理員權限認證'''

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

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

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

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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。

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