loom-kernel 0.12.1__tar.gz → 0.12.2__tar.gz

{loom_kernel-0.12.1 → loom_kernel-0.12.2}/CHANGELOG.md

@@ -1,3 +1,20 @@
+# 🚀 Release 0.12.2 ([#68](https://github.com/the-reacher-data/loom-py/pull/68)) ([`4f067d0`](https://github.com/the-reacher-data/loom-py/commit/4f067d060347334128455503d6a4975c1bcc6285))
+
+
+
+## 🐛 Fixes
+### prefect
+- **prefect:** observe maintenance logs in flow runs
+
+
+
+
+
+
+
+
+
+
 # 🚀 Release 0.12.1 ([#66](https://github.com/the-reacher-data/loom-py/pull/66)) ([`f3c045e`](https://github.com/the-reacher-data/loom-py/commit/f3c045ec6b9e915e5d2c9f55d7134a7be37c1475))
 
 

loom_kernel-0.12.2/CHANGELOG_RELEASE.md

@@ -0,0 +1,15 @@
+# 🚀 Release 0.12.2 ([#68](https://github.com/the-reacher-data/loom-py/pull/68)) ([`4f067d0`](https://github.com/the-reacher-data/loom-py/commit/4f067d060347334128455503d6a4975c1bcc6285))
+
+
+
+## 🐛 Fixes
+### prefect
+- **prefect:** observe maintenance logs in flow runs
+
+
+
+
+
+
+
+

