pandaai-cli 0.1.0__py3-none-any.whl

cli.py

@@ -0,0 +1,442 @@
+#!/usr/bin/env python3
+"""
+PandaAI 因子分析 CLI
+
+用法:
+  python cli.py login [--phone PHONE] [--password PWD]
+  python cli.py factor_create --formula "..." [--name NAME]
+  python cli.py factor_info <factor_id>
+  python cli.py factor_update <factor_id> [--name NAME] [--adjustment-cycle N]
+  python cli.py factor_run <factor_id> [--download]
+  python cli.py factor_list [--limit N]
+  python cli.py balance
+  python cli.py factor_result <run_id>
+  python cli.py factor_delete <factor_id> [factor_id...] [--pattern PATTERN]
+
+通用参数: --config PATH, --json
+"""
+
+import argparse
+import getpass
+import json
+import sys
+import os
+import traceback
+from typing import Optional, Dict, Any, Tuple
+
+import requests
+import yaml
+
+from template import build_factor_json
+from pandaai import (
+    STATUS_SUCCESS,
+    load_config, save_token, config_path,
+    do_login, login,
+    get_balance,
+    create_factor, run_factor, poll_factor_run, get_factor_output,
+    get_factor, update_factor, extract_factor_info, apply_factor_updates,
+    get_factor_run_results, format_factor_analysis_results,
+    list_factors, delete_factors, format_factor_list, enrich_factors_with_runs,
+    format_success, format_failure,
+    download_csv,
+)
+
+
+def _fail(error_type: str, message: str) -> None:
+    """输出错误 JSON 并退出"""
+    print(json.dumps(format_failure(status="ERROR", error_type=error_type, message=message), ensure_ascii=False))
+    sys.exit(1)
+
+
+def main():
+    parser = argparse.ArgumentParser(description="PandaAI 因子分析 CLI")
+    parser.add_argument("--config", type=str, default=None, help="配置文件路径")
+    parser.add_argument("--json", action="store_true", help="仅输出 JSON")
+
+    sub = parser.add_subparsers(dest="command", help="可用命令")
+
+    # ── login ──
+    p = sub.add_parser("login", help="登录并保存 token")
+    p.add_argument("--phone", type=str, default=None, help="手机号")
+    p.add_argument("--password", type=str, default=None, help="密码")
+
+    # ── factor_create ──
+    p = sub.add_parser("factor_create", help="创建因子分析（不执行），返回 factor_id")
+    cg = p.add_mutually_exclusive_group(required=True)
+    cg.add_argument("--code", type=str, help="Python 因子代码")
+    cg.add_argument("--formula", type=str, help="因子公式")
+    cg.add_argument("--file", type=str, help="从文件读取代码/公式")
+    p.add_argument("--name", type=str, default="新建因子分析", help="因子分析名称")
+    p.add_argument("--start-date", type=str, default=None, help="因子构建开始日期 YYYYMMDD")
+    p.add_argument("--end-date", type=str, default=None, help="因子构建结束日期 YYYYMMDD")
+    p.add_argument("--adjustment-cycle", type=str, default="1", help="调仓周期 1-10（默认 1）")
+    p.add_argument("--factor-direction", type=str, default="1", help="因子方向 0=负向 1=正向（默认 1）")
+
+    # ── factor_info ──
+    p = sub.add_parser("factor_info", help="查看因子详情（代码/公式 + 所有参数）")
+    p.add_argument("factor_id", type=str, help="因子分析 ID")
+
+    # ── factor_update ──
+    p = sub.add_parser("factor_update", help="修改因子参数")
+    p.add_argument("factor_id", type=str, help="因子分析 ID")
+    p.add_argument("--name", type=str, default=None, help="新名称")
+    ug = p.add_mutually_exclusive_group()
+    ug.add_argument("--code", type=str, default=None, help="新 Python 因子代码")
+    ug.add_argument("--formula", type=str, default=None, help="新因子公式")
+    ug.add_argument("--file", type=str, default=None, help="从文件读取新代码/公式")
+    p.add_argument("--start-date", type=str, default=None, help="因子构建开始日期 YYYYMMDD")
+    p.add_argument("--end-date", type=str, default=None, help="因子构建结束日期 YYYYMMDD")
+    p.add_argument("--adjustment-cycle", type=str, default=None, help="调仓周期 1-10")
+    p.add_argument("--factor-direction", type=str, default=None, help="因子方向 0=负向 1=正向")
+
+    # ── factor_run ──
+    p = sub.add_parser("factor_run", help="执行已有因子分析（启动 + 轮询 + 结果）")
+    p.add_argument("factor_id", type=str, help="因子分析 ID")
+    p.add_argument("--download", type=str, nargs="?", const="", default=None, help="下载 CSV（默认 ~/Downloads）")
+    p.add_argument("--poll-interval", type=int, default=2, help="轮询间隔（秒）")
+    p.add_argument("--timeout", type=int, default=600, help="超时时间（秒）")
+
+    # ── factor_list ──
+    p = sub.add_parser("factor_list", help="列出所有因子分析")
+    p.add_argument("--limit", type=int, default=100, help="每页条数（最大100）")
+    p.add_argument("--offset", type=int, default=0, help="偏移量")
+    p.add_argument("--no-detail", action="store_true", help="跳过获取分析摘要（更快）")
+
+    # ── balance ──
+    sub.add_parser("balance", help="查询算力余额")
+
+    # ── factor_result ──
+    p = sub.add_parser("factor_result", help="查询运行结果（14 个分析接口）")
+    p.add_argument("run_id", type=str, help="运行 ID")
+    p.add_argument("--download", type=str, nargs="?", const="", default=None, help="下载 CSV")
+
+    # ── factor_delete ──
+    p = sub.add_parser("factor_delete", help="删除因子分析")
+    p.add_argument("factor_id", type=str, nargs="*", help="因子分析 ID（可多个）")
+    p.add_argument("--pattern", type=str, default=None, help="按名称前缀批量删除")
+    p.add_argument("--yes", action="store_true", help="跳过确认")
+
+    args = parser.parse_args()
+
+    if not args.command:
+        parser.print_help()
+        sys.exit(1)
+
+    # ═══════════════════════════════════════════
+    # 加载配置
+    # ═══════════════════════════════════════════
+    config = load_config(args.config)
+
+    # ═══════════════════════════════════════════
+    # login
+    # ═══════════════════════════════════════════
+    if args.command == "login":
+        phone = args.phone or input("手机号: ").strip()
+        password = args.password
+        if not password:
+            password = getpass.getpass("密码: ").strip()
+        if not phone or not password:
+            _fail("LOGIN_FAILED", "手机号和密码不能为空")
+        token, uid, _ = do_login(config, phone, password, config.get("country_code", "86"))
+        save_token(token, uid, args.config)
+        print(f"✅ 登录成功! userId={uid}")
+        return
+
+    # ═══════════════════════════════════════════
+    # 认证
+    # ═══════════════════════════════════════════
+    token, uid, _ = login(config)
+    if not args.json:
+        pass  # token 有效，静默继续
+
+    # ═══════════════════════════════════════════
+    # create
+    # ═══════════════════════════════════════════
+    if args.command == "factor_create":
+        if args.file:
+            try:
+                with open(args.file, "r") as f:
+                    content = f.read()
+            except Exception as e:
+                _fail("INPUT_ERROR", f"读取文件失败: {e}")
+        elif args.code:
+            content = args.code
+        else:
+            content = args.formula
+        mode = "formula" if args.formula else "code"
+        if not content or not content.strip():
+            _fail("INPUT_ERROR", "代码/公式内容为空")
+
+        # 校验调仓周期
+        adj_cycle = args.adjustment_cycle
+        try:
+            adj_int = int(adj_cycle)
+            if adj_int < 1 or adj_int > 10:
+                _fail("INPUT_ERROR", "调仓周期必须在 1-10 之间")
+        except ValueError:
+            _fail("INPUT_ERROR", f"调仓周期必须为整数: {adj_cycle}")
+
+        factor_json = build_factor_json(content, mode, name=args.name,
+                                        start_date=args.start_date, end_date=args.end_date,
+                                        adjustment_cycle=args.adjustment_cycle,
+                                        factor_direction=args.factor_direction)
+        factor_id = create_factor(config, token, uid, factor_json)
+        if args.json:
+            print(json.dumps({"success": True, "factor_id": factor_id}, ensure_ascii=False))
+        else:
+            print(f"✅ 因子分析已创建: {factor_id}")
+        return
+
+    # ═══════════════════════════════════════════
+    # info
+    # ═══════════════════════════════════════════
+    if args.command == "factor_info":
+        factor_data = get_factor(config, token, uid, args.factor_id)
+        if factor_data is None:
+            _fail("INFO_FAILED", f"获取因子信息失败: {args.factor_id}")
+
+        info = extract_factor_info(factor_data)
+
+        if args.json:
+            print(json.dumps({"success": True, **info}, ensure_ascii=False, indent=2))
+        else:
+            from datetime import datetime as dt
+            mode_label = "Python 代码" if info["mode"] == "code" else "因子公式"
+            created = ""
+            if info.get("create_at"):
+                try:
+                    created = dt.fromtimestamp(info["create_at"] / 1000).strftime("%Y-%m-%d %H:%M:%S")
+                except Exception:
+                    created = str(info["create_at"])
+
+            print(f"名称:     {info['name']}")
+            print(f"ID:       {info['_id']}")
+            print(f"类型:     {mode_label}")
+            print(f"创建时间: {created}")
+            print(f"调仓周期: {info['adjustment_cycle']}")
+            print(f"分组数量: {info['group_number']}")
+            print(f"因子方向: {info['factor_direction']} ({'正向' if info['factor_direction'] == '1' else '负向'})")
+            print(f"股票池:   {info['stock_pool']}")
+            print(f"开始日期: {info['start_date']}")
+            print(f"结束日期: {info['end_date']}")
+            if info["last_run_id"]:
+                print(f"最后运行: {info['last_run_id']}")
+            print(f"\n{'─' * 50}")
+            print(f"【{mode_label}】")
+            print(info["content"])
+        return
+
+    # ═══════════════════════════════════════════
+    # update
+    # ═══════════════════════════════════════════
+    if args.command == "factor_update":
+        factor_data = get_factor(config, token, uid, args.factor_id)
+        if factor_data is None:
+            _fail("UPDATE_FAILED", f"获取因子信息失败: {args.factor_id}")
+
+        updates = {}
+
+        if args.name is not None:
+            updates["name"] = args.name
+
+        if args.file:
+            try:
+                with open(args.file, "r") as f:
+                    updates["content"] = f.read()
+            except Exception as e:
+                _fail("INPUT_ERROR", f"读取文件失败: {e}")
+            updates["mode"] = "code"
+        elif args.code is not None:
+            updates["content"] = args.code
+            updates["mode"] = "code"
+        elif args.formula is not None:
+            updates["content"] = args.formula
+            updates["mode"] = "formula"
+
+        if args.start_date is not None:
+            updates["start_date"] = args.start_date
+        if args.end_date is not None:
+            updates["end_date"] = args.end_date
+
+        if args.adjustment_cycle is not None:
+            try:
+                adj_int = int(args.adjustment_cycle)
+                if adj_int < 1 or adj_int > 10:
+                    _fail("INPUT_ERROR", "调仓周期必须在 1-10 之间")
+            except ValueError:
+                _fail("INPUT_ERROR", f"调仓周期必须为整数: {args.adjustment_cycle}")
+            updates["adjustment_cycle"] = args.adjustment_cycle
+
+        if args.factor_direction is not None:
+            if args.factor_direction not in ("0", "1"):
+                _fail("INPUT_ERROR", "因子方向必须为 0(负向) 或 1(正向)")
+            updates["factor_direction"] = args.factor_direction
+
+        if not updates:
+            print("没有需要更新的参数")
+            return
+
+        factor_json = apply_factor_updates(factor_data, updates)
+        factor_id = update_factor(config, token, uid, args.factor_id, factor_json)
+        if args.json:
+            print(json.dumps({"success": True, "factor_id": factor_id, "updated": list(updates.keys())}, ensure_ascii=False))
+        else:
+            print(f"✅ 因子分析已更新: {factor_id}")
+            print(f"   更新字段: {', '.join(updates.keys())}")
+        return
+
+    # ═══════════════════════════════════════════
+    # run
+    # ═══════════════════════════════════════════
+    if args.command == "factor_run":
+        factor_id = args.factor_id
+
+        balance_start = get_balance(config, token)
+        if not args.json:
+            power_start = balance_start.get("computingPower") if isinstance(balance_start, dict) else "?"
+            print(f"[INFO] 当前算力: {power_start}", file=sys.stderr)
+
+        run_id, run_error = run_factor(config, token, uid, factor_id)
+        if run_id is None:
+            balance_end = get_balance(config, token) or balance_start
+            power_end = balance_end.get("computingPower") if isinstance(balance_end, dict) else None
+            error_msg = run_error or "启动因子分析失败"
+            if not args.json:
+                print(f"[ERROR] {error_msg}", file=sys.stderr)
+            result = {
+                "success": False, "status": "RUN_FAILED",
+                "factor_id": factor_id, "factor_run_id": None,
+                "error": {"type": "RUN_FAILED", "message": error_msg},
+                "billing": {"balance": power_end, "status": "ok"}
+            }
+            print(json.dumps(result, ensure_ascii=False, indent=2))
+            sys.exit(1)
+
+        if not args.json:
+            print(f"[INFO] 因子分析已启动: {run_id}", file=sys.stderr)
+            print(f"[INFO] 等待执行完成...", file=sys.stderr)
+
+        run_result = poll_factor_run(config, token, uid, run_id, args.poll_interval, args.timeout)
+
+        output = None
+        full_results = None
+        if run_result["status"] == STATUS_SUCCESS:
+            output = get_factor_output(config, token, uid, factor_id)
+            full_results = get_factor_run_results(config, token, uid, run_id)
+
+        balance_end = get_balance(config, token) or balance_start
+
+        s = run_result["status"]
+        if s == STATUS_SUCCESS:
+            result = format_success(factor_id, run_id, run_result, balance_start, balance_end, output)
+            result["results"] = full_results
+        elif s == 3:  # STATUS_FAILED
+            errors = run_result.get("errors", [])
+            msg = "; ".join(e.get("error_message", e.get("message", "")) for e in errors) if errors else "因子分析执行失败"
+            result = format_failure("FAILED", "WORKFLOW_FAILED", msg, factor_id, run_id, run_result, balance_start, balance_end)
+        elif s == 6:  # STATUS_BILLING_STOP
+            result = format_failure("BILLING_STOP", "INSUFFICIENT_BALANCE", "余额不足", factor_id, run_id, run_result, balance_start, balance_end)
+        elif s == "TIMEOUT":
+            result = format_failure("TIMEOUT", "TIMEOUT", f"轮询超时 ({args.timeout}s)", factor_id, run_id, run_result, balance_start, balance_end)
+        else:
+            result = format_failure("UNKNOWN", "UNKNOWN", f"未知状态: {s}", factor_id, run_id, run_result, balance_start, balance_end)
+
+        print(json.dumps(result, ensure_ascii=False, indent=2))
+        if s == STATUS_SUCCESS and full_results and not args.json:
+            print("\n" + format_factor_analysis_results(full_results), file=sys.stderr)
+            download_csv(full_results, args.download)
+        return
+
+    # ═══════════════════════════════════════════
+    # list
+    # ═══════════════════════════════════════════
+    if args.command == "factor_list":
+        factors, total = list_factors(config, token, uid, limit=args.limit, offset=args.offset)
+        if factors is None:
+            _fail("LIST_FAILED", "获取因子列表失败")
+
+        if not args.no_detail and factors:
+            if not args.json:
+                print(f"[INFO] 正在获取分析摘要...", file=sys.stderr)
+            factors = enrich_factors_with_runs(config, token, uid, factors)
+
+        if args.json:
+            print(json.dumps({"success": True, "total": total, "count": len(factors),
+                             "limit": args.limit, "offset": args.offset, "factors": factors},
+                            ensure_ascii=False, indent=2))
+        else:
+            print(format_factor_list(factors, total_count=total, limit=args.limit, offset=args.offset))
+        return
+
+    # ═══════════════════════════════════════════
+    # balance
+    # ═══════════════════════════════════════════
+    if args.command == "balance":
+        balance = get_balance(config, token)
+        if balance is None:
+            _fail("BALANCE_FAILED", "查询算力余额失败")
+        if args.json:
+            print(json.dumps({"success": True, "balance": balance}, ensure_ascii=False))
+        else:
+            power = balance.get("computingPower", "?")
+            print(f"算力: {power}")
+        return
+
+    # ═══════════════════════════════════════════
+    # factor_result
+    # ═══════════════════════════════════════════
+    if args.command == "factor_result":
+        results = get_factor_run_results(config, token, uid, args.run_id)
+        if args.json:
+            print(json.dumps({"success": "error" not in results, **results}, ensure_ascii=False, indent=2))
+        else:
+            print(format_factor_analysis_results(results))
+            download_csv(results, args.download)
+        return
+
+    # ═══════════════════════════════════════════
+    # delete
+    # ═══════════════════════════════════════════
+    if args.command == "factor_delete":
+        if args.pattern:
+            ids = []
+            offset = 0
+            while True:
+                factors, _ = list_factors(config, token, uid, limit=100, offset=offset)
+                if not factors:
+                    break
+                for f in factors:
+                    if f.get("name", "").startswith(args.pattern):
+                        ids.append(f["_id"])
+                offset += 100
+            if not ids:
+                print(f"没有匹配 '{args.pattern}' 的因子分析")
+                return
+        elif args.factor_id:
+            ids = args.factor_id
+        else:
+            print("用法: factor_delete <factor_id> [factor_id...]  或  factor_delete --pattern <前缀>")
+            sys.exit(1)
+
+        if not args.yes and not args.json:
+            print(f"将删除 {len(ids)} 个因子分析: {', '.join(ids[:5])}{'...' if len(ids) > 5 else ''}")
+            confirm = input("确认? (y/N): ").strip().lower()
+            if confirm not in ("y", "yes"):
+                print("已取消")
+                return
+
+        deleted, error = delete_factors(config, token, uid, ids)
+        if error:
+            _fail("DELETE_FAILED", f"删除失败 (已删{deleted}): {error}")
+        print(f"✅ 已删除 {deleted} 个因子分析")
+        return
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except SystemExit:
+        pass
+    except Exception:
+        print(json.dumps(format_failure("ERROR", "UNKNOWN", f"CLI内部错误: {traceback.format_exc()}"), ensure_ascii=False))
+        sys.exit(1)

