sqlmesh 0.209.1.dev43__tar.gz → 0.209.1.dev45__tar.gz

{sqlmesh-0.209.1.dev43 → sqlmesh-0.209.1.dev45}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlmesh
-Version: 0.209.1.dev43
+Version: 0.209.1.dev45
 Summary: Next-generation data transformation framework
 Author-email: "TobikoData Inc." <engineering@tobikodata.com>
 License:                                  Apache License

{sqlmesh-0.209.1.dev43 → sqlmesh-0.209.1.dev45}/pyproject.toml

@@ -301,3 +301,4 @@ banned-module-level-imports = [
 "sqlmesh/lsp/**/*.py" = ["TID251"]
 "tests/lsp/**/*.py" = ["TID251"]
 "benchmarks/lsp*.py" = ["TID251"]
+"sqlmesh/dbt/builtin.py" = ["T100"]

{sqlmesh-0.209.1.dev43 → sqlmesh-0.209.1.dev45}/sqlmesh/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.209.1.dev43'
-__version_tuple__ = version_tuple = (0, 209, 1, 'dev43')
+__version__ = version = '0.209.1.dev45'
+__version_tuple__ = version_tuple = (0, 209, 1, 'dev45')

{sqlmesh-0.209.1.dev43 → sqlmesh-0.209.1.dev45}/sqlmesh/dbt/builtin.py

@@ -14,13 +14,14 @@ from dbt.adapters.base import BaseRelation, Column
 from ruamel.yaml import YAMLError
 from sqlglot import Dialect
 
+from sqlmesh.core.console import get_console
 from sqlmesh.core.engine_adapter import EngineAdapter
 from sqlmesh.core.snapshot.definition import DeployabilityIndex
 from sqlmesh.dbt.adapter import BaseAdapter, ParsetimeAdapter, RuntimeAdapter
 from sqlmesh.dbt.relation import Policy
 from sqlmesh.dbt.target import TARGET_TYPE_TO_CONFIG_CLASS
 from sqlmesh.dbt.util import DBT_VERSION
-from sqlmesh.utils import AttributeDict, yaml
+from sqlmesh.utils import AttributeDict, debug_mode_enabled, yaml
 from sqlmesh.utils.date import now
 from sqlmesh.utils.errors import ConfigError, MacroEvalError
 from sqlmesh.utils.jinja import JinjaMacroRegistry, MacroReference, MacroReturnVal
@@ -170,7 +171,13 @@ def env_var(name: str, default: t.Optional[str] = None) -> t.Optional[str]:
 
 
 def log(msg: str, info: bool = False) -> str:
-    logger.debug(msg)
+    if info:
+        # Write to both log file and stdout
+        logger.info(msg)
+        get_console().log_status_update(msg)
+    else:
+        logger.debug(msg)
+
     return ""
 
 
@@ -309,6 +316,15 @@ def _try_literal_eval(value: str) -> t.Any:
         return value
 
 
+def debug() -> str:
+    import sys
+    import ipdb  # type: ignore
+
+    frame = sys._getframe(3)
+    ipdb.set_trace(frame)
+    return ""
+
+
 BUILTIN_GLOBALS = {
     "dbt_version": version.__version__,
     "env_var": env_var,
@@ -329,6 +345,10 @@ BUILTIN_GLOBALS = {
     "zip_strict": lambda *args: list(zip(*args)),
 }
 
+# Add debug function conditionally both with dbt or sqlmesh equivalent flag
+if os.environ.get("DBT_MACRO_DEBUGGING") or debug_mode_enabled():
+    BUILTIN_GLOBALS["debug"] = debug
+
 BUILTIN_FILTERS = {
     "as_bool": as_bool,
     "as_native": _try_literal_eval,

{sqlmesh-0.209.1.dev43 → sqlmesh-0.209.1.dev45}/sqlmesh.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlmesh
-Version: 0.209.1.dev43
+Version: 0.209.1.dev45
 Summary: Next-generation data transformation framework
 Author-email: "TobikoData Inc." <engineering@tobikodata.com>
 License:                                  Apache License

{sqlmesh-0.209.1.dev43 → sqlmesh-0.209.1.dev45}/tests/dbt/test_transformation.py

@@ -854,13 +854,40 @@ def test_logging(sushi_test_project: Project, runtime_renderer: t.Callable):
     renderer = runtime_renderer(context, engine_adapter=engine_adapter)
 
     logger = logging.getLogger("sqlmesh.dbt.builtin")
-    with patch.object(logger, "debug") as mock_logger:
-        assert renderer('{{ log("foo") }}') == ""
-    assert "foo" in mock_logger.call_args[0][0]
 
-    with patch.object(logger, "debug") as mock_logger:
+    # Test log with info=False (default), should only log to file with debug and not to console
+    with (
+        patch.object(logger, "debug") as mock_debug,
+        patch.object(logger, "info") as mock_info,
+        patch.object(get_console(), "log_status_update") as mock_console,
+    ):
+        assert renderer('{{ log("foo") }}') == ""
+        mock_debug.assert_called_once()
+        assert "foo" in mock_debug.call_args[0][0]
+        mock_info.assert_not_called()
+        mock_console.assert_not_called()
+
+    # Test log with info=True, should log to info and also call log_status_update
+    with (
+        patch.object(logger, "debug") as mock_debug,
+        patch.object(logger, "info") as mock_info,
+        patch.object(get_console(), "log_status_update") as mock_console,
+    ):
+        assert renderer('{{ log("output to be logged with info", info=true) }}') == ""
+        mock_info.assert_called_once()
+        assert "output to be logged with info" in mock_info.call_args[0][0]
+        mock_debug.assert_not_called()
+        mock_console.assert_called_once()
+        assert "output to be logged with info" in mock_console.call_args[0][0]
+
+    # Test print function as well, should use debug
+    with (
+        patch.object(logger, "debug") as mock_logger,
+        patch.object(get_console(), "log_status_update") as mock_console,
+    ):
         assert renderer('{{ print("bar") }}') == ""
-    assert "bar" in mock_logger.call_args[0][0]
+        assert "bar" in mock_logger.call_args[0][0]
+        mock_console.assert_not_called()
 
 
 @pytest.mark.xdist_group("dbt_manifest")