diff --git a/app/config.py b/app/config.py index 9890c28..0966459 100755 --- 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,