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
看完上述内容是否对您有帮助呢?如果还想对相关知识有进一步的了解或阅读更多相关文章,请关注创新互联行业资讯频道,感谢您对创新互联的支持。
网页标题:django中怎么自定义处理中间件-创新互联
当前网址:http://pwwzsj.com/article/ddoiph.html