from rest_framework import permissions


class IsAdminUserOrReadOnly(permissions.BasePermission):
    """
    仅管理员用户可以进行修改
    其他用户仅可以查看
    """
    # 钩子方法
    def has_permission(self, request, view):
        # 所有人允许 GET, HEAD, OPTIONS 请求
        if request.method in permissions.SAFE_METHODS:
            return True
        # 仅管理员可以其他操作
        return request.user.is_superuser