django中怎么自定義處理中間件

django中怎么自定義處理中間件，很多新手對此不是很清楚，為了幫助大家解決這個難題，下面小編將為大家詳細講解，有這方面需求的人可以來學習下，希望你能有所收獲。

Django基礎中間件

django.utils.deprecation.py

class MiddlewareMixin(object):
 def __init__(self, get_response=None):
  self.get_response = get_response
  super(MiddlewareMixin, self).__init__()

 def __call__(self, request):
  response = None
  if hasattr(self, 'process_request'):
   response = self.process_request(request)
  if not response:
   response = self.get_response(request)
  if hasattr(self, 'process_response'):
   response = self.process_response(request, response)
  return response

以上為Django基礎中間件源碼，要習慣于看源碼，上面的這段代碼并不復雜，下面我們來一一解釋。

def __init__(self, get_response=None):
 self.get_response = get_response
 super(MiddlewareMixin, self).__init__()

熟悉 python 類的都不陌生 __init__ 方法, 這里主要是 一次性配置和初始化

def __call__(self, request):
  response = None
  if hasattr(self, 'process_request'):
    response = self.process_request(request)
  if not response:
    response = self.get_response(request)
  if hasattr(self, 'process_response'):
    response = self.process_response(request, response)
  return response

__call__ 為每個請求/響應執行的代碼

self.process_request(request) 為每個請求到調用視圖之前的操作，通常可以在這里做一些用戶請求頻率的控制。

self.get_response(request) 為調用視圖

self.process_response(request, response) 為調用視圖完成后的操作

自定義中間件

剛才了解了基礎中間件，現在就開始編寫我們自己的中間件。

通常我們回去繼承基礎中間件來實現自己的功能

from django.utils.deprecation import MiddlewareMixin

class PermissionMiddlewareMixin(MiddlewareMixin):
  """
  django 中間件
  """

  def process_request(self, request):
    pass

  def process_response(self, request, response):
    return response

如果你要在請求之前做處理，需要定義 process_request() 方法，去實現相關功能

如果你要在視圖調用之后做處理，需要定義 process_response() 方法，去實現相關功能

:warning:注意 定義 process_response() 方法一定要 return response

需要將你編寫的中間件添加到 settings 中的 MIDDLEWARE 里

我這里寫了一個通過中間件限制客戶端請求頻率，有興趣的可以看一下

django中間件客戶端請求頻率限制

通過redis lua腳本對客戶端IP請求頻率限制

# coding:utf-8

__author__ = 'carey@akhack.com'

from django.utils.deprecation import MiddlewareMixin
from django.http.response import HttpResponse
from django_redis import get_redis_connection
from hashlib import md5


class RequestBlockMiddlewareMixin(MiddlewareMixin):
  """
  django中間件客戶端請求頻率限制
  """

  limit = 4 # 單位時間內允許請求次數
  expire = 1 # 限制時間
  cache = "default" # 獲取django cache

  def process_request(self, request):
    num = self.set_key(request)
    if num > self.limit:
      return HttpResponse("請求頻率過快，請稍后重試", status=503)

  @staticmethod
  def get_ident(request):
    """
    Identify the machine making the request by parsing HTTP_X_FORWARDED_FOR
    if present and number of proxies is > 0. If not use all of
    HTTP_X_FORWARDED_FOR if it is available, if not use REMOTE_ADDR.
    """
    NUM_PROXIES = 1
    xff = request.META.get('HTTP_X_FORWARDED_FOR')
    remote_addr = request.META.get('REMOTE_ADDR')
    num_proxies = NUM_PROXIES

    if num_proxies is not None:
      if num_proxies == 0 or xff is None:
        return remote_addr
      addrs = xff.split(',')
      client_addr = addrs[-min(num_proxies, len(addrs))]
      return client_addr.strip()

    return ''.join(xff.split()) if xff else remote_addr

  def get_md5(self, request):
    """
    獲取IP md5值
    :param request:
    :return:
    """
    ip_str = self.get_ident(request)
    ip_md5 = md5()
    ip_md5.update(ip_str.encode("utf-8"))
    return ip_md5.hexdigest()

  def set_key(self, request):
    """
    通過redis lua腳本設置請求請求次數和限制時間
    :param request:
    :return: 限制時間內請求次數
    """
    lua = """
      local current
      current = redis.call("incr",KEYS[1])
      if tonumber(current) == 1 then
        redis.call("expire",KEYS[1],ARGV[1])
      end
      return tonumber(redis.call("get", KEYS[1]))
      """
    key = self.get_md5(request)
    redis_cli = get_redis_connection(self.cache)
    data = redis_cli.eval(lua, 1, key, self.expire, self.limit)
    return data

看完上述內容是否對您有幫助呢？如果還想對相關知識有進一步的了解或閱讀更多相關文章，請關注億速云行業資訊頻道，感謝您對億速云的支持。