{loom_kernel-0.12.1 → loom_kernel-0.12.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: loom-kernel
-Version: 0.12.1
+Version: 0.12.2
 Summary: Loom Python project
 License-File: LICENSE
 Classifier: Typing :: Typed

{loom_kernel-0.12.1 → loom_kernel-0.12.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "loom-kernel"
-version = "0.12.1"
+version = "0.12.2"
 description = "Loom Python project"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -268,7 +268,7 @@ markers = [
 
 [tool.commitizen]
 name = "cz_conventional_commits"
-version = "0.12.1"
+version = "0.12.2"
 tag_format = "v$version"
 version_files = [
     "pyproject.toml:project.version",

{loom_kernel-0.12.1 → loom_kernel-0.12.2}/src/loom/prefect/flow/_body.py

@@ -16,6 +16,7 @@ from loom.etl.runner import ETLRunner
 from loom.prefect._ctx import FlowCtx
 from loom.prefect._placeholders import resolve_placeholder
 from loom.prefect._summary import set_run_summary
+from loom.prefect.flow._common import prefect_flow_run_id
 from loom.prefect.flow._run_name import compute_correlation_id
 from loom.prefect.flow._signature import normalize_datetime_fields
 from loom.prefect.manifest import (
@@ -89,7 +90,7 @@ def build_flow_body(
             _maybe_delete_manifest(manifest_store, ctx.correlation_id)
             return
 
-        flow_run_id = _current_flow_run_id()
+        flow_run_id = prefect_flow_run_id()
         install_log_bridge(flow_run_id)
         try:
             observers = _build_observers(flow_run_id, manifest_store, manifest)
@@ -158,15 +159,6 @@ def _maybe_build_prometheus_adapter() -> Any:
     return PrometheusLifecycleAdapter(pushgateway_url=pushgateway)
 
 
-def _current_flow_run_id() -> Any:
-    try:
-        from prefect.runtime import flow_run as _fr  # noqa: PLC0415
-
-        return _fr.id
-    except (ImportError, AttributeError):
-        return None
-
-
 def _invoke_runner(
     config_path: str,
     pipeline: type[ETLPipeline[Any]],

{loom_kernel-0.12.1 → loom_kernel-0.12.2}/src/loom/prefect/flow/_common.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import Any
+import uuid
+from typing import Any, cast
 
 
 def coerce_tags(raw: Any) -> tuple[str, ...]:
@@ -17,4 +18,18 @@ def coerce_tags(raw: Any) -> tuple[str, ...]:
     return tuple(raw)
 
 
-__all__ = ["coerce_tags"]
+def prefect_flow_run_id() -> uuid.UUID | None:
+    """Return the current Prefect flow run identifier.
+
+    Returns:
+        The active flow run UUID, or ``None`` outside a Prefect flow run.
+    """
+    try:
+        from prefect.runtime import flow_run  # noqa: PLC0415
+
+        return cast(uuid.UUID | None, flow_run.id)
+    except (ImportError, AttributeError):
+        return None
+
+
+__all__ = ["coerce_tags", "prefect_flow_run_id"]

{loom_kernel-0.12.1 → loom_kernel-0.12.2}/src/loom/prefect/flow/_maintenance.py

@@ -19,10 +19,12 @@ from loom.prefect._summary import set_run_summary
 from loom.prefect.deploy._schedule import extract_pool_config
 from loom.prefect.deploy._yaml import read_yaml
 from loom.prefect.flow._common import coerce_tags as _coerce_tags
+from loom.prefect.flow._common import prefect_flow_run_id
 from loom.prefect.flow._hooks import make_notification_hooks, pause_schedule_on_failure
 from loom.prefect.flow._run_name import make_run_name_callback
 from loom.prefect.flow._signature import normalize_datetime_fields, signature_from_params_type
 from loom.prefect.notify import build_notifiers
+from loom.prefect.observer._logging_bridge import install_log_bridge, uninstall_log_bridge
 
 
 def maintenance_flow(
@@ -76,8 +78,13 @@ def maintenance_flow(
         resolved = normalize_datetime_fields(resolved, params_type)
         params = msgspec.convert(resolved, type=params_type)
         actual_path = os.environ.get("LOOM_STORAGE_CONFIG_PATH") or storage_config_path
-        storage_config, _ = _load_yaml(actual_path)
-        report = MaintenanceRunner.from_config(storage_config).run(step, params=params)
+        # Bridge loom logs into the Prefect run (mirrors etl_flow).
+        install_log_bridge(prefect_flow_run_id())
+        try:
+            storage_config, _ = _load_yaml(actual_path)
+            report = MaintenanceRunner.from_config(storage_config).run(step, params=params)
+        finally:
+            uninstall_log_bridge()
         set_run_summary(_maintenance_summary(report, resolved))
         report.raise_if_errors()
 

loom_kernel-0.12.2/tests/unit/prefect/test_maintenance_flow.py

@@ -0,0 +1,84 @@
+"""Tests for Prefect maintenance-flow observability."""
+
+from __future__ import annotations
+
+import uuid
+from pathlib import Path
+from unittest.mock import MagicMock
+
+import msgspec
+import pytest
+
+from loom.etl.maintenance._runner import MaintenanceReport
+from loom.etl.maintenance._step import MaintenanceStep
+from loom.prefect.flow import _maintenance
+
+
+class _MaintenanceParams(msgspec.Struct, frozen=True):
+    dry_run: bool = False
+
+
+class _MaintenanceStep(MaintenanceStep[_MaintenanceParams]):
+    operations = []
+
+
+def _write_config(tmp_path: Path) -> Path:
+    config_path = tmp_path / "maintenance.yaml"
+    config_path.write_text("params: {}\n", encoding="utf-8")
+    return config_path
+
+
+def _build_flow(tmp_path: Path) -> object:
+    return _maintenance.maintenance_flow(
+        name="maintenance",
+        step=_MaintenanceStep,
+        params_type=_MaintenanceParams,
+        config_path=str(_write_config(tmp_path)),
+        source_file=__file__,
+    )
+
+
+def test_maintenance_flow_bridges_logs_to_current_flow_run(
+    monkeypatch: pytest.MonkeyPatch,
+    tmp_path: Path,
+) -> None:
+    flow_run_id = uuid.uuid4()
+    runner = MagicMock()
+    runner.run.return_value = MaintenanceReport()
+    install = MagicMock()
+    uninstall = MagicMock()
+
+    monkeypatch.setattr(_maintenance, "prefect_flow_run_id", lambda: flow_run_id)
+    monkeypatch.setattr(_maintenance, "install_log_bridge", install)
+    monkeypatch.setattr(_maintenance, "uninstall_log_bridge", uninstall)
+    monkeypatch.setattr(_maintenance, "_load_yaml", lambda path: (MagicMock(), path))
+    monkeypatch.setattr(_maintenance.MaintenanceRunner, "from_config", lambda config: runner)
+
+    flow = _build_flow(tmp_path)
+    flow.fn(dry_run=True)  # type: ignore[attr-defined]
+
+    install.assert_called_once_with(flow_run_id)
+    uninstall.assert_called_once_with()
+    runner.run.assert_called_once()
+
+
+def test_maintenance_flow_uninstalls_log_bridge_after_failure(
+    monkeypatch: pytest.MonkeyPatch,
+    tmp_path: Path,
+) -> None:
+    uninstall = MagicMock()
+
+    monkeypatch.setattr(_maintenance, "prefect_flow_run_id", uuid.uuid4)
+    monkeypatch.setattr(_maintenance, "install_log_bridge", MagicMock())
+    monkeypatch.setattr(_maintenance, "uninstall_log_bridge", uninstall)
+    monkeypatch.setattr(
+        _maintenance,
+        "_load_yaml",
+        MagicMock(side_effect=RuntimeError("storage config failed")),
+    )
+
+    flow = _build_flow(tmp_path)
+    with pytest.raises(RuntimeError, match="storage config failed"):
+        flow.fn(dry_run=False)  # type: ignore[attr-defined]
+
+    uninstall.assert_called_once_with()

{loom_kernel-0.12.1 → loom_kernel-0.12.2}/uv.lock

@@ -1583,7 +1583,7 @@ wheels = [
 
 [[package]]
 name = "loom-kernel"
-version = "0.12.1"
+version = "0.12.2"
 source = { editable = "." }
 dependencies = [
     { name = "aiocache" },
@@ -2315,8 +2315,8 @@ name = "pendulum"
 version = "3.2.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "python-dateutil", marker = "python_full_version < '3.13'" },
-    { name = "tzdata", marker = "python_full_version < '3.13'" },
+    { name = "python-dateutil" },
+    { name = "tzdata" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/cb/72/9a51afa0a822b09e286c4cb827ed7b00bc818dac7bd11a5f161e493a217d/pendulum-3.2.0.tar.gz", hash = "sha256:e80feda2d10fa3ff8b1526715f7d33dcb7e08494b3088f2c8a3ac92d4a4331ce", size = 86912, upload-time = "2026-01-30T11:22:24.093Z" }
 wheels = [
@@ -3951,8 +3951,8 @@ name = "whenever"
 version = "0.10.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "tzdata", marker = "python_full_version >= '3.13' and sys_platform == 'win32'" },
-    { name = "tzlocal", marker = "python_full_version >= '3.13' and sys_platform != 'darwin' and sys_platform != 'linux'" },
+    { name = "tzdata", marker = "sys_platform == 'win32'" },
+    { name = "tzlocal", marker = "sys_platform != 'darwin' and sys_platform != 'linux'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/fd/13/281bc0bc19f0e2e9559a22137607271d6e684b0f9bcbaf0a95b95a5e9258/whenever-0.10.0.tar.gz", hash = "sha256:a5ef2b5493531de95294080495d4d9bce9691b4b3d681c6952c7a8b3d52d7a04", size = 429588, upload-time = "2026-04-05T18:43:50.199Z" }
 wheels = [

loom_kernel-0.12.1/CHANGELOG_RELEASE.md

@@ -1,15 +0,0 @@
-# 🚀 Release 0.12.1 ([#66](https://github.com/the-reacher-data/loom-py/pull/66)) ([`f3c045e`](https://github.com/the-reacher-data/loom-py/commit/f3c045ec6b9e915e5d2c9f55d7134a7be37c1475))
-
-
-
-## 🐛 Fixes
-### runner
-- **runner:** wire ClickHouseClientExecutor in from_config for ClientStep
-
-
-
-
-
-
-
-