DRF之权限

2022-04-08 23:33:07 阅读：168 来源： 互联网

标签：UserPermission classes user import 权限 class DRF

操作步骤

　　第一步：

　　　　写一个类，继承BasePermission，重写has_permission，判断如果有权限，返回True，如果没有权限，返回False

　　第二步:局部使用和全局使用

　　　　局部使用

    class BookDetailView(GenericViewSet, CreateModelMixin, DestroyModelMixin, UpdateModelMixin):
    permission_classes = [UserPermission, ]

　　　　全局使用

      REST_FRAMEWORK={
            "DEFAULT_PERMISSION_CLASSES":["app01.auth.UserPermission",]
        }

代码(需要先登录)

　　auth.py

from rest_framework.permissions import BasePermission


class UserPermission(BasePermission):
    def has_permission(self, request, view):
        # self.message = '你是普通用户，没有权限'  # 没有权限，提示信息
        self.message = '你是%s用户,没有权限' % request.user.get_user_type_display()  # 没有权限，提示信息
        # 如果有权限返回True，没有权限返回False
        # 权限类，在认证类知之后，request.user有了当前登录用户
        user_type = request.user.user_type
        if user_type < 3:  # 只要不是1和2，就没有权限
            return True
        else:
            return False

　　views.py

from rest_framework.viewsets import GenericViewSet
from rest_framework.mixins import ListModelMixin, CreateModelMixin, DestroyModelMixin, RetrieveModelMixin, \
    UpdateModelMixin


# 普通用户只能查看
class BookView(GenericViewSet, ListModelMixin, RetrieveModelMixin):
    authentication_classes = [LoginAuth, ]
    queryset = Book.objects.all()
    serializer_class = BookSerializer


from .auth import UserPermission


# 普通用户及以上才能修改
class BookDataView(GenericViewSet, CreateModelMixin, DestroyModelMixin):
    authentication_classes = [LoginAuth, ]
    permission_classes = [UserPermission, ]
    queryset = Book.objects.all()
    serializer_class = BookSerializer

　　urls.py

from django.urls import path, include
from app01 import views
from rest_framework.routers import SimpleRouter

router = SimpleRouter()
router.register('user', views.UserView, 'user')
router.register('books', views.BookView, 'books')
router.register('booksdata', views.BookDataView, 'booksdata')
urlpatterns = [
    path('admin/', admin.site.urls),
    path('', include(router.urls)),
]

代码需求说明

# 登录成功----》所有必须登录能访问---》每个视图类上加认证类
# 用户是普通用户---》普通用户可以访问所有和单条
# 普通管理员和超级用户可以操作所有，除了访问单条和所有的那个视图类，加上认证类
# books：查看一条，和所有
# booksdata路由下有：删除，新增，修改---》权限类加在这里

# book 5个接口，必须登录才能访问
# 5个接口分成了俩视图写：
        -BookView：获取所有，获取单条
      -BookDetailView：删除，修改，新增
        -这俩视图都需要登录：authentication_classes = [LoginAuth, ]
    -BookView只要登陆就可以操作
    -BookDetailView必须有权限才能，加了一个权限，permission_classes = [UserPermission, ]

标签：UserPermission,classes,user,import,权限,class,DRF
来源： https://www.cnblogs.com/zhaoyuanshi/p/16120320.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

DRF之权限