Added blueprint exception and middleware support, and tests

sanic/blueprints.py

@@ -1,6 +1,4 @@
-
-
-class BlueprintSetup():
+class BlueprintSetup:
     """
     """
 
@@ -17,26 +15,41 @@ class BlueprintSetup():
         #: blueprint.
         self.url_prefix = url_prefix
 
-    def add_url_rule(self, uri, methods=None, handler=None, **options):
-        """A helper method to register a handler to the application url routes.
-
+    def add_route(self, handler, uri, methods):
+        """
+        A helper method to register a handler to the application url routes.
         """
         if self.url_prefix:
             uri = self.url_prefix + uri
 
         self.app.router.add(uri, methods, handler)
 
+    def add_exception(self, handler, *args, **kwargs):
+        """
+        Registers exceptions to sanic
+        """
+        self.app.exception(*args, **kwargs)(handler)
 
-class Blueprint():
+    def add_middleware(self, middleware, *args, **kwargs):
+        """
+        Registers middleware to sanic
+        """
+        if args or kwargs:
+            self.app.middleware(*args, **kwargs)(middleware)
+        else:
+            self.app.middleware(middleware)
+
+
+class Blueprint:
     def __init__(self, name, url_prefix=None):
         self.name = name
         self.url_prefix = url_prefix
         self.deferred_functions = []
 
     def record(self, func):
-        """Registers a callback function that is invoked when the blueprint is
+        """
+        Registers a callback function that is invoked when the blueprint is
         registered on the application.
-
         """
         self.deferred_functions.append(func)
 
@@ -57,12 +70,30 @@ class Blueprint():
         """
         """
         def decorator(handler):
-            self.add_url_rule(uri=uri, methods=methods, handler=handler)
+            self.record(lambda s: s.add_route(handler, uri, methods))
             return handler
         return decorator
 
-    def add_url_rule(self, uri, methods=None, handler=None):
+    def middleware(self, *args, **kwargs):
         """
         """
-        self.record(lambda s:
-            s.add_url_rule(uri, methods, handler))
+
+        def register_middleware(middleware):
+            self.record(lambda s: s.add_middleware(middleware, *args, **kwargs))
+            return middleware
+
+        # Detect which way this was called, @middleware or @middleware('AT')
+        if len(args) == 1 and len(kwargs) == 0 and callable(args[0]):
+            args = []
+            return register_middleware(args[0])
+        else:
+            return register_middleware
+
+    def exception(self, *args, **kwargs):
+        """
+        """
+        def decorator(handler):
+            self.record(lambda s: s.add_exception(handler, *args, **kwargs))
+            return handler
+        return decorator
+