import os
import tempfile
import uuid

import requests
from qiniu import Auth, put_file, etag, BucketManager

# 七牛云配置
QINIU_ACCESS_KEY = '-8ezB_d-8-eUFTMvhOGbGzgeQRPeKQnaQ3DBcUxo'
QINIU_SECRET_KEY = 'KFhkYxTAJ2ZPN3ZS3euTsfWk8-C92rKgkhAMkDRN'
QINIU_BUCKET_NAME = 'bzdcoldverify'
QINIU_BUCKET_DOMAIN = 'https://bzdcoldverifyoss.baozhida.cn'  # 例如：'http://xxx.clouddn.com'

# 初始化七牛云Auth
q = Auth(QINIU_ACCESS_KEY, QINIU_SECRET_KEY)

def download_file(url):
    """下载文件到临时目录"""
    local_filename = os.path.join("./temp", str(uuid.uuid4()) + '.pdf')
    with requests.get(url, stream=True) as r:
        r.raise_for_status()
        with open(local_filename, 'wb') as f:
            for chunk in r.iter_content(chunk_size=8192):
                f.write(chunk)
    return local_filename

def upload_to_qiniu(local_file, key):
    """上传文件到七牛云"""
    token = q.upload_token(QINIU_BUCKET_NAME, key)
    ret, info = put_file(token, key, local_file)
    if info.status_code == 200:
        return f"{QINIU_BUCKET_DOMAIN}/{key}"
    else:
        raise Exception(f"七牛云上传失败: {info}")