Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

优化ucloud相关

Browse Source
This commit is contained in:
liuyebo
2024-07-12 12:59:23 +08:00
parent eb690ad02c
commit 0f672c7961
2 changed files with 11 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
config/ucloud.py
View File

@@ -8,3 +8,5 @@ BUCKET = "drg100"
UPLOAD_SUFFIX = ".cn-sh2.ufileos.com" UPLOAD_SUFFIX = ".cn-sh2.ufileos.com"
# 下载后缀 # 下载后缀
DOWNLOAD_SUFFIX = ".cn-sh2.ufileos.com" DOWNLOAD_SUFFIX = ".cn-sh2.ufileos.com"
# 私空间文件地址过期时间(秒)
PRIVATE_EXPIRES = 3600

22
ucloud/ucloud.py
View File

@@ -3,14 +3,14 @@ import logging
from ufile import filemanager from ufile import filemanager
from config.ucloud import PUBLIC_KEY, PRIVATE_KEY, UPLOAD_SUFFIX, DOWNLOAD_SUFFIX, BUCKET from config.ucloud import PUBLIC_KEY, PRIVATE_KEY, UPLOAD_SUFFIX, DOWNLOAD_SUFFIX, BUCKET, PRIVATE_EXPIRES
UFILE_HANDLER = filemanager.FileManager(PUBLIC_KEY, PRIVATE_KEY, UPLOAD_SUFFIX, DOWNLOAD_SUFFIX)
def get_private_url(key): def get_private_url(key, bucket=BUCKET):
get_ufile_handler = filemanager.FileManager(PUBLIC_KEY, PRIVATE_KEY, UPLOAD_SUFFIX, DOWNLOAD_SUFFIX)
# 判断文件是否存在 # 判断文件是否存在
_, resp = get_ufile_handler.head_file(BUCKET, key) _, resp = UFILE_HANDLER.head_file(bucket, key)
if resp.status_code != 200: if resp.status_code != 200:
logging.warning(f"uCloud中未找到({key})! status: {resp.status_code} error: {resp.error}") logging.warning(f"uCloud中未找到({key})! status: {resp.status_code} error: {resp.error}")
return None return None
@@ -19,15 +19,13 @@ def get_private_url(key):
# url = get_ufile_handler.public_download_url(bucket, key) # url = get_ufile_handler.public_download_url(bucket, key)
# 获取私有空间下载url, expires为下载链接有效期，单位为秒 # 获取私有空间下载url, expires为下载链接有效期，单位为秒
url = get_ufile_handler.private_download_url(BUCKET, key, expires=3600) url = UFILE_HANDLER.private_download_url(bucket, key, expires=PRIVATE_EXPIRES)
return url return url
def copy_file(source_bucket, source_key, target_bucket, target_key): def copy_file(source_bucket, source_key, target_bucket, target_key):
copy_ufile_handler = filemanager.FileManager(PUBLIC_KEY, PRIVATE_KEY, UPLOAD_SUFFIX, DOWNLOAD_SUFFIX)
# 拷贝文件 # 拷贝文件
ret, resp = copy_ufile_handler.copy(target_bucket, target_key, source_bucket, source_key) ret, resp = UFILE_HANDLER.copy(target_bucket, target_key, source_bucket, source_key)
if resp.status_code != 200: if resp.status_code != 200:
logging.warning( logging.warning(
f"将({source_key})从({source_bucket})拷贝到({target_bucket})失败! status: {resp.status_code} error: {resp.error}") f"将({source_key})从({source_bucket})拷贝到({target_bucket})失败! status: {resp.status_code} error: {resp.error}")
@@ -35,11 +33,9 @@ def copy_file(source_bucket, source_key, target_bucket, target_key):
return True return True
def upload_file(key, file_path): def upload_file(key, file_path, bucket=BUCKET):
putufile_handler = filemanager.FileManager(PUBLIC_KEY, PRIVATE_KEY, UPLOAD_SUFFIX, DOWNLOAD_SUFFIX)
# 普通上传文件至空间 # 普通上传文件至空间
ret, resp = putufile_handler.putfile(BUCKET, key, file_path, header=None) ret, resp = UFILE_HANDLER.putfile(bucket, key, file_path, header=None)
if resp.status_code != 200: if resp.status_code != 200:
logging.warning(f"上传({key})失败! status: {resp.status_code} error: {resp.error}") logging.warning(f"上传({key})失败! status: {resp.status_code} error: {resp.error}")
return False return False