重庆分公司,新征程启航
为企业提供网站建设、域名注册、服务器等服务
本篇文章给大家分享的是有关Django如何限制API访问频率,小编觉得挺实用的,因此分享给大家学习,希望大家阅读完这篇文章后可以有所收获,话不多说,跟着小编一起来看看吧。
成都创新互联公司是一家专业提供蔡甸企业网站建设,专注与成都网站设计、成都网站制作、成都h5网站建设、小程序制作等业务。10年已为蔡甸众多企业、政府机构等服务。创新互联专业网站建设公司优惠进行中。解决思路
1.Django官方插件库中有个django-ratelimit插件可以满足要求, django-ratelimit文档地址,很灵活很强大。只需要在我们的views函数上加上
@ratelimit(key='ip', rate='1/30s', block=True)
装饰器就可以了,网站上的资料不多,大部分都是英文的。在这里稍微解释下参数:
key='ip', 必填项,标识按照IP划分,我理解的是同一IP,遵循后面参数设定的规则。
rate='1/30s',必填项,设置的频率值,这个意思是30秒内执行一次,也可以按照 “分”,“时”,“日” 等划分,很灵活的配置,比如每分钟执行5次,可以这样写rate='5/m',这里参考文档足够弄明白
block=True,在这里吃了个亏,默认是False,加上了装饰器没写该参数,访问不受限制,没有达到间隔时间内不能再访问的预期效果,果断回去翻文档,
谷歌翻译:False是否阻止请求而不是注释。
我理解大概的意思是,当访问进来的时候是否去阻止它,把block=True之后,在次测试访问,可以看到403,确实是阻止了。
还有其它的参数,有更多需求的话可以看看,这是其一。
2.通过session存储访问时间
这里其实有两种方式,第一是写在中间件中,第二是装饰器,每个人需求不一样,我这一大堆函数就几个需要给外部调用的,干脆就做了装饰器,先贴码:
def limit(seconds = 5): ''' @func: 限制访问频率装饰器 ''' def rate_limit(func): def func_limit(request): now=time.time() request_time = request.session.get('request_time',0) interval_time = int(now - request_time) if interval_time < seconds: ret = ret_content(20008,'%s 秒后可再次访问'%(seconds-interval_time)) return JsonResponse(ret) else: request.session['request_time'] = time.time() ret = func(request) return ret return func_limit return rate_limit def ret_content(ret_code,message): return {'ret_code':ret_code,'message':message}