Merge pull request #424 from growingdever/blueprint-support-view

support view instance for blueprint add_route method
This commit is contained in:
Eli Uriegas 2017-02-14 10:26:05 -06:00 committed by GitHub
commit b139810b6a
2 changed files with 21 additions and 6 deletions

View File

@ -131,8 +131,8 @@ can be used to implement our API versioning scheme.
from sanic.response import text
from sanic import Blueprint
blueprint_v1 = Blueprint('v1')
blueprint_v2 = Blueprint('v2')
blueprint_v1 = Blueprint('v1', url_prefix='/v1')
blueprint_v2 = Blueprint('v2', url_prefix='/v2')
@blueprint_v1.route('/')
async def api_v1_root(request):

View File

@ -1,5 +1,7 @@
from collections import defaultdict, namedtuple
from sanic.constants import HTTP_METHODS
from sanic.views import CompositionView
FutureRoute = namedtuple('Route', ['handler', 'uri', 'methods', 'host'])
FutureListener = namedtuple('Listener', ['handler', 'uri', 'methods', 'host'])
@ -82,15 +84,28 @@ class Blueprint:
return handler
return decorator
def add_route(self, handler, uri, methods=None, host=None):
def add_route(self, handler, uri, methods=frozenset({'GET'}), host=None):
"""
Creates a blueprint route from a function.
:param handler: Function to handle uri request.
:param handler: Function for handling uri requests. Accepts function,
or class instance with a view_class method.
:param uri: Endpoint at which the route will be accessible.
:param methods: List of acceptable HTTP methods.
:return: function or class instance
"""
route = FutureRoute(handler, uri, methods, host)
self.routes.append(route)
# Handle HTTPMethodView differently
if hasattr(handler, 'view_class'):
methods = set()
for method in HTTP_METHODS:
if getattr(handler.view_class, method.lower(), None):
methods.add(method)
# handle composition view differently
if isinstance(handler, CompositionView):
methods = handler.handlers.keys()
self.route(uri=uri, methods=methods, host=host)(handler)
return handler
def listener(self, event):