feat: context support pathlib.Path paths (#1942)

Author: eakmanrq

sqlmesh/core/context.py

@@ -256,7 +256,7 @@ def __init__(
         engine_adapter: t.Optional[EngineAdapter] = None,
         notification_targets: t.Optional[t.List[NotificationTarget]] = None,
         state_sync: t.Optional[StateSync] = None,
-        paths: t.Union[str, t.Iterable[str]] = "",
+        paths: t.Union[str | Path, t.Iterable[str | Path]] = "",
         config: t.Optional[t.Union[Config, str, t.Dict[Path, Config]]] = None,
         gateway: t.Optional[str] = None,
         concurrent_tasks: t.Optional[int] = None,

tests/conftest.py

@@ -243,7 +243,7 @@ def init_and_plan_context(copy_to_temp_path, mocker) -> t.Callable:
     def _make_function(
         paths: str | t.List[str] | Path | t.List[Path], config="test_config"
     ) -> t.Tuple[Context, Plan]:
-        sushi_context = Context(paths=[str(x) for x in copy_to_temp_path(paths)], config=config)
+        sushi_context = Context(paths=copy_to_temp_path(paths), config=config)
         confirm = mocker.patch("sqlmesh.core.console.Confirm")
         confirm.ask.return_value = False
 

tests/core/test_context.py

@@ -285,7 +285,7 @@ def test():
     config = Config(
         ignore_patterns=["models/ignore/*.sql", "macro_ignore.py", ".ipynb_checkpoints/*"]
     )
-    context = Context(paths=str(tmp_path), config=config)
+    context = Context(paths=tmp_path, config=config)
 
     assert ['"memory"."db"."actual_test"'] == list(context.models)
     assert "test" in context._macros

tests/core/test_format.py

@@ -22,7 +22,7 @@ def test_format_files(tmp_path: pathlib.Path):
     )
 
     config = Config()
-    context = Context(paths=str(tmp_path), config=config)
+    context = Context(paths=tmp_path, config=config)
     context.load()
     assert isinstance(context.get_model("this.model"), SqlModel)
     assert isinstance(context.get_model("other.model"), SqlModel)

tests/core/test_integration.py

@@ -1743,7 +1743,7 @@ def get_current_df(context: Context):
         str(tmp_path / "models" / "marketing.sql"),
     )
 
-    context = Context(paths=[str(tmp_path)], config="test_config")
+    context = Context(paths=tmp_path, config="test_config")
     context.engine_adapter.create_schema("sushi")
     replace_source_table(
         context,

tests/core/test_test.py

@@ -451,7 +451,7 @@ def test_test_generation(tmp_path: Path) -> None:
         model_defaults=ModelDefaultsConfig(dialect="duckdb"),
     )
 
-    context = Context(paths=[str(tmp_path)], config=config)
+    context = Context(paths=tmp_path, config=config)
     context.plan(auto_apply=True)
 
     input_queries = {

tests/web/test_main.py

@@ -39,7 +39,7 @@ def get_settings_override() -> Settings:
 
 @pytest.fixture
 def project_context(project_tmp_path: Path) -> Context:
-    context = Context(paths=str(project_tmp_path), console=api_console)
+    context = Context(paths=project_tmp_path, console=api_console)
 
     def get_loaded_context_override() -> Context:
         return context