Query Task Status #

import requests
import time

APPUID = "your_appuid"
BASE_URL = "https://openapi.kujiale.com/v2/aholo"

def wait_for_task(task_id, timeout=600, interval=5):
    """
    等待单个任务完成
    
    Args:
        task_id: 任务ID（注意：接口实际需要 projectId）
        timeout: 最大等待时间（秒）
        interval: 轮询间隔（秒）
    
    Returns:
        result: 任务结果（包含 plyPath, spzPath, sogPath）
    """
    
    print(f"⏳ 等待任务完成: {task_id}")
    
    start_time = time.time()
    
    while time.time() - start_time < timeout:
        response = requests.post(
            f"{BASE_URL}/project/task/list/by-ids",
            params={"appuid": APPUID},
            json=[task_id]
            # 注意：实际需添加签名认证
        )
        
        if response.json()["c"] != "0":
            raise Exception("查询失败")
        
        tasks = response.json()["d"]
        
        if not tasks:
            print("   任务未找到，继续等待...")
            time.sleep(interval)
            continue
        
        task = tasks[0]
        status = task["status"]
        
        # 终态判断
        if status == 3:  # 成功
            print(f"   ✓ 任务完成!")
            return task["result"]
        
        elif status == 4:  # 失败
            raise Exception("任务执行失败")
        
        elif status == 5:  # 取消
            raise Exception("任务被取消")
        
        elif status == 6:  # 超时
            raise Exception("任务超时")
        
        elif status == 7:  # 拒绝
            raise Exception("任务被拒绝")
        
        # 非终态，继续等待
        elapsed = int(time.time() - start_time)
        status_text = {
            0: "排队中",
            1: "等待执行",
            2: "执行中",
            8: "预处理中"
        }.get(status, f"状态{status}")
        
        print(f"   [{elapsed}s] {status_text}")
        time.sleep(interval)
    
    raise Exception(f"超时 ({timeout}s)")

# 使用示例
result = wait_for_task("3FO4K4WBNNYV")

print("\n📦 3D 资产:")
print(f"   PLY: {result['plyPath']}")
print(f"   SPZ: {result['spzPath']}")
print(f"   SOG: {result['sogPath']}")

def query_tasks(task_ids):
    """批量查询多个任务状态"""
    
    response = requests.post(
        f"{BASE_URL}/project/task/list/by-ids",
        params={"appuid": APPUID},
        json=task_ids
    )
    
    return response.json()["d"]

# 查询多个任务
tasks = query_tasks(["task1", "task2", "task3"])

for task in tasks:
    print(f"任务 {task['taskId']}: 状态 {task['status']}")
    
    if task['status'] == 3:
        print(f"  ✓ PLY: {task['result']['plyPath']}")

def create_and_wait(image_url, project_name, project_type="recon", **kwargs):
    """
    创建项目并等待完成
    
    Args:
        image_url: 上传后的图片URL
        project_name: 项目名称
        project_type: "recon" 或 "gen"
        kwargs: 其他参数（scene, quality, prompt 等）
    
    Returns:
        result: 3D 资产结果
    """
    
    # 创建项目
    if project_type == "recon":
        project_id = create_3d_recon_project(
            image_url, project_name,
            scene=kwargs.get("scene", "model"),
            quality=kwargs.get("quality", "NORMAL")
        )
    else:
        project_id = create_3d_gen_project(
            image_url, project_name,
            prompt=kwargs.get("prompt")
        )
    
    # 等待完成
    result = wait_for_task(project_id)
    
    return {
        "projectId": project_id,
        "assets": result
    }

# 使用示例
output = create_and_wait(
    image_url="https://cdn.kujiale.com/object.jpg",
    project_name="工艺品",
    project_type="recon",
    scene="model",
    quality="HIGH"
)

print(f"项目 ID: {output['projectId']}")
print(f"SPZ 文件: {output['assets']['spzPath']}")