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

目前成都創(chuàng)新互聯(lián)公司已為千余家的企業(yè)提供了網(wǎng)站建設(shè)、域名、網(wǎng)頁空間、網(wǎng)站改版維護(hù)、企業(yè)網(wǎng)站設(shè)計(jì)、當(dāng)雄網(wǎng)站維護(hù)等服務(wù)，公司將堅(jiān)持客戶導(dǎo)向、應(yīng)用為本的策略，正道將秉承"和諧、參與、激情"的文化，與客戶和合作伙伴齊心協(xié)力一起成長，共同發(fā)展。

Django基礎(chǔ)中間件

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基礎(chǔ)中間件源碼，要習(xí)慣于看源碼，上面的這段代碼并不復(fù)雜，下面我們來一一解釋。

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__ 為每個(gè)請(qǐng)求/響應(yīng)執(zhí)行的代碼

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

self.get_response(request) 為調(diào)用視圖

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

自定義中間件

剛才了解了基礎(chǔ)中間件，現(xiàn)在就開始編寫我們自己的中間件。

通常我們回去繼承基礎(chǔ)中間件來實(shí)現(xiàn)自己的功能

from django.utils.deprecation import MiddlewareMixin

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

  def process_request(self, request):
    pass

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

如果你要在請(qǐng)求之前做處理，需要定義 process_request() 方法，去實(shí)現(xiàn)相關(guān)功能

如果你要在視圖調(diào)用之后做處理，需要定義 process_response() 方法，去實(shí)現(xiàn)相關(guān)功能

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

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

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

django中間件客戶端請(qǐng)求頻率限制

通過redis lua腳本對(duì)客戶端IP請(qǐng)求頻率限制

# 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中間件客戶端請(qǐng)求頻率限制
  """

  limit = 4 # 單位時(shí)間內(nèi)允許請(qǐng)求次數(shù)
  expire = 1 # 限制時(shí)間
  cache = "default" # 獲取django cache

  def process_request(self, request):
    num = self.set_key(request)
    if num > self.limit:
      return HttpResponse("請(qǐng)求頻率過快，請(qǐng)稍后重試", 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腳本設(shè)置請(qǐng)求請(qǐng)求次數(shù)和限制時(shí)間
    :param request:
    :return: 限制時(shí)間內(nèi)請(qǐng)求次數(shù)
    """
    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

看完上述內(nèi)容是否對(duì)您有幫助呢？如果還想對(duì)相關(guān)知識(shí)有進(jìn)一步的了解或閱讀更多相關(guān)文章，請(qǐng)關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道，感謝您對(duì)創(chuàng)新互聯(lián)的支持。

網(wǎng)頁名稱：django中怎么自定義處理中間件-創(chuàng)新互聯(lián)
標(biāo)題來源：http://weahome.cn/article/ddoiph.html