下面是对单个的视图进行的设置的&＃xff1a;
请求的时候用postman然后发送信息
我们下面所有的举例都是在用户对Comment这个表的操作
首先先生成一个类似于COOKIE的字符串 发送给前端浏览器 然后下次它再访问带着这个认证字符串
登陆视图
#登陆视图class LoginView(APIView):def post(self,request):ret &＃61; {"status":0}print(request.data)username &＃61; request.data.get("username")password &＃61; request.data.get(&＃39;password&＃39;)user_obj &＃61; models.UserInfo.objects.filter(username &＃61; username,password &＃61; password).first() # 校验看传递进来的是否对的print(11111)if user_obj:print(2222222)#校验成功 加密并且写入 Token中它的token &＃61; get_token_code(username)models.Token.objects.update_or_create(defaults&＃61;{&＃39;token&＃39;:token},user &＃61; user_obj) # 有就更新 没有就创建ret[&＃39;token&＃39;] &＃61; token #加密的内容存起来else:ret[&＃39;status&＃39;] &＃61; 1ret[&＃39;error&＃39;] &＃61; "用户名或密码错误"return Response(ret)
生成请求的token字符串&＃xff1a;
#生成Token函数
def get_token_code(username):&＃39;&＃39;&＃39;根据用户名和时间戳生成用户登陆成功过的随机字符串:param username: 字符串格式用户名:return: 字符串格式Token&＃39;&＃39;&＃39;import timeimport hashlibtimestamp &＃61; str(time.time())m &＃61; hashlib.md5(bytes(username,encoding&＃61;"utf8"))m.update(bytes(timestamp,encoding&＃61;"utf8"))return m.hexdigest()
认证&＃xff1a; BaseAuthentication
from rest_framework.authentication import BaseAuthentication # 认证模块
抛错模块&＃xff1a;
from rest_framework.exceptions import AuthenticationFailed # 模块报错 返回信息
我们可以根据这个来设置 登陆人的信息 只有登陆成功后才能进行提交post之类的操作
这个时候就用到了我们的权限设置
我们写一个文件来存放登陆的信息设置&＃xff1a;
from rest_framework.authentication import BaseAuthentication # 认证模块from first import modelsfrom rest_framework.exceptions import AuthenticationFailed # 模块报错 返回信息
import logging
logger &＃61; logging.getLogger(__name__)
class MyAuth(BaseAuthentication):def authenticate(self, request):print(request.method)if request.method in ["POST","PUT","DELETE"]:token &＃61; request.data.get("token") #获取你登陆的信息token_obj &＃61; models.Token.objects.filter(token &＃61; token).first() #去数据库中阿奎那有没有这个数据if token_obj:return token_obj.user,tokenelse:raise AuthenticationFailed(&＃39;无效的token&＃39;)else:return None,None
views:
就是把你的认证设置导入进来 然后再用到post get操作的总操作的集合模块对操作设置
from first.utils.auth import MyAuth #导入你的 认证信息设置class CommentViewSet(ModelViewSet):queryset &＃61; models.Comment.objects.all()serializer_class &＃61; first_serializers.Commentserializer #这个是你的对Comment的定义设置的
authentication_classes &＃61; [MyAuth] # 让你设置的认证生效
视图级别认证
class CommentViewSet(ModelViewSet):queryset &＃61; models.Comment.objects.all()serializer_class &＃61; app01_serializers.CommentSerializerauthentication_classes &＃61; [MyAuth, ]
全局级别认证
# 在settings.py中配置
REST_FRAMEWORK &＃61; {"DEFAULT_AUTHENTICATION_CLASSES": ["app01.utils.MyAuth", ]
}
权限管理&＃xff1a;BasePermission
from rest_framework.permissions import BasePermission #导入权限设置的模块
也是设置一个权限管理的模块&＃xff1a;
from rest_framework.permissions import BasePermission #导入权限设置的模块class MyPermission(BaseException):message &＃61; &＃39;快滚&＃xff0c;没有权限&＃xff01;&＃39;def has_permission(self,request,view):&＃39;&＃39;&＃39;判断该用户有没有权限如果有就返回True如果没有就返回False&＃39;&＃39;&＃39;print(&＃39;我要进行自定义设置了&＃39;)return Truedef has_object_permission(self,request,view,obj): # 重写父类的方法 名字必须这样&＃39;&＃39;&＃39;obj是你的当前文章的作者对象判断当前的评论用户作者是不是你当前的用户只有评论的作者才能删除自己的评论&＃39;&＃39;&＃39;if request.method in [&＃39;PUT&＃39;,"DELETE"]:if obj.user &＃61;&＃61; request.user:# 当前要删除的评论的作者就是当前登陆的用户return Trueelse:return Falseelse:return True
views:
from first.utils.auth import MyAuth #导入你的 认证信息设置
from first.utils.permission import MyPermission #导入权限管理的模块class CommentViewSet(ModelViewSet):queryset &＃61; models.Comment.objects.all()serializer_class &＃61; first_serializers.Commentserializer #这个是你的对Comment的定义设置的
authentication_classes &＃61; [MyAuth,]permission_classes &＃61; [MyPermission,] # 把你权限管理的设置导入进来
这个时候你就要给你的url设置了 就像之前解除的 继承全局的然后并不需要设置很多url 设置一个就可以
ursl&＃xff1a;
from rest_framework.routers import DefaultRouterrouter &＃61; DefaultRouter()
router.register(r&＃39;comment&＃39;, views.CommentViewSet) #这个就是设置你的输入并且查找的视图
urlpatterns &＃43;&＃61; router.urls
限制:
from rest_framework.throttling import SimpleRateThrottle
定义一个访问次数限制&＃xff1a;
自定以的限制&＃xff1a;
class VisitThorttle(SimpleRateThrottle):scope &＃61; "ooo" #这个名字随便起 是你设置全局的时候的给对应的def get_cache_key(self, request, view):return self.get_ident(request) # 求当前访问的IP
其实上面是执行了这个流程&＃xff1a;
&＃39;&＃39;&＃39;
自定义访问限制类
&＃39;&＃39;&＃39;from rest_framework.throttling import BaseThrottle,SimpleRateThrottlefrom rest_framework.throttling import SimpleRateThrottleimport timeD &＃61; {}class MyThorttle(BaseThrottle):def allow_request(self, request, view):&＃39;&＃39;&＃39;返回True就放行 返回False 表示被限制了&＃39;&＃39;&＃39;# 获取当前的ipip &＃61; request.META.get("REMOTE_ADDR") # 获取请求的IPprint(&＃39;这是自定义希纳是类的allow_request&＃39;)print(ip)#获取当前时间now &＃61; time.time()#判断当前ip是否有访问记录if ip in D:D[ip] &＃61; [] #初始化一个空的访问历史列表#高端history &＃61; D[ip]while history and now -history[-1] > 30: #有ip并且当前时间减去储存的最后一个时间
history.pop()if len(history) >&＃61; 3:return Falseelse:# 把这一次的访问时间加到访问历史列表的第一位
D[ip].insert(0,now)return Trueclass VisitThorttle(SimpleRateThrottle):scope &＃61; "ooo" #这个名字随便起def get_cache_key(self, request, view):return self.get_ident(request) # 求当前访问的IP
views:
from first.utils.throttle import VisitThorttle # 导入自定义访问限制class CommentViewSet(ModelViewSet):queryset &＃61; models.Comment.objects.all()serializer_class &＃61; first_serializers.Commentserializer #这个是你的对Comment的定义设置的throttle_classes &＃61; [VisitThorttle,] #把自定义的访问限制类
全局设置 全局的请求设置是你只要进去就会触发 而我们在views中的设置 只是有针对的对一些视图进行设置
全局设置&＃xff1a;
在settings设置;
# rest framework相关的配置项
REST_FRAMEWORK &＃61; { 关于认证的全局配置 &＃39;DEFAULT_AUTHENTICATION_CLASSES&＃39;: [&＃39;app01.utils.auth.MyAuth&＃39;, ], &＃39;DEFAULT_PERMISSION_CLASSES&＃39;: [&＃39;app01.utils.permission.MyPermission&＃39;], #"DEFAULT_THROTTLE_CLASSES": ["app01.utils.throttle.MyThrottle", ],"DEFAULT_THROTTLE_RATES": { # 这个和上面的都可以设置 "ooo": "5/m", # ooo对应你后自定以的设置的scope"xxx": &＃39;10/m&＃39;}
}
京公网安备 11010802041100号