feat(vscode): handling python errors well

[ci skip]

Author: benfdking

sqlmesh/core/config/loader.py

@@ -122,6 +122,9 @@ def load_config_from_paths(
                     validation_error_message(e, f"Invalid project config '{config_name}':")
                     + "\n\nVerify your config.py."
                 )
+            except ConfigError:
+                # Re-raise ConfigError as is (it already has the proper location and message)
+                raise
         else:
             raise ConfigError(
                 f"Unsupported config file extension '{extension}' in config file '{path}'."
@@ -184,8 +187,19 @@ def load_config_from_python_module(
     module_path: Path,
     config_name: str = "config",
 ) -> C:
-    with sys_path(module_path.parent):
-        config_module = import_python_file(module_path, module_path.parent)
+    try:
+        with sys_path(module_path.parent):
+            config_module = import_python_file(module_path, module_path.parent)
+    except SyntaxError as e:
+        raise ConfigError(
+            f"Python syntax error in config file: {e.msg} ({e.filename}, line {e.lineno})",
+            location=module_path,
+        )
+    except Exception as e:
+        raise ConfigError(
+            f"Failed to load config file: {str(e)}",
+            location=module_path,
+        )
 
     try:
         config_obj = getattr(config_module, config_name)

vscode/extension/tests/broken_project.spec.ts

@@ -400,4 +400,43 @@ test.describe('Bad config.py/config.yaml file issues', () => {
       .first()
       .isVisible({ timeout: 5_000 })
   })
+
+  test('sushi example, bad config.py', async ({ page, sharedCodeServer }) => {
+    const tempDir = await fs.mkdtemp(
+      path.join(os.tmpdir(), 'vscode-test-tcloud-'),
+    )
+    await fs.copy(SUSHI_SOURCE_PATH, tempDir)
+    await createPythonInterpreterSettingsSpecifier(tempDir)
+
+    const configPyPath = path.join(tempDir, 'config.py')
+    // Write an invalid Python to config.py
+    await fs.writeFile(configPyPath, 'invalid_python_code = [1, 2, 3')
+
+    await page.goto(
+      `http://127.0.1:${sharedCodeServer.codeServerPort}/?folder=${tempDir}`,
+    )
+    await page.waitForLoadState('networkidle')
+
+    // Open customers.sql model
+    await page
+      .getByRole('treeitem', { name: 'models', exact: true })
+      .locator('a')
+      .click()
+    await page
+      .getByRole('treeitem', { name: 'customers.sql', exact: true })
+      .locator('a')
+      .click()
+
+    // Expect the error to appear
+    await page.waitForSelector('text=Error creating context')
+
+    // Open the problems view
+    await runCommand(page, 'View: Focus Problems')
+
+    // Assert that the error is present in the problems view
+    await page
+      .getByText('Python syntax error in config file:', { exact: true })
+      .first()
+      .isVisible({ timeout: 5_000 })
+  })
 })