pandaai/__init__.py

@@ -0,0 +1,12 @@
+"""PandaAI CLI 功能模块"""
+
+from .constants import *
+from .config import load_config, save_token, config_path
+from .auth import do_login, login, api_url, make_headers
+from .billing import get_balance
+from .factor_core import create_factor, run_factor, poll_factor_run, get_factor_output, get_factor, update_factor, extract_factor_info, apply_factor_updates
+from .output import get_factor_run_detail, get_node_output, extract_task_id, fetch_factor_analysis, get_factor_run_results
+from .factor import FACTOR_ANALYSIS_ENDPOINTS, format_factor_analysis_results
+from .factor_list import list_factors, delete_factors, format_factor_list, enrich_factors_with_runs
+from .formatting import format_success, format_failure, _calc_duration
+from .download import download_csv

pandaai/auth.py

@@ -0,0 +1,103 @@
+"""认证：登录、token 验证"""
+
+import hashlib
+import sys
+import json
+from typing import Optional, Tuple
+
+import requests
+
+from .config import save_token
+
+
+def _api_url(config: dict, path: str) -> str:
+    """拼接完整 API 地址"""
+    base = config["gateway_url"].rstrip("/")
+    return f"{base}{path}"
+
+
+def _headers(token: str = None, uid: str = None, extra: dict = None) -> dict:
+    """构造请求头"""
+    h = {"Content-Type": "application/json"}
+    if token:
+        h["Authorization"] = token
+    if uid:
+        h["uid"] = str(uid)
+    if extra:
+        h.update(extra)
+    return h
+
+
+def _fail(error_type: str, message: str) -> None:
+    """输出错误 JSON 并退出（导入自 formatting）"""
+    from .formatting import format_failure
+    print(json.dumps(format_failure(status="ERROR", error_type=error_type, message=message), ensure_ascii=False))
+    sys.exit(1)
+
+
+def do_login(config: dict, phone: str, password: str, country_code: str = "86") -> Tuple[str, str, dict]:
+    """
+    执行密码登录，返回 (token, uid, userinfo)
+    """
+    url = _api_url(config, "/login/pw")
+    body = {
+        "phone": phone,
+        "password": hashlib.md5(password.encode()).hexdigest(),
+        "countryCode": country_code,
+    }
+    try:
+        resp = requests.post(url, json=body, timeout=10)
+        data = resp.json()
+    except Exception as e:
+        return _fail("LOGIN_FAILED", f"登录请求失败: {e}")
+
+    if data.get("code") != "200":
+        return _fail("LOGIN_FAILED", f"登录失败: {data.get('message', '未知错误')}")
+
+    token = data.get("data")
+    if not token:
+        return _fail("LOGIN_FAILED", "登录响应中没有 token")
+
+    user_id, userinfo = _get_user_info(config, token)
+    if user_id is None or userinfo is None:
+        return _fail("LOGIN_FAILED", "获取用户信息失败")
+
+    return token, user_id, userinfo
+
+
+def login(config: dict) -> Tuple[str, str, dict]:
+    """
+    认证入口：优先用已保存的 token，失败则提示用户登录
+    返回 (token, uid, userinfo)
+    """
+    saved_token = config.get("token", "")
+    saved_uid = config.get("uid", "")
+
+    if saved_token and saved_uid:
+        uid, userinfo = _get_user_info(config, saved_token)
+        if uid is not None and userinfo is not None:
+            return saved_token, uid, userinfo
+
+    return _fail("LOGIN_REQUIRED",
+                 "未登录或 token 已过期，请先执行: python cli.py login --phone <手机号> --password <密码>")
+
+
+def _get_user_info(config: dict, token: str) -> Tuple[Optional[str], Optional[dict]]:
+    """调 /user/info 获取 userId 和用户信息"""
+    url = _api_url(config, "/user/info")
+    try:
+        resp = requests.get(url, headers=_headers(token=token), timeout=10)
+        data = resp.json()
+        if data.get("code") == "200":
+            user_data = data.get("data", {})
+            if isinstance(user_data, dict):
+                uid = user_data.get("id")
+                return (str(uid) if uid else None), user_data
+    except Exception:
+        pass
+    return None, None
+
+
+# 导出工具函数供其他模块使用
+api_url = _api_url
+make_headers = _headers

