60 lines
2.5 KiB
Python
60 lines
2.5 KiB
Python
import sys
|
|
import os
|
|
import uuid
|
|
import requests
|
|
# 添加项目根目录到Python路径
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
|
from Util.ObsUtil import ObsUploader
|
|
|
|
def main():
|
|
# 测试视频URL - 使用用户提供的可播放视频URL
|
|
test_video_url = "https://v3-vvecloud.yangyi08.com/3e4dec4994152917baa2f14302927595/68a69142/video/tos/cn/tos-cn-v-242bcc/oEiAEhzf0xyEw7FaNgU0QAiGSAZBOCQgKm1tmA/?a=7073&ch=0&cr=0&dr=0&er=0&lr=default&cd=0%7C0%7C0%7C0&br=4542&bt=4542&cs=0&ds=4&ft=xztlUQhhe6BMyqjL_jkJD12Nzj&mime_type=video_mp4&qs=13&rc=M2w1Nm85cjk7NTgzNGczM0BpM2w1Nm85cjk7NTgzNGczM0A2YWo1MmRzM2dhLS1kXjBzYSM2YWo1MmRzM2dhLS1kXjBzcw%3D%3D&btag=80000e00008000&dy_q=1755743021&l=202508211023414B2120DB0DDDEF74FB7F"
|
|
|
|
# 使用UUID生成唯一文件名
|
|
unique_id = uuid.uuid4()
|
|
test_object_key = f"HuangHai/JiMeng/{unique_id}.mp4"
|
|
|
|
# 临时文件保存路径
|
|
temp_file_path = os.path.join(os.path.dirname(__file__), f"{unique_id}.mp4")
|
|
|
|
try:
|
|
# 1. 下载视频到本地临时文件
|
|
print(f"开始下载视频: {test_video_url}")
|
|
with requests.get(test_video_url, stream=True, timeout=3600) as r:
|
|
r.raise_for_status()
|
|
with open(temp_file_path, 'wb') as f:
|
|
for chunk in r.iter_content(chunk_size=8192):
|
|
f.write(chunk)
|
|
print(f"视频下载完成，保存至: {temp_file_path}")
|
|
|
|
# 2. 实例化ObsUploader
|
|
obs_uploader = ObsUploader()
|
|
|
|
print(f"开始上传文件到OBS: {temp_file_path}\n对象键: {test_object_key}")
|
|
|
|
# 3. 调用文件上传方法
|
|
result = obs_uploader.upload_file(
|
|
file_path=temp_file_path,
|
|
object_key=test_object_key
|
|
)
|
|
|
|
# 4. 打印结果
|
|
if hasattr(result, 'get') and result.get('success'):
|
|
print(f"上传成功! 视频地址: {result.get('url')}")
|
|
else:
|
|
print(f"上传失败: {str(result)}")
|
|
|
|
except Exception as e:
|
|
print(f"测试过程中发生异常: {str(e)}")
|
|
finally:
|
|
# 5. 清理临时文件
|
|
if os.path.exists(temp_file_path):
|
|
os.remove(temp_file_path)
|
|
print(f"临时文件已删除: {temp_file_path}")
|
|
|
|
def test_upload_file():
|
|
# 单元测试函数，可用于集成到测试框架
|
|
main()
|
|
|
|
if __name__ == "__main__":
|
|
main() |