|
|
|
|
|
# https://help.aliyun.com/zh/oss/use-cases/add-signatures-on-the-client-by-using-javascript-and-upload-data-to-oss?spm=a2c4g.11186623.help-menu-31815.d_6_1_0_1.2dd15d03SrLg4Q#4f036801celh7
|
|
|
|
|
|
import hashlib
|
|
|
|
|
|
import requests
|
|
|
|
|
|
import time
|
|
|
|
|
|
import hmac
|
|
|
|
|
|
import base64
|
|
|
|
|
|
import urllib.parse
|
|
|
|
|
|
from hashlib import sha1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import oss2
|
|
|
|
|
|
|
|
|
|
|
|
from WxMini.Milvus.Config.MulvusConfig import *
|
|
|
|
|
|
|
|
|
|
|
|
# 初始化 OSS Bucket
|
|
|
|
|
|
auth = oss2.Auth(ACCESS_KEY_ID, ACCESS_KEY_SECRET)
|
|
|
|
|
|
bucket = oss2.Bucket(auth, ENDPOINT, BUCKET_NAME)
|
|
|
|
|
|
|
|
|
|
|
|
def upload_mp3_to_oss(file_path, oss_object_name):
|
|
|
|
|
|
"""
|
|
|
|
|
|
上传本地 MP3 文件到 OSS
|
|
|
|
|
|
:param file_path: 本地 MP3 文件路径
|
|
|
|
|
|
:param oss_object_name: OSS 中存储的文件名
|
|
|
|
|
|
"""
|
|
|
|
|
|
try:
|
|
|
|
|
|
# 上传文件
|
|
|
|
|
|
with open(file_path, 'rb') as file:
|
|
|
|
|
|
bucket.put_object(oss_object_name, file)
|
|
|
|
|
|
print(f"文件 {file_path} 已成功上传到 OSS，存储为 {oss_object_name}")
|
|
|
|
|
|
except Exception as e:
|
|
|
|
|
|
print(f"上传失败: {e}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def upload_mp3_to_oss_from_memory(oss_object_name, audio_data):
|
|
|
|
|
|
"""
|
|
|
|
|
|
从内存上传 MP3 数据到 OSS
|
|
|
|
|
|
:param oss_object_name: OSS 中存储的文件名
|
|
|
|
|
|
:param audio_data: 音频数据（bytes 类型）
|
|
|
|
|
|
"""
|
|
|
|
|
|
try:
|
|
|
|
|
|
# 上传音频数据
|
|
|
|
|
|
bucket.put_object(oss_object_name, audio_data)
|
|
|
|
|
|
print(f"音频数据已成功上传到 OSS，存储为 {oss_object_name}")
|
|
|
|
|
|
except Exception as e:
|
|
|
|
|
|
print(f"上传失败: {e}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def hmacsha256(key, data):
|
|
|
|
|
|
"""
|
|
|
|
|
|
计算HMAC-SHA256哈希值的函数
|
|
|
|
|
|
|
|
|
|
|
|
:param key: 用于计算哈希的密钥，字节类型
|
|
|
|
|
|
:param data: 要进行哈希计算的数据，字符串类型
|
|
|
|
|
|
:return: 计算得到的HMAC-SHA256哈希值，字节类型
|
|
|
|
|
|
"""
|
|
|
|
|
|
try:
|
|
|
|
|
|
mac = hmac.new(key, data.encode(), hashlib.sha256)
|
|
|
|
|
|
hmacBytes = mac.digest()
|
|
|
|
|
|
return hmacBytes
|
|
|
|
|
|
except Exception as e:
|
|
|
|
|
|
raise RuntimeError(f"Failed to calculate HMAC-SHA256 due to {e}")
|
|
|
|
|
|
|
|
|
|
|
|
def upload_to_oss(access_key_id, access_key_secret, security_token, endpoint=ENDPOINT, bucket_name=BUCKET_NAME, file_key="Upload/example.txt", file_content="Hello, OSS!"):
|
|
|
|
|
|
"""
|
|
|
|
|
|
上传文件到 OSS
|
|
|
|
|
|
:param access_key_id: 临时 AccessKey ID
|
|
|
|
|
|
:param access_key_secret: 临时 AccessKey Secret
|
|
|
|
|
|
:param security_token: 临时 SecurityToken
|
|
|
|
|
|
:param endpoint: OSS Endpoint (默认值: ENDPOINT)
|
|
|
|
|
|
:param bucket_name: OSS Bucket 名称 (默认值: BUCKET_NAME)
|
|
|
|
|
|
:param file_key: 文件在 OSS 中的 Key (默认值: "Upload/example.txt")
|
|
|
|
|
|
:param file_content: 文件内容 (默认值: "Hello, OSS!")
|
|
|
|
|
|
:return: 上传结果（成功或失败）
|
|
|
|
|
|
"""
|
|
|
|
|
|
# 使用临时凭证初始化 OSS 客户端
|
|
|
|
|
|
auth = oss2.StsAuth(access_key_id, access_key_secret, security_token)
|
|
|
|
|
|
bucket = oss2.Bucket(auth, endpoint, bucket_name)
|
|
|
|
|
|
|
|
|
|
|
|
# 上传文件
|
|
|
|
|
|
try:
|
|
|
|
|
|
bucket.put_object(file_key, file_content)
|
|
|
|
|
|
return "文件上传成功"
|
|
|
|
|
|
except oss2.exceptions.AccessDenied as e:
|
|
|
|
|
|
return f"文件上传失败：{e}"
|
