mirror of
https://github.com/huge-dream/django-vue3-admin.git
synced 2026-09-26 14:23:46 +00:00
57 lines
2.8 KiB
Python
57 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
from rest_framework.exceptions import ValidationError
|
|
|
|
from application import dispatch
|
|
from qcloud_cos import CosConfig
|
|
from qcloud_cos import CosS3Client
|
|
|
|
|
|
# 进度条
|
|
# 当无法确定待上传的数据长度时，total_bytes的值为None。
|
|
def percentage(consumed_bytes, total_bytes):
|
|
if total_bytes:
|
|
rate = int(100 * (float(consumed_bytes) / float(total_bytes)))
|
|
print('\r{0}% '.format(rate), end='')
|
|
|
|
def tencent_cos_upload(file, file_name):
|
|
try:
|
|
file.seek(0)
|
|
file_read = file.read()
|
|
except Exception as e:
|
|
file_read = file
|
|
if not file:
|
|
raise ValidationError('请上传文件')
|
|
# 生成文件名
|
|
path_prefix = dispatch.get_system_config_values("file_storage.tencent_path")
|
|
if not path_prefix.endswith('/'):
|
|
path_prefix = path_prefix + '/'
|
|
if path_prefix.startswith('/'):
|
|
path_prefix = path_prefix[1:]
|
|
base_fil_name = f'{path_prefix}{file_name}'
|
|
# 获取cos配置
|
|
# 1. 设置用户属性, 包括 secret_id, secret_key, region等。Appid 已在 CosConfig 中移除，请在参数 Bucket 中带上 Appid。Bucket 由 BucketName-Appid 组成
|
|
secret_id = dispatch.get_system_config_values("file_storage.tencent_secret_id") # 用户的 SecretId，建议使用子账号密钥，授权遵循最小权限指引，降低使用风险。子账号密钥获取可参见 https://cloud.tencent.com/document/product/598/37140
|
|
secret_key = dispatch.get_system_config_values("file_storage.tencent_secret_key") # 用户的 SecretKey，建议使用子账号密钥，授权遵循最小权限指引，降低使用风险。子账号密钥获取可参见 https://cloud.tencent.com/document/product/598/37140
|
|
region = dispatch.get_system_config_values("file_storage.tencent_region") # 替换为用户的 region，已创建桶归属的 region 可以在控制台查看，https://console.cloud.tencent.com/cos5/bucket # COS 支持的所有 region 列表参见https://cloud.tencent.com/document/product/436/6224
|
|
bucket = dispatch.get_system_config_values("file_storage.tencent_bucket") # 要访问的桶名称
|
|
config = CosConfig(Region=region, SecretId=secret_id, SecretKey=secret_key)
|
|
client = CosS3Client(config)
|
|
# 访问地址
|
|
base_file_url = f'https://{bucket}.cos.{region}.myqcloud.com'
|
|
# 生成外网访问的文件路径
|
|
if base_file_url.endswith('/'):
|
|
file_path = base_file_url + base_fil_name
|
|
else:
|
|
file_path = f'{base_file_url}/{base_fil_name}'
|
|
# 这个是阿里提供的SDK方法 bucket是调用的4.1中配置的变量名
|
|
try:
|
|
response = client.put_object(
|
|
Bucket=bucket,
|
|
Body=file_read,
|
|
Key=base_fil_name,
|
|
EnableMD5=False
|
|
)
|
|
return file_path
|
|
except:
|
|
return None
|