feat: guarantee extension order

shanbay · Jun 7, 2024 · b0406aa · b0406aa
1 parent 630673e
commit b0406aa
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/sea/app.py b/sea/app.py
@@ -137,7 +137,15 @@ def load_extensions_in_module(self, module):
         def is_ext(ins):
             return not inspect.isclass(ins) and hasattr(ins, "init_app")
 
-        for n, ext in inspect.getmembers(module, is_ext):
+        # guarantee the order of extension initialization by load_order attribute
+        def sort_ext(member):
+            _, ext = member
+            DEFAULT_LOWER_ORDER = 10
+            return getattr(ext, "load_order", DEFAULT_LOWER_ORDER)
+
+        for n, ext in sorted(
+            inspect.getmembers(module, is_ext), key=sort_ext
+        ):
             self._register_extension(n, ext)
         return self.extensions