Upload File #

import hashlib
import requests

def upload_single_file(file_path, upload_info):
    """单文件上传"""
    
    # 计算 MD5
    with open(file_path, "rb") as f:
        file_data = f.read()
        md5 = hashlib.md5(file_data).hexdigest()
    
    # 上传
    with open(file_path, "rb") as f:
        response = requests.post(
            f"https://{upload_info['global_domain']}/ous/api/v2/single/upload",
            headers={"ous-token-v2": upload_info["ous_token"]},
            files={"file": f},
            data={"md5": md5}
        )
    
    if response.json()["c"] != "0":
        raise Exception("上传失败")
    
    return response.json()["d"]["taskId"]

def upload_large_file(file_path, upload_info):
    """分片上传大文件"""
    
    file_size = os.path.getsize(file_path)
    block_size = upload_info["block_size"]
    blocks = math.ceil(file_size / block_size)
    
    # 计算 MD5
    with open(file_path, "rb") as f:
        md5 = hashlib.md5(f.read()).hexdigest()
    
    # 初始化
    init_response = requests.post(
        f"https://{upload_info['global_domain']}/ous/api/v2/block/upload/init",
        headers={"ous-token-v2": upload_info["ous_token"]},
        data={
            "md5": md5,
            "blocks": blocks,
            "size": file_size,
            "name": os.path.basename(file_path)
        }
    )
    
    task_id = init_response.json()["d"]["taskId"]
    
    # 上传各分片（并发控制在2以内）
    with open(file_path, "rb") as f:
        for block_num in range(1, blocks + 1):
            chunk = f.read(block_size)
            
            response = requests.post(
                f"https://{upload_info['global_domain']}/ous/api/v2/block/upload/part",
                headers={"ous-token-v2": upload_info["ous_token"]},
                files={"file": chunk},
                data={"block": block_num}
            )
            
            print(f"分片 {block_num}/{blocks} 上传完成")
    
    return task_id

def wait_for_upload(upload_info, timeout=30):
    """等待上传完成"""
    
    start_time = time.time()
    
    while time.time() - start_time < timeout:
        response = requests.get(
            f"https://{upload_info['global_domain']}/ous/api/v2/upload/status",
            headers={"ous-token-v2": upload_info["ous_token"]}
        )
        
        data = response.json()["d"]
        status = data["status"]
        
        if status == 5:  # 成功
            return data["url"]
        elif status in [6, 8]:  # 失败或中止
            raise Exception(f"上传失败: status={status}")
        
        progress = data.get("blockUpload", {}).get("progressPercent", 0)
        print(f"状态: {status}, 进度: {progress}%")
        
        time.sleep(0.5)  # 轮询间隔建议 200ms 以上
    
    raise Exception("上传超时")

# 使用示例
file_url = wait_for_upload(upload_info)
print(f"✓ 文件上传成功: {file_url}")

import os
import hashlib
import requests
import time
import math

def upload_file(file_path, upload_info):
    """
    完整上传流程：自动判断单文件/分片上传，并等待完成
    
    Args:
        file_path: 本地文件路径
        upload_info: 包含 ous_token, global_domain, block_size
    
    Returns:
        url: 上传成功后的文件 URL
    """
    
    file_size = os.path.getsize(file_path)
    block_size = upload_info["block_size"]
    
    # 计算 MD5
    with open(file_path, "rb") as f:
        md5 = hashlib.md5(f.read()).hexdigest()
    
    print(f"📤 上传文件: {os.path.basename(file_path)} ({file_size/1024/1024:.2f}MB)")
    
    # 判断上传方式
    if file_size < block_size:
        # 单文件上传
        print("   使用单文件上传")
        task_id = upload_single_file(file_path, upload_info)
    else:
        # 分片上传
        print(f"   使用分片上传 ({math.ceil(file_size/block_size)} 个分片)")
        task_id = upload_large_file(file_path, upload_info)
    
    # 等待上传完成
    url = wait_for_upload(upload_info)
    
    print(f"   ✓ 上传成功: {url}")
    return url