79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
# pip install esdk-obs-python --trusted-host pypi.org
|
||
import traceback
|
||
import logging
|
||
|
||
from obs import ObsClient
|
||
from obs import PutObjectHeader
|
||
|
||
from Config.Config import OBS_AK, OBS_SK, OBS_SERVER, OBS_BUCKET
|
||
|
||
# 配置日志
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class ObsUploader:
|
||
def __init__(self, access_key_id, secret_access_key, server):
|
||
"""
|
||
初始化OBS客户端
|
||
:param access_key_id: 访问密钥ID
|
||
:param secret_access_key: 秘密访问密钥
|
||
:param server: OBS服务器地址
|
||
"""
|
||
self.obs_client = ObsClient(
|
||
access_key_id=access_key_id,
|
||
secret_access_key=secret_access_key,
|
||
server=server
|
||
)
|
||
|
||
def upload_file(self, bucket_name, object_key, file_path, metadata=None, headers=None):
|
||
"""
|
||
上传文件到OBS
|
||
:param bucket_name: 桶名称
|
||
:param object_key: 对象键（上传后的文件名）
|
||
:param file_path: 本地文件路径
|
||
:param meta元数据
|
||
:param headers: 请求头
|
||
:return: 成功返回True和响应信息，失败返回False和错误信息
|
||
"""
|
||
try:
|
||
# 初始化默认参数
|
||
if metadata is None:
|
||
metadata = {}
|
||
if headers is None:
|
||
headers = PutObjectHeader()
|
||
|
||
# 执行文件上传
|
||
resp = self.obs_client.putFile(
|
||
bucket_name,
|
||
object_key,
|
||
file_path,
|
||
metadata,
|
||
headers
|
||
)
|
||
|
||
# 检查响应状态
|
||
if resp.status < 300:
|
||
logger.info(f'文件上传成功: {file_path} -> {bucket_name}/{object_key}')
|
||
logger.info(f'requestId: {resp.requestId}')
|
||
logger.info(f'etag: {resp.body.etag}')
|
||
return True, {
|
||
'requestId': resp.requestId,
|
||
'etag': resp.body.etag,
|
||
'versionId': resp.body.versionId,
|
||
'storageClass': resp.body.storageClass
|
||
}
|
||
else:
|
||
error_msg = f'文件上传失败: {resp.errorCode} - {resp.errorMessage}'
|
||
logger.error(error_msg)
|
||
logger.error(f'requestId: {resp.requestId}')
|
||
return False, {
|
||
'requestId': resp.requestId,
|
||
'errorCode': resp.errorCode,
|
||
'errorMessage': resp.errorMessage
|
||
}
|
||
except Exception as e:
|
||
error_msg = f'文件上传异常: {str(e)}'
|
||
logger.error(error_msg)
|
||
logger.error(traceback.format_exc())
|
||
return False, {'error': str(e), 'traceback': traceback.format_exc()}
|