feat: expose classmethod to build tortoise config

Author: waketzheng

tortoise/config.py

@@ -2,10 +2,14 @@
 
 from collections.abc import Mapping
 from dataclasses import dataclass, field
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from tortoise.exceptions import ConfigurationError
 
+if TYPE_CHECKING:
+    from collections.abc import Iterable
+    from types import ModuleType
+
 
 @dataclass(frozen=True)
 class DBUrlConfig:
@@ -202,3 +206,36 @@ def from_dict(cls, data: Mapping[str, Any]) -> TortoiseConfig:
             use_tz=data.get("use_tz"),
             timezone=data.get("timezone"),
         )
+
+    @classmethod
+    def merge_args(
+        cls,
+        config: dict[str, Any] | TortoiseConfig | None = None,
+        config_file: str | None = None,
+        db_url: str | None = None,
+        modules: dict[str, Iterable[str | ModuleType]] | None = None,
+    ) -> TortoiseConfig:
+        # Handle config_file: load it as config dict
+        if config_file is not None:
+            from tortoise import Tortoise
+
+            if config is not None:
+                raise ConfigurationError("Cannot specify both 'config' and 'config_file'")
+            config = Tortoise._get_config_from_config_file(config_file)
+
+        # Convert input to TortoiseConfig for typed access
+        typed_config: TortoiseConfig
+        if config is None:
+            from tortoise.backends.base.config_generator import generate_config
+
+            if db_url is None or modules is None:
+                raise ConfigurationError(
+                    "Must provide either 'config', 'config_file', or both 'db_url' and 'modules'"
+                )
+            config_dict = generate_config(db_url, app_modules=modules)
+            typed_config = cls.from_dict(config_dict)
+        elif isinstance(config, dict):
+            typed_config = cls.from_dict(config)
+        else:
+            typed_config = config
+        return typed_config

tortoise/context.py

@@ -303,26 +303,7 @@ async def init(
         """
         from tortoise.apps import Apps
 
-        # Handle config_file: load it as config dict
-        if config_file is not None:
-            if config is not None:
-                raise ConfigurationError("Cannot specify both 'config' and 'config_file'")
-            config = self._get_config_from_config_file(config_file)
-
-        # Convert input to TortoiseConfig for typed access
-        typed_config: TortoiseConfig
-        if config is None:
-            if db_url is None or modules is None:
-                raise ConfigurationError(
-                    "Must provide either 'config', 'config_file', or both 'db_url' and 'modules'"
-                )
-            config_dict = generate_config(db_url, app_modules=modules)
-            typed_config = TortoiseConfig.from_dict(config_dict)
-        elif isinstance(config, TortoiseConfig):
-            typed_config = config
-        else:
-            typed_config = TortoiseConfig.from_dict(config)
-
+        typed_config = TortoiseConfig.merge_args(config, config_file, db_url, modules)
         config_dict = typed_config.to_dict()
         connections_config = config_dict["connections"]
         apps_config = config_dict["apps"]