feat: support middlewares from env

shanbay · Dec 13, 2024 · 913f98a · 913f98a
1 parent a85eff1
commit 913f98a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/sea/app.py b/sea/app.py
@@ -127,7 +127,8 @@ def _register_extension(self, name, ext):
         self._extensions[name] = ext
 
     def load_middlewares(self):
-        mids = ["sea.middleware.GuardMiddleware"] + self.config.get("MIDDLEWARES")
+        env_mids = os.environ.get("SEA_MIDDLEWARES", "").split(",")
+        mids = ["sea.middleware.GuardMiddleware"] + [mid for mid in env_mids if mid] + self.config.get("MIDDLEWARES")
         for mn in mids:
             m = utils.import_string(mn)
             self._middlewares.insert(0, m)

diff --git a/tests/test_app.py b/tests/test_app.py
@@ -29,9 +29,11 @@ def test_baseapp(caplog):
     assert _app.config["PORT"] == 4001
     assert _app.debug
     assert _app.testing
+    os.environ["SEA_MIDDLEWARES"] = "sea.middleware.BaseMiddleware,"
     _app.load_middlewares()
 
-    assert len(_app.middlewares) == 3
+    assert len(_app.middlewares) == 5
+    print(_app.middlewares)
 
     with mock.patch('sea._app', new=_app):
         from app import extensions, servicers