首页
学习
活动
专区
圈层
工具
发布
社区首页 >专栏 >Django限制API访问频率的几种思路

Django限制API访问频率的几种思路

作者头像
kirin
发布于 2021-02-04 16:04:40
发布于 2021-02-04 16:04:40
2.4K0
举报
文章被收录于专栏:Kirin博客Kirin博客

需求描述:

平台中需要编写接口供第三方调用,需要控制调用频率,需求为5s内调用一次后不得再次调用。

解决思路

1.Django官方插件库中有个django-ratelimit插件可以满足要求, django-ratelimit文档地址,很灵活很强大。只需要在我们的views函数上加上

代码语言:javascript
复制
@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存储访问时间

这里其实有两种方式,第一是写在中间件中,第二是装饰器,每个人需求不一样,我这一大堆函数就几个需要给外部调用的,干脆就做了装饰器,先贴码:

代码语言:javascript
复制
  1. def limit(seconds = 5):
  2. ”’
  3. @func: 限制访问频率装饰器
  4. ”’
  5. def rate_limit(func):
  6. def func_limit(request):
  7. now=time.time()
  8. request_time = request.session.get(‘request_time’,0)
  9. interval_time = int(now – request_time)
  10. if interval_time < seconds:
  11. ret = ret_content(20008,‘%s 秒后可再次访问’%(seconds-interval_time))
  12. return JsonResponse(ret)
  13. else:
  14. request.session[‘request_time’] = time.time()
  15. ret = func(request)
  16. return ret
  17. return func_limit
  18. return rate_limit
代码语言:javascript
复制
代码语言:javascript
复制
  1. def ret_content(ret_code,message):
  2. return {‘ret_code’:ret_code,‘message’:message}
代码语言:javascript
复制

大概思路是:将当前访问的时间存session,设置时间间隔,当在时间间隔之内的时候,不让其刷新,并返回json,超过时间间隔,更新session,让其执行。使用如下:

代码语言:javascript
复制
@limit(seconds=30)

30s内执行一次,我这个乞丐版的没第一种方法灵活了,不过在禁止期间返回的结果值会比较友好,不是403。程序处理会比较方便,另外针对IP的情况,我这里没做处理,需要的话自己可以改改,把request_time换成IP + Salt方式就可以了。

外部IP地址获取代码这里也贴出来:

代码语言:javascript
复制
  1. def get_remote_cli_ip(request):
  2. ”’
  3. @func:获取客户端ip
  4. ”’
  5. ip = request.META.get(‘HTTP_X_FORWARDED_FOR’,0)
  6. if ip == 0:
  7. return request.META[‘REMOTE_ADDR’]
  8. else:
  9. return ip
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021/02/03 ,如有侵权请联系 cloudcommunity@tencent.com 删除
目录
  • 需求描述:
  • 解决思路
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档