pandaai/billing.py

@@ -0,0 +1,20 @@
+"""算力余额查询"""
+
+from typing import Optional
+
+import requests
+
+from .auth import api_url, make_headers
+
+
+def get_balance(config: dict, token: str) -> Optional[dict]:
+    """查询用户算力余额"""
+    url = api_url(config, "/userWallet/myWallet")
+    try:
+        resp = requests.get(url, headers=make_headers(token=token), timeout=10)
+        data = resp.json()
+        if data.get("code") == "200":
+            return data.get("data", {})
+    except Exception:
+        pass
+    return None

pandaai/config.py

@@ -0,0 +1,40 @@
+"""配置加载与 token 持久化"""
+
+import os
+import sys
+import json
+import yaml
+
+
+def config_path(custom_path: str = None) -> str:
+    """获取配置文件路径。默认 ~/.pandaai/config.yaml，可通过 --config 覆盖"""
+    if custom_path is None:
+        return os.path.join(os.path.expanduser("~"), ".pandaai", "config.yaml")
+    return custom_path
+
+
+def load_config(config_path_arg: str = None) -> dict:
+    """加载配置文件"""
+    path = config_path(config_path_arg)
+    if not os.path.exists(path):
+        print(json.dumps({"success": False, "error": {"type": "CONFIG_ERROR", "message": f"配置文件不存在: {path}"}}))
+        sys.exit(1)
+    with open(path, "r") as f:
+        return yaml.safe_load(f)
+
+
+def save_token(token: str, uid: str, config_path_arg: str = None) -> None:
+    """保存 token 和 uid 到配置文件（目录和文件不存在则自动创建）"""
+    path = config_path(config_path_arg)
+    os.makedirs(os.path.dirname(path), exist_ok=True)
+    if os.path.exists(path):
+        with open(path, "r") as f:
+            config = yaml.safe_load(f) or {}
+    else:
+        config = {}
+    config["token"] = token
+    config["uid"] = uid
+    # 确保至少有 gateway_url
+    config.setdefault("gateway_url", "https://www.pandaaiquant.com/pandaApi")
+    with open(path, "w") as f:
+        yaml.dump(config, f, allow_unicode=True, default_flow_style=False)

