Test of failing functionality

flask_injector/__init__.py

@@ -53,6 +53,20 @@ def wrapper(*args: Any, **kwargs: Any) -> Any:
     return wrapper  # type: ignore
 
 
+def _resolve_wraped_function(func: Callable) -> Callable:
+    """
+    Decorators that use functools wraps will not have
+    a func.__code__ object that matches what View.as_view
+    creates breaking code.
+
+    This gets the original undecorated function.
+    """
+    if hasattr(func, '__wrapped__'):
+        return _resolve_wraped_function(func.__wrapped__)
+    else:
+        return func
+
+
 def wrap_fun(fun: T, injector: Injector) -> T:
     if isinstance(fun, LocalProxy):
         return fun  # type: ignore
@@ -101,6 +115,7 @@ def wrap_class_based_view(fun: Callable, injector: Injector) -> Callable:
     cls = cast(Any, fun).view_class
     name = fun.__name__
 
+    fun = _resolve_wraped_function(fun)
     closure_contents = (c.cell_contents for c in cast(Any, fun).__closure__)
     fun_closure = dict(zip(fun.__code__.co_freevars, closure_contents))
     try:

flask_injector/tests.py

@@ -1,7 +1,7 @@
 import gc
 import json
 import warnings
-from functools import partial
+from functools import partial, wraps
 from typing import NewType
 
 import flask_restful
@@ -284,6 +284,36 @@ def dispatch_request(self, dispatch_arg):
     eq_(response.data, b'aaa bbb')
 
 
+def test_view_decoration_works():
+
+    def decorator(view):
+        @wraps(view)
+        def wrapper(*args, **kwargs):
+            return view(*args, **kwargs)
+
+        return wrapper
+
+    class MyView(View):
+        decorators = [decorator]
+
+        def __init__(self, class_arg):
+            self.class_arg = class_arg
+
+        def dispatch_request(self, dispatch_arg):
+            return '%s %s' % (self.class_arg, dispatch_arg)
+
+
+    app = Flask(__name__)
+    app.add_url_rule('/<dispatch_arg>', view_func=MyView.as_view('view', class_arg='aaa'))
+
+    FlaskInjector(app=app)
+
+    client = app.test_client()
+    response = client.get('/bbb')
+    print(response.data)
+    eq_(response.data, b'aaa bbb')
+
+
 def test_flask_restful_integration_works():
     class HelloWorld(flask_restful.Resource):
         @inject