Django中間件_ZenDei技術網路在線

中間件是面向切麵編程的好例子，它是一個可以介入Django的request和response處理過程的鉤子框架，一個輕量級、底層的“插件”系統，用於在全局修改Django的輸入或輸出。要使用中間件，首先要在settings中設置：上述是Django項目的預設設置，每一項字元串都代表一個中間件。中 ...

中間件是面向切麵編程的好例子，它是一個可以介入Django的request和response處理過程的鉤子框架，一個輕量級、底層的“插件”系統，用於在全局修改Django的輸入或輸出。

要使用中間件，首先要在settings中設置：

MIDDLEWARE = [
    'django.middleware.security.SecurityMiddleware',
    'django.contrib.sessions.middleware.SessionMiddleware',
    'django.middleware.common.CommonMiddleware',
    'django.middleware.csrf.CsrfViewMiddleware',
    'django.contrib.auth.middleware.AuthenticationMiddleware',
    'django.contrib.messages.middleware.MessageMiddleware',
    'django.middleware.clickjacking.XFrameOptionsMiddleware',
]

上述是Django項目的預設設置，每一項字元串都代表一個中間件。中間件就像洋蔥一樣，每個中間件都是一個層，在用戶請求階段，調用定義的view函數之前，請求以自上而下的順序通過所有的層，view函數處理之後，響應以自下而上的順序通過所有的層，期間經過的每個中間件都會對請求或者響應進行處理。

自定義中間件

要自定義中間件，寫一個類即可：

class SimpleMiddleware(object):
    def __init__(self, get_response):
        self.get_response = get_response
        # One-time configuration and initialization.

    def __call__(self, request):
        # Code to be executed for each request before
        # the view (and later middleware) are called.

        response = self.get_response(request)

        # Code to be executed for each request/response after
        # the view is called.

        return response

__init__初始化方法只在web服務啟動時調用一次，get_response參數是必需的，這個參數指的是下一個中間件或者view函數(如果是最後一個中間件)。

每個請求都會調用一次__call__，先對請求做處理，然後將請求發往下一個中間件(get_response)，一層層的迭代，就像棧一樣，返回時對響應做處理。

如果其中一個層的中間件決定短路並返迴響應而不調用其get_response，那麼該層下麵的層都不會看到請求或響應，只有該中間件和之上的中間件才會看到響應。

短路可以起到一些ip過濾之類的效果：

class IpFilterMiddleware(object):
    def __init__(self, get_response):
        self.get_response = get_response
        # One-time configuration and initialization.

    def __call__(self, request):
        if request.META.has_key('HTTP_X_FORWARDED_FOR'):
            ip = request.META['HTTP_X_FORWARDED_FOR']
        else:  
            ip = request.META['REMOTE_ADDR']
        if ip == '127.0.0.1':
            return HttpResponse('You are forbidden')
        response = self.get_response(request)
        return response

設置：

