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看完上述内容是否对您有帮助呢？如果还想对相关知识有进一步的了解或阅读更多相关文章，请关注亿速云行业资讯频道，感谢您对亿速云的支持。
亿速云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
