rest_framework 访问频率的限制(节流)
对于用户访问频率的显示，rest_framework 也有自己的类进行约束
- 先来一个自己的基于它的类的节流类的限制
class VisitControl(BaseThrottle):
"""
访问评论限制
"""
def __init__(self):
self.history = None
def allow_request(self, request, view):
ctime = time.time()
remote_addr = self.get_ident(request)
if remote_addr not in VISIT_DICT:
VISIT_DICT[remote_addr] = [ctime, ]
return True
history_list = VISIT_DICT.get(remote_addr)
self.history = history_list
"""
history_list[-1] < ctime - 10
这个--> 如果当前的时间 减去 60秒 之后 与列表里面的 最后一个 就是最早访问的 时间 比较
如果 比他大 证明 这个 时间 无效了 ，把他 去除 ，然后在比较前一个 直到
当前时间 减去60秒之后 比 最后一个小, 证明 这个时间在60秒内，第一个条件就符合了
然后 剩下的 列表里面的值就都符合了(因为 列表里面最前面的元素 时间约靠近当下时间)
然后拿着这个列表里面的值 去判断 (因为这些值都是 60 秒内访问的) 如果小于3 证明 你还有一次机会
让他访问 ，但是 需要 把他加到 列表里面去,
以便下次判断列表个数的时候使用(每次符合要求的时间都需要加到列表里面,共后续使用)
"""
while history_list and history_list[-1] < ctime - 10:
history_list.pop()
if len(history_list) < 3:
history_list.insert(0, ctime)
return True
def wait(self):
ctime = time.time()
return 60 - (ctime - self.history[-1])
其实rest_framework已经为我们提供了基于用户的节流类，以及基于IP的节流类，我们只需要继承这个两个类 分别实现各自的方法即可:
class UserThrottle(SimpleRateThrottle):
scope = "User" # 登录用户访问频率的限制
def get_cache_key(self, request, view):
return request.user.username
class IpThrottle(SimpleRateThrottle):
scope = "AsyUser" # 匿名用户访问频率的限制
def get_cache_key(self, request, view):
return self.get_ident(request)
"""
注意的是全局的settings配置如下
"""
"DEFAULT_AUTHENTICATION_CLASSES": ["api.utils.permission.MyAuthtication",], # 认证全局配置
"DEFAULT_PERMISSION_CLASSES": [], # 权限的 全局配置
# 节流的频率控制
"DEFAULT_THROTTLE_RATES": {
"Luffy": "10/m",
"User": "20/m"
},
# 节流的类，默认是没有的！
"DEFAULT_THROTTLE_CLASSES": ["api.utils.permission.UserThrottle",]
rest_framework版本控制
首先是自己写在url的get请求的版本控制比如
url="xxxxxx/?version=v1"
对应的类方法如下：
class MyVersion(BaseVersioning):
"""
自己 的类的方法
"""
def determine_version(self, request, *args, **kwargs):
version = request.query_params.get("version")
return version
看一下其他部分的源码:
## 版本的控制的
## version 是版本号，这个scheme 是你的解析版本的类
## 解析版本的类只允许有一个
version, scheme = self.determine_version(request, *args, **kwargs)
# 版本赋值给 request.version 类赋值给request.versioning_scheme
request.version, request.versioning_scheme = version, scheme
一般使用这个内之类即可
from rest_framework.versioning import URLPathVersioning
class URLPathVersioning(BaseVersioning):
"""
To the client this is the same style as `NamespaceVersioning`.
The difference is in the backend - this implementation uses
Django's URL keyword arguments to determine the version.
An example URL conf for two views that accept two different versions.
urlpatterns = [
url(r'^(?P<version>[v1|v2]+)/users/$', users_list, name='users-list'),
url(r'^(?P<version>[v1|v2]+)/users/(?P<pk>[0-9]+)/$', users_detail, name='users-detail')
]
GET /1.0/something/ HTTP/1.1
Host: example.com
Accept: application/json
"""
invalid_version_message = _('Invalid version in URL path.')
def determine_version(self, request, *args, **kwargs):
version = kwargs.get(self.version_param, self.default_version)
if not self.is_allowed_version(version):
raise exceptions.NotFound(self.invalid_version_message)
return version
def reverse(self, viewname, args=None, kwargs=None, request=None, format=None, **extra):
if request.version is not None:
kwargs = {} if (kwargs is None) else kwargs
kwargs[self.version_param] = request.version
return super(URLPathVersioning, self).reverse(
viewname, args, kwargs, request, format, **extra
)