pandaai/constants.py

@@ -0,0 +1,15 @@
+"""状态常量"""
+
+STATUS_PENDING = 0
+STATUS_RUNNING = 1
+STATUS_SUCCESS = 2
+STATUS_FAILED = 3
+STATUS_BILLING_STOP = 6  # 余额不足终止
+
+STATUS_MAP = {
+    STATUS_PENDING: "PENDING",
+    STATUS_RUNNING: "RUNNING",
+    STATUS_SUCCESS: "SUCCESS",
+    STATUS_FAILED: "FAILED",
+    STATUS_BILLING_STOP: "BILLING_STOP",
+}

pandaai/download.py

@@ -0,0 +1,41 @@
+"""CSV 下载"""
+
+import os
+import sys
+
+import requests
+
+
+def download_csv(results: dict, download_arg: str) -> None:
+    """从结果中提取 CSV URL 并下载"""
+    csv_url = None
+    for _uuid, data in results.get("nodes", {}).items():
+        if isinstance(data, dict) and "url" in data:
+            csv_url = data["url"]
+            break
+
+    if not csv_url:
+        return
+
+    if download_arg == "":
+        save_dir = os.path.join(os.path.expanduser("~"), "Downloads")
+        filename = csv_url.split("/")[-1] if "/" in csv_url else "factor_data.csv"
+        save_path = os.path.join(save_dir, filename)
+    elif download_arg:
+        if os.path.isdir(download_arg) or download_arg.endswith("/"):
+            filename = csv_url.split("/")[-1] if "/" in csv_url else "factor_data.csv"
+            save_path = os.path.join(download_arg, filename)
+        else:
+            save_path = download_arg
+    else:
+        return
+
+    try:
+        resp = requests.get(csv_url, timeout=60)
+        resp.raise_for_status()
+        os.makedirs(os.path.dirname(save_path) or ".", exist_ok=True)
+        with open(save_path, "wb") as f:
+            f.write(resp.content)
+        print(f"📥 CSV已保存: {save_path} ({len(resp.content)} bytes)", file=sys.stderr)
+    except Exception as e:
+        print(f"⚠ CSV下载失败: {e}", file=sys.stderr)