Extract register_middleware into a method.

This commit is contained in:
Johnny Rocketfingers 2017-05-18 06:34:28 +02:00
parent fa1b7de52a
commit bf5438d573
2 changed files with 13 additions and 11 deletions

View File

@ -272,24 +272,25 @@ class Sanic:
return response return response
# Decorator def register_middleware(self, middleware, attach_to='request'):
def middleware(self, middleware_or_request):
"""Decorate and register middleware to be called before a request.
Can either be called as @app.middleware or @app.middleware('request')
"""
def register_middleware(middleware, attach_to='request'):
if attach_to == 'request': if attach_to == 'request':
self.request_middleware.append(middleware) self.request_middleware.append(middleware)
if attach_to == 'response': if attach_to == 'response':
self.response_middleware.appendleft(middleware) self.response_middleware.appendleft(middleware)
return middleware return middleware
# Decorator
def middleware(self, middleware_or_request):
"""Decorate and register middleware to be called before a request.
Can either be called as @app.middleware or @app.middleware('request')
"""
# Detect which way this was called, @middleware or @middleware('AT') # Detect which way this was called, @middleware or @middleware('AT')
if callable(middleware_or_request): if callable(middleware_or_request):
return register_middleware(middleware_or_request) return self.register_middleware(middleware_or_request)
else: else:
return partial(register_middleware, return partial(Sanic.register_middleware, self,
attach_to=middleware_or_request) attach_to=middleware_or_request)
# Static Files # Static Files

View File

@ -65,10 +65,11 @@ class Blueprint:
# Middleware # Middleware
for future in self.middlewares: for future in self.middlewares:
if future.args or future.kwargs: if future.args or future.kwargs:
app.middleware(*future.args, app.register_middleware(future.middleware,
**future.kwargs)(future.middleware) *future.args,
**future.kwargs)
else: else:
app.middleware(future.middleware) app.register_middleware(future.middleware)
# Exceptions # Exceptions
for future in self.exceptions: for future in self.exceptions: