56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-# pip install oss2
|
||
import os
|
||
import oss2
|
||
from oss2.credentials import EnvironmentVariableCredentialsProvider
|
||
|
||
from Config.Config import ALY_AK, ALY_SK, ALY_ENDPOINT, ALY_REGION, ALY_BUCKET, ALY_OSS_PREFIX
|
||
|
||
|
||
def upload_to_oss(local_file_path: str, object_name: str) -> bool:
|
||
"""
|
||
将本地文件上传到阿里云OSS
|
||
@param local_file_path: 本地文件的完整路径
|
||
@param object_name: OSS上的对象名称，完整路径中不能包含Bucket名称
|
||
@return: 上传成功返回True，失败返回False
|
||
"""
|
||
try:
|
||
# 设置环境变量获取访问凭证
|
||
os.environ['OSS_ACCESS_KEY_ID'] = ALY_AK
|
||
os.environ['OSS_ACCESS_KEY_SECRET'] = ALY_SK
|
||
auth = oss2.ProviderAuthV4(EnvironmentVariableCredentialsProvider())
|
||
|
||
# 创建Bucket对象
|
||
bucket = oss2.Bucket(auth, ALY_ENDPOINT, ALY_BUCKET, region=ALY_REGION)
|
||
|
||
# 上传文件
|
||
print(f"开始上传文件: {local_file_path} 到 OSS: {object_name}")
|
||
bucket.put_object_from_file(object_name, local_file_path)
|
||
print(f"文件上传成功: {object_name}")
|
||
return True
|
||
except Exception as e:
|
||
print(f"文件上传失败: {str(e)}")
|
||
return False
|
||
|
||
|
||
def get_url(file_name):
|
||
key = f'{ALY_OSS_PREFIX}/{file_name}'
|
||
aly_endpoint = ALY_ENDPOINT.replace("https://", "")
|
||
url = f"https://{ALY_BUCKET}.{aly_endpoint}/{key}"
|
||
return url
|
||
|
||
|
||
# 示例用法
|
||
if __name__ == '__main__':
|
||
# 本地文件的完整路径
|
||
local_file_path = r'/Dsideal/handwriting.jpg'
|
||
|
||
# 填写Object完整路径，完整路径中不能包含Bucket名称
|
||
object_name = 'HuangHai/handwriting.jpg'
|
||
|
||
# 调用函数上传文件
|
||
upload_success = upload_to_oss(local_file_path, object_name)
|
||
if upload_success:
|
||
print("上传操作完成")
|
||
else:
|
||
print("上传操作失败")
|