| |
| """Preflight checks before Hugging Face Space deploy.""" |
|
|
| from __future__ import annotations |
|
|
| import sys |
| from pathlib import Path |
|
|
| ROOT = Path(__file__).resolve().parent.parent |
| sys.path.insert(0, str(ROOT)) |
|
|
|
|
| def main() -> int: |
| errors: list[str] = [] |
|
|
| for rel in ( |
| "app.py", |
| "requirements.txt", |
| "README_HF_SPACE.md", |
| "demo/handlers.py", |
| "demo/theme.py", |
| "morphsql/__init__.py", |
| "model/risk_classifier.joblib", |
| "examples/vertica_legacy", |
| ): |
| if not (ROOT / rel).exists(): |
| errors.append(f"missing required path: {rel}") |
|
|
| try: |
| import app as space_app |
| from app import _build_demo, demo |
|
|
| assert demo is not None |
| built = _build_demo() |
| assert built is not None |
| except Exception as exc: |
| errors.append(f"app import/build failed: {exc}") |
|
|
| try: |
| from demo.handlers import convert_for_ui, convert_upload_for_ui |
| import tempfile |
| from pathlib import Path as P |
|
|
| notes, output, status, share, preview, path, nb, api = convert_for_ui( |
| "SELECT COALESCE(a, 0) AS a FROM t", |
| "snowflake", |
| "pandas", |
| ) |
| assert "import pandas" in output and path.endswith(".py") |
|
|
| with tempfile.TemporaryDirectory() as td: |
| sql = P(td) / "sample.sql" |
| sql.write_text("SELECT COALESCE(x, 0) AS x FROM t", encoding="utf-8") |
| batch = convert_upload_for_ui(str(sql), "", "snowflake", "pyspark") |
| assert batch[6].endswith(".py") |
| except Exception as exc: |
| errors.append(f"convert smoke failed: {exc}") |
|
|
| if errors: |
| print("Space preflight FAILED:") |
| for e in errors: |
| print(f" - {e}") |
| return 1 |
|
|
| print("Space preflight OK — ready to deploy.") |
| return 0 |
|
|
|
|
| if __name__ == "__main__": |
| raise SystemExit(main()) |
|
|
