AutoDL 14 天自动续命脚本

#!/usr/bin/env python3
"""AutoDL 14天自动续命 — 重建实例并恢复环境"""
import requests, os, time, json

TOKEN = os.environ["AUTODL_TOKEN"]
HOST = "https://api.autodl.com"
HEADERS = {"Authorization": TOKEN, "Content-Type": "application/json"}

# ======== 按你的实例改这里 ========
REGION = "beijingDC2"          # 地区
GPU_SPEC = "pro6000-p"         # GPU规格ID（见API文档附录）
IMAGE_ID = "image-xxxxxxxxx"   # 镜像UUID（私有镜像列表里查）
INSTANCE_NAME = "auto-renew"
DATA_CENTER = ["beijingDC2"]   # 可多选
CUDA_MIN = 113                 # CUDA >= 11.3
GPU_COUNT = 1
DISK_EXPAND = 0               # 系统盘扩容(GB)
START_CMD = "sleep 1"          # 开机后自动执行的命令
# =================================


def _post(path, data=None):
    r = requests.post(f"{HOST}{path}", headers=HEADERS, json=data or {})
    return r.json()


def get_old_instance():
    """查当前实例列表，取最新一个"""
    r = _post("/api/v1/dev/instance/pro/list")
    items = r.get("data", {}).get("list", [])
    if not items:
        raise Exception("没找到实例，先去控制台手动创建一个")
    return items[0]


def release_instance(uuid):
    """先关机，再释放"""
    _post("/api/v1/dev/instance/pro/power_off", {"instance_uuid": uuid})
    time.sleep(10)
    _post("/api/v1/dev/instance/pro/release", {"instance_uuid": uuid})
    print(f"  已释放 {uuid}")


def create_instance():
    """用预设配置创建新实例"""
    data = {
        "data_center_list": DATA_CENTER,
        "req_gpu_amount": GPU_COUNT,
        "expand_system_disk_by_gb": DISK_EXPAND,
        "gpu_spec_uuid": GPU_SPEC,
        "image_uuid": IMAGE_ID,
        "cuda_v_from": CUDA_MIN,
        "instance_name": INSTANCE_NAME,
        "start_command": START_CMD,
    }
    r = _post("/api/v1/dev/instance/pro/create", data)
    info = r.get("data", {})
    uuid = info.get("instance_uuid")
    print(f"  新实例 {uuid} 创建成功")
    return info


def wait_until_running(uuid, timeout=300):
    """等实例开机就绪"""
    for _ in range(timeout // 10):
        r = _post("/api/v1/dev/instance/pro/status", {"instance_uuid": uuid})
        status = r.get("data", {}).get("status")
        if status == "running":
            return True
        time.sleep(10)
    raise Exception("实例启动超时")


def main():
    print("=== AutoDL 14天续命 ===")

    print("[1/4] 查旧实例...")
    old = get_old_instance()
    old_uuid = old["instance_uuid"]
    print(f"  旧实例: {old_uuid}")

    print("[2/4] 创建新实例...")
    new_info = create_instance()
    new_uuid = new_info["instance_uuid"]

    print("[3/4] 等新实例就绪...")
    wait_until_running(new_uuid)
    # 新实例开机后可通过 START_CMD 自动挂载文件存储、拉代码
    # 例如: START_CMD = "bash /root/setup.sh"

    print("[4/4] 释放旧实例...")
    release_instance(old_uuid)

    print(f"=== 完成，新实例 {new_uuid}，倒计时重置 ===")


if __name__ == "__main__":
    main()