fix settings management issue

ScholCommLab · Apr 7, 2021 · 3af5848 · 3af5848
1 parent 3d7a1ee
commit 3af5848
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/app/config.py b/app/config.py
@@ -1,4 +1,7 @@
 import os
+from typing import Type
+from typing import Union
+
 from pydantic import BaseSettings
 
 
@@ -141,7 +144,26 @@ def init_app(cls, app):
         DockerConfig.init_app(app)
 
 
-def get_config_class(config_name="default"):
+ConfigTypes: Type[
+    Union[
+        DevelopmentConfig,
+        TestingConfig,
+        ProductionConfig,
+        UnixConfig,
+        DockerConfig,
+        DockerComposeConfig,
+    ]
+] = Union[
+    DevelopmentConfig,
+    TestingConfig,
+    ProductionConfig,
+    UnixConfig,
+    DockerConfig,
+    DockerComposeConfig,
+]
+
+
+def get_config_class(config_name: str = "default") -> ConfigTypes:
     configs = {
         "development": DevelopmentConfig,
         "testing": TestingConfig,