MIDDLEWARE = [
    'path.to.IpFilterMiddleware',

上述中間件把當地訪問攔截了，實際應用中可以為ip建立黑名單，然後檢查訪問ip是否在黑名單中。

我們也可以將中間件標記為未使用，只要在__init__中引發一個django.core.exceptions.MiddlewareNotUsed異常即可。

其它的中間件鉤子

除了前面描述的基本請求/響應中間件模式，您還可以向基於類的中間件添加三種其他特殊方法：

process_view(request, view_func, view_args, view_kwargs)

在調用view之前被調用。

它應該返回一個None 或一個HttpResponse對象。如果返回None，Django 將會繼續處理這個請求，執行其它的process_view() 中間件，然後調用對應的視圖。如果它返回一個HttpResponse對象，也會起到短路作用。

process_exception(request, exception)

當一個view引發異常時，Django會調用process_exception()來處理。返回一個None或一個HttpResponse對象。如果返回HttpResponse對象，會將響應交給處理響應的中間件處理。由於處理響應時是從下到上的，此層以上的process_exception()是不會被調用的。

process_template_response(request, response)

response參數應該是一個由view或者中間件返回的TemplateResponse對像（或等價的對象）。

如果響應的實例有render()方法，process_template_response()會在view剛好執行完畢之後被調用。這個方法必須返回一個實現了render方法的響應對象。

開發階段通常將DEBUG打開以調試錯誤，部署時會關閉DEBUG以免用戶看到錯誤信息。我們可以寫一個異常中間件，讓有許可權的管理員可以在DEBUG關閉的情況下看到錯誤信息，這樣就避免了切換的麻煩；

import sys

from django.views.debug import technical_500_response

class ExceptionMiddleware(object):
    def __init__(self, get_response):
        self.get_response = get_response
        # One-time configuration and initialization.

    def __call__(self, request):
        response = self.get_response(request)
        return response

    def process_exception(self, request, exception):
        if request.user.is_admin:
            return technical_500_response(request, *sys.exc_info())

置於中間件的最底層，這樣管理員可以看到用戶無法看到的錯誤詳細信息。

關於異常處理

Django會自動將視圖或中間件引發的異常轉換為具有錯誤狀態代碼的適當HTTP響應，這種轉換髮生在每個中間件之前和之後，我們是無需使用try捕捉異常的。

與舊版本中間件的相容

在Django 1.10版本之前，中間件設置名為MIDDLEWARE_CLASSES，是長這樣的：

class SessionMiddleware(object):
    def __init__(self):
        engine = import_module(settings.SESSION_ENGINE)
        self.SessionStore = engine.SessionStore

    def process_request(self, request):
        session_key = request.COOKIES.get(settings.SESSION_COOKIE_NAME)
        request.session = self.SessionStore(session_key)

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

Django提供django.utils.deprecation.MiddlewareMixin以簡化與MIDDLEWARE和舊的MIDDLEWARE_CLASSES相容的中間件類。 Django中包含的所有中間件類都相容這兩種設置。

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

新版繼承自MiddlewareMixin的SessionMiddleware：

class SessionMiddleware(MiddlewareMixin):
    def __init__(self, get_response=None):
        self.get_response = get_response
        engine = import_module(settings.SESSION_ENGINE)
        self.SessionStore = engine.SessionStore

    def process_request(self, request):
        session_key = request.COOKIES.get(settings.SESSION_COOKIE_NAME)
        request.session = self.SessionStore(session_key)

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

混合類提供一個__init__()方法，它接受一個可選的get_response參數，並將其存儲在self.get_response中。

__call__()方法：

調用self.process_request(request)（已定義的話）。
調用self.get_response(request)獲取後續中間件或者視圖的響應。
調用self.process_response（request， response）（已定義的話）。
返迴響應。

在大多數情況下，繼承這種混合類足以使舊式中間件與新系統相容，具有足夠的向後相容性。

上述的異常中間件也可以寫為：

from django.utils.deprecation import MiddlewareMixin

class ExceptionMiddleware(MiddlewareMixin):
    def process_exception(self, request, exception):
        if request.user.is_admin:
            return technical_500_response(request, *sys.exc_info())

使用MIDDLEWARE和MIDDLEWARE_CLASSES之間的行為差異：

在MIDDLEWARE_CLASSES下，即使上層中間件的process_request方法短路，下層的中間件也會調用它的process_response方法。在MIDDLEWARE下，如果中間件發生短路，則只有中間件和之上的中間件才會看到響應。
在MIDDLEWARE_CLASSES下，process_exception適用於從中間件process_request方法引發的異常。在MIDDLEWARE下，process_exception僅適用於view或者TemplateResponse的render方法引發的異常。從中間件引發的異常轉換為適當的HTTP響應，然後傳遞給下一個中間件。
在MIDDLEWARE_CLASSES下，如果process_response方法引發異常，則會跳過所有較早中間件的process_response方法然後返回500錯誤，而在MIDDLEWARE下，從中間件引發的異常將立即轉換為適當的HTTP響應，然後下一個中間件將會看到該響應。由於中間件引發異常，中間件不會被跳過。

處理流式響應

StreamingHttpResponse並沒有content屬性，使用時必須加以判斷：

if response.streaming:
    response.streaming_content = wrap_streaming_content(response.streaming_content)
else:
    response.content = alter_content(response.content)

用生成器處理過大的內容：

def wrap_streaming_content(content):
    for chunk in content:
        yield alter_content(chunk)