Chore: In GenericContext make the config type a class member (#2216)

Author: izeigerman

sqlmesh/cli/main.py

@@ -11,7 +11,7 @@
 from sqlmesh.cli import error_handler
 from sqlmesh.cli import options as opt
 from sqlmesh.cli.example_project import ProjectTemplate, init_example_project
-from sqlmesh.core.config import Config, load_configs
+from sqlmesh.core.config import load_configs
 from sqlmesh.core.context import Context
 from sqlmesh.utils.date import TimeLike
 from sqlmesh.utils.errors import MissingDependencyError
@@ -77,7 +77,7 @@ def cli(
         elif ctx.invoked_subcommand in ("create_external_models", "migrate", "rollback", "ui"):
             load = False
 
-    configs = load_configs(config, Config, paths)
+    configs = load_configs(config, Context.CONFIG_TYPE, paths)
     log_limit = list(configs.values())[0].log_limit
     configure_logging(debug, ignore_warnings, log_to_stdout, log_limit=log_limit)
 

sqlmesh/core/context.py

@@ -254,9 +254,10 @@ class GenericContext(BaseContext, t.Generic[C]):
         config_type: The type of config object to use (default Config).
     """
 
+    CONFIG_TYPE: t.Type[C]
+
     def __init__(
         self,
-        config_type: t.Type[C],
         engine_adapter: t.Optional[EngineAdapter] = None,
         notification_targets: t.Optional[t.List[NotificationTarget]] = None,
         state_sync: t.Optional[StateSync] = None,
@@ -271,7 +272,7 @@ def __init__(
     ):
         self.console = console or get_console()
         self.configs = (
-            config if isinstance(config, dict) else load_configs(config, config_type, paths)
+            config if isinstance(config, dict) else load_configs(config, self.CONFIG_TYPE, paths)
         )
         self.dag: DAG[str] = DAG()
         self._models: UniqueKeyDict[str, Model] = UniqueKeyDict("models")
@@ -1762,5 +1763,4 @@ def _register_notification_targets(self) -> None:
 
 
 class Context(GenericContext[Config]):
-    def __init__(self, *args: t.Any, **kwargs: t.Any):
-        super().__init__(Config, *args, **kwargs)
+    CONFIG_TYPE = Config

sqlmesh/magics.py

@@ -19,7 +19,7 @@
 
 from sqlmesh.cli.example_project import ProjectTemplate, init_example_project
 from sqlmesh.core import constants as c
-from sqlmesh.core.config import Config, load_configs
+from sqlmesh.core.config import load_configs
 from sqlmesh.core.console import get_console
 from sqlmesh.core.context import Context
 from sqlmesh.core.dialect import format_model_expressions, parse
@@ -97,7 +97,7 @@ def context(self, line: str) -> None:
         from sqlmesh import configure_logging
 
         args = parse_argstring(self.context, line)
-        configs = load_configs(args.config, Config, args.paths)
+        configs = load_configs(args.config, Context.CONFIG_TYPE, args.paths)
         log_limit = list(configs.values())[0].log_limit
         configure_logging(args.debug, args.ignore_warnings, log_limit=log_limit)
         try: