mirror of
https://github.com/hanzoai/insights.git
synced 2026-10-30 16:15:31 +00:00
zeekay
bc4ddf332b
Mechanically re-derived: bin/debrand runs over a fresh import of posthog/posthog@fa1a9a60d4 (2026-08-06). The pipeline (.hanzo/workflows), the Dockerfiles and bin/docker* entrypoints, compose files, LLM.md and the squashed initial migration are restored from the shipped tree, same as the previous import. The prune's import-closure guard kept 21 products the new core now imports; dropping them is a refactor now, recorded rather than forced.
93 lines
3.1 KiB
Python
Executable File
93 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# ruff: noqa: T201 allow print statements
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
import time
|
|
from importlib import util as importlib_util
|
|
from pathlib import Path
|
|
from types import ModuleType
|
|
|
|
import duckdb
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parent.parent
|
|
sys.path.insert(0, str(REPO_ROOT))
|
|
MAX_RETRIES = 30
|
|
|
|
|
|
def _load_ducklake_common() -> ModuleType:
|
|
"""Load managed warehouse helpers directly from the repo without relying on editable installs."""
|
|
module_path = REPO_ROOT / "products" / "managed_warehouse" / "backend" / "common.py"
|
|
spec = importlib_util.spec_from_file_location("managed_warehouse_common", module_path)
|
|
if not spec or not spec.loader:
|
|
raise ImportError(f"Unable to locate DuckLake helpers at {module_path}")
|
|
module = importlib_util.module_from_spec(spec)
|
|
spec.loader.exec_module(module)
|
|
return module
|
|
|
|
|
|
ducklake_common = _load_ducklake_common()
|
|
|
|
HEALTHCHECK_ALIAS = "ducklake_dev_health"
|
|
|
|
|
|
def check_once(config: dict[str, str]) -> bool:
|
|
reset_performed = False
|
|
conn = duckdb.connect()
|
|
try:
|
|
conn.execute("LOAD ducklake")
|
|
ducklake_common.attach_catalog(conn, config, alias=HEALTHCHECK_ALIAS)
|
|
ducklake_common.run_smoke_check(conn, alias=HEALTHCHECK_ALIAS)
|
|
except (duckdb.NotImplementedException, duckdb.InvalidInputException) as exc:
|
|
if ducklake_common.is_version_mismatch(exc):
|
|
print(f"DuckLake version mismatch detected, resetting local catalog: {exc}")
|
|
conn.close()
|
|
ducklake_common.reset_ducklake_catalog(config)
|
|
print("DuckLake catalog reset complete, retrying health check...")
|
|
conn = duckdb.connect()
|
|
conn.execute("LOAD ducklake")
|
|
ducklake_common.attach_catalog(conn, config, alias=HEALTHCHECK_ALIAS)
|
|
ducklake_common.run_smoke_check(conn, alias=HEALTHCHECK_ALIAS)
|
|
reset_performed = True
|
|
else:
|
|
raise
|
|
finally:
|
|
conn.close()
|
|
|
|
return reset_performed
|
|
|
|
|
|
def main() -> int:
|
|
for attempt in range(1, MAX_RETRIES + 1):
|
|
config = ducklake_common.get_config()
|
|
try:
|
|
reset_performed = check_once(config)
|
|
except KeyboardInterrupt:
|
|
raise
|
|
except Exception as exc: # noqa: BLE001 - any failure should trigger a retry
|
|
print(f"Awaiting DuckLake warmup... ({exc})")
|
|
if attempt >= MAX_RETRIES:
|
|
print(f"DuckLake failed to start after {MAX_RETRIES} attempts, giving up.")
|
|
return 1
|
|
try:
|
|
initialized = ducklake_common.initialize_ducklake(config, alias="ducklake_dev")
|
|
if initialized:
|
|
print("Reinitialized DuckLake catalog, retrying health check...")
|
|
except Exception as setup_exc: # noqa: BLE001
|
|
print(f"DuckLake initialization attempt failed ({setup_exc})")
|
|
time.sleep(1)
|
|
continue
|
|
|
|
if reset_performed:
|
|
print("DuckLake is up after catalog reset.")
|
|
else:
|
|
print("DuckLake is up!")
|
|
return 0
|
|
|
|
return 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|