dataease/scripts/multi_deploy.py

import sys
import os
import json
import subprocess

# Add local path for engine import
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from multi_engine import MultiDataEaseChartEngine

# --- Configuration from Environment ---
def load_dotenv():
    env_path = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), ".env")
    if os.path.exists(env_path):
        with open(env_path) as f:
            for line in f:
                line = line.strip()
                if line and not line.startswith('#') and '=' in line:
                    key, value = line.split('=', 1)
                    os.environ[key.strip()] = value.strip().strip('"').strip("'")

load_dotenv()

ACCESS_KEY = os.environ.get("DATAEASE_ACCESS_KEY")
SECRET_KEY = os.environ.get("DATAEASE_SECRET_KEY")
BASE_URL   = os.environ.get("DATAEASE_BASE_URL")

# Screenshot configuration - auto-detect OpenClaw workspace for MEDIA: display
def _default_output_dir():
    """优先使用 OpenClaw workspace 目录，否则用 skill 本地 output 目录"""
    env_dir = os.environ.get("DATAEASE_OUTPUT_DIR")
    if env_dir:
        return env_dir
    home = os.path.expanduser("~")
    workspace_dir = os.path.join(home, ".openclaw", "workspace", "dataease-output")
    if os.path.isdir(os.path.join(home, ".openclaw", "workspace")):
        os.makedirs(workspace_dir, exist_ok=True)
        return workspace_dir
    return os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "output")

OUTPUT_DIR = _default_output_dir()
DEFAULT_PIXEL = "2560*1440"  # 更高分辨率，图片更清晰

def capture_dashboard(dashboard_id, output_format="jpeg"):
    """调用 capture_dashboard.py 进行截图"""
    scripts_dir = os.path.dirname(os.path.abspath(__file__))
    capture_script = os.path.join(scripts_dir, "capture_dashboard.py")

    if not os.path.exists(capture_script):
        return None, "Screenshot script not found"

    # 确保输出目录存在
    os.makedirs(OUTPUT_DIR, exist_ok=True)

    # capture_dashboard.py 使用相同的 BASE_URL（不含 /de2api 后缀）
    capture_base_url = BASE_URL.rstrip("/")

    # 调用截图脚本
    result_format = "0" if output_format == "jpeg" else "1"  # 0=jpeg, 1=pdf
    cmd = [
        "python3", capture_script,
        "capture",
        "--resource-id", str(dashboard_id),
        "--busi-type", "dashboard",
        "--output-dir", OUTPUT_DIR,
        "--result-format", result_format,
        "--base-url", capture_base_url,
        "--pixel", DEFAULT_PIXEL
    ]

    try:
        result = subprocess.run(cmd, capture_output=True, text=True, timeout=120)
        if result.returncode == 0:
            output = json.loads(result.stdout)
            if output.get("ok"):
                return output.get("saved_file"), None
            else:
                return None, output.get("error", "Unknown capture error")
        else:
            return None, result.stderr or "Capture failed"
    except subprocess.TimeoutExpired:
        return None, "Capture timeout"
    except Exception as e:
        return None, str(e)

def main():
    if not all([ACCESS_KEY, SECRET_KEY, BASE_URL]):
        print("Error: Missing required environment variables.")
        sys.exit(1)

    if len(sys.argv) < 3:
        print("Usage: python3 multi_deploy.py <dashboard_title> <charts_json_config> [--no-screenshot]")
        print("Example: python3 multi_deploy.py '综合分析' '[{\"type\":\"bar\",\"dataset_name\":\"ds1\",\"x_axis\":[\"f1\"],\"y_axis\":[\"f2\"]}]'")
        sys.exit(1)

    title = sys.argv[1]
    try:
        charts_config = json.loads(sys.argv[2])
    except Exception as e:
        print(f"Error parsing JSON config: {e}")
        sys.exit(1)

    # 检查是否禁用截图
    no_screenshot = "--no-screenshot" in sys.argv

    engine = MultiDataEaseChartEngine(BASE_URL, ACCESS_KEY, SECRET_KEY)

    try:
        did, url = engine.deploy_multi(title, charts_config)
        print(f"\n✅ Successfully deployed multi-chart dashboard!")
        print(f"Dashboard ID: {did}")
        print(f"URL: {url}")

        # 自动截图
        if not no_screenshot:
            print("\n📸 Capturing screenshot...")
            screenshot_path, error = capture_dashboard(did)
            if screenshot_path:
                print(f"Screenshot saved: {screenshot_path}")
                # 输出 JSON 结果供 Agent 解析
                result = {
                    "ok": True,
                    "dashboard_id": str(did),
                    "url": url,
                    "screenshot": screenshot_path,
                    "title": title,
                    "charts_count": len(charts_config)
                }
                print(f"\n__RESULT_JSON__\n{json.dumps(result, ensure_ascii=False)}\n__END_JSON__")
            else:
                print(f"⚠️ Screenshot failed: {error}")
                result = {
                    "ok": True,
                    "dashboard_id": str(did),
                    "url": url,
                    "screenshot": None,
                    "screenshot_error": error,
                    "title": title,
                    "charts_count": len(charts_config)
                }
                print(f"\n__RESULT_JSON__\n{json.dumps(result, ensure_ascii=False)}\n__END_JSON__")

    except Exception as e:
        print(f"\n❌ Deployment failed: {str(e)}")
        sys.exit(1)

if __name__ == "__main__":
    main()