feat: guarantee extension order

shanbay · Jun 7, 2024 · 737a969 · 737a969
1 parent 630673e
commit 737a969
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/sea/app.py b/sea/app.py
@@ -137,7 +137,19 @@ 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
+            ORDER_ATTR = "load_order"
+            DEFAULT_LOWER_ORDER = 10
+
+            if hasattr(ext, ORDER_ATTR):
+                return getattr(ext, ORDER_ATTR)
+            return DEFAULT_LOWER_ORDER
+
+        for n, ext in sorted(
+            inspect.getmembers(module, is_ext), key=sort_ext
+        ):
             self._register_extension(n, ext)
         return self.extensions