1. 程式人生 > 程式設計 >Django 對IP訪問頻率進行限制的例子

Django 對IP訪問頻率進行限制的例子

REST_FRAMEWORK 配置

對使用 rest_framework 框架的專案來說,可以使用框架的設定來對api的訪問頻率進行限制

REST_FRAMEWORK = {
   'DEFAULT_PARSER_CLASSES': (
    'rest_framework.parsers.JSONParser','rest_framework.parsers.FormParser','rest_framework.parsers.MultiPartParser',),'DEFAULT_AUTHENTICATION_CLASSES': (
    # 'lecare.core.rest_auth.CrossSiteSessionAuthentication','DEFAULT_PERMISSION_CLASSES': [
    # 'rest_framework.permissions.IsAuthenticated','rest_framework.permissions.AllowAny',],'PAGE_SIZE': 20,'UNICODE_JSON': False,# 'COERCE_DECIMAL_TO_STRING': False,# 'EXCEPTION_HANDLER': 'lecare.core.custom_exception_handler.custom_exception_handler','JWT_EXPIRATION_DELTA': datetime.timedelta(hours = 2),'JWT_REFRESH_EXPIRATION_DELTA': datetime.timedelta(days = 360),'JWT_ALLOW_REFRESH': False,'JWT_AUTH_HEADER_PREFIX': 'JWT','JWT_PAYLOAD_HANDLER': 'consumer.jwt_conf.jwt_payload_handler','JWT_RESPONSE_PAYLOAD_HANDLER': 'consumer.jwt_conf.jwt_response_payload_handler','JWT_GET_USER_SECRET_KEY': 'consumer.jwt_conf.jwt_get_secret_key',# 'DEFAULT_THROTTLE_CLASSES': (
  #   # 開啟匿名使用者介面請求頻率限制
  #   'rest_framework.throttling.AnonRateThrottle',#   # 開啟授權使用者介面請求頻率限制
  #   'rest_framework.throttling.UserRateThrottle'
  # ),# 'DEFAULT_THROTTLE_RATES': {
  #   # 頻率限制有second,minute,hour,day
  #   # 匿名使用者請求頻率
  #   'anon': '30/second',#   # 授權使用者請求頻率
  #   'user': '30/second'
  # }
}

使用middleware中介軟體來限制IP頻率

import time
from django.utils.deprecation import MiddlewareMixin
MAX_REQUEST_PER_SECOND=2 #每秒訪問次數

class RequestBlockingMiddleware(MiddlewareMixin):
  def process_request(self,request):
    now=time.time()
    request_queue = request.session.get('request_queue',[])
    if len(request_queue) < MAX_REQUEST_PER_SECOND:
      request_queue.append(now)
      request.session['request_queue']=request_queue
    else:
      time0=request_queue[0]
      if (now-time0)<1:
        time.sleep(5)
      request_queue.append(time.time())
      request.session['request_queue']=request_queue[1:]

#啟用RequestBlocking中介軟體
IDDLEWARE = [
  'django.middleware.security.SecurityMiddleware','django.contrib.sessions.middleware.SessionMiddleware','django.middleware.common.CommonMiddleware','django.middleware.csrf.CsrfViewMiddleware','common.middleware.RequestBlockingMiddleware',#在sessions之後,auth之前
  'django.contrib.auth.middleware.AuthenticationMiddleware','django.contrib.messages.middleware.MessageMiddleware','django.middleware.clickjacking.XFrameOptionsMiddleware',]

以上這篇Django 對IP訪問頻率進行限制的例子就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。