adding middleware setup

jstacoder · jstacoder · commit d35f84c78845 · 2018-06-25T18:05:08.000-07:00
diff --git a/flask_xxl/main.py b/flask_xxl/main.py
@@ -61,9 +61,18 @@ def get_app(self, app_module_name, **kwargs):
         self._register_context_processors()
         self._register_template_filters()
         self._register_template_extensions()
+        self._register_middlewares()
 
         return self.app
 
+    def _register_middlewares(self):
+        for middleware_path in self.app.config.get('MIDDLEWARE', []):
+            module_name, middleware_name = middleware_path.rsplit('.', 1)
+            module_name, _ = self._get_imported_stuff_by_path(middleware_path)
+            middleware = getattr(module_name, middleware_name, None)
+            if middleware is not None:
+                self.app.wsgi_app = middleware(self.app.wsgi_app)
+
     def _set_path(self):
         sys.path.append(self.app.config.get('ROOT_PATH',''))
 
diff --git a/setup.py b/setup.py
@@ -1,4 +1,4 @@
-VERSION = '0,10,0'
+VERSION = '0,10,1'
 import os
 from setuptools import setup, find_packages,findall
 from glob import glob
@@ -26,7 +26,7 @@ def get_pkg_data():
     return pkg_data
 
 config = dict(
-        name='flask-xxl',
+        name='flask-xl',
         version=get_version(),#'0.0.9',
         include_package_data=True,
         author='Kyle Roux',
