87 lines
3.6 KiB
Python
87 lines
3.6 KiB
Python
# -*- coding=utf-8
|
||
from qcloud_cos import CosConfig
|
||
from qcloud_cos import CosS3Client
|
||
import sys
|
||
import os
|
||
import logging
|
||
|
||
# 正常情况日志级别使用 INFO，需要定位时可以修改为 DEBUG，此时 SDK 会打印和服务端的通信信息
|
||
# logging.basicConfig(level=logging.INFO, stream=sys.stdout)
|
||
|
||
g_cos_root = 'https://xinglu-1324629296.cos.ap-beijing.myqcloud.com'
|
||
g_cos_bucket = 'xinglu-1324629296'
|
||
|
||
def get_client():
|
||
# 1. 设置用户属性, 包括 secret_id, secret_key, region等。Appid 已在 CosConfig 中移除，请在参数 Bucket 中带上 Appid。Bucket 由 BucketName-Appid 组成
|
||
# secret_id = os.environ['COS_SECRET_ID'] # 用户的 SecretId，建议使用子账号密钥，授权遵循最小权限指引，降低使用风险。子账号密钥获取可参见 https://cloud.tencent.com/document/product/598/37140
|
||
# secret_key = os.environ['COS_SECRET_KEY'] # 用户的 SecretKey，建议使用子账号密钥，授权遵循最小权限指引，降低使用风险。子账号密钥获取可参见 https://cloud.tencent.com/document/product/598/37140
|
||
"""
|
||
SecretId:AKIDFLFvOfjbFi171mLnEXp370rHZozymN2c
|
||
SecretKey:a6RYUDl6EAtQmurb8To4AFRCoDOM6g3D
|
||
"""
|
||
secret_id = 'AKIDFLFvOfjbFi171mLnEXp370rHZozymN2c'
|
||
secret_key = 'a6RYUDl6EAtQmurb8To4AFRCoDOM6g3D'
|
||
region = 'ap-beijing' # 替换为用户的 region，已创建桶归属的 region 可以在控制台查看，https://console.cloud.tencent.com/cos5/bucket
|
||
# COS 支持的所有 region 列表参见https://cloud.tencent.com/document/product/436/6224
|
||
token = None # 如果使用永久密钥不需要填入 token，如果使用临时密钥需要填入，临时密钥生成和使用指引参见 https://cloud.tencent.com/document/product/436/14048
|
||
scheme = 'https' # 指定使用 http/https 协议来访问 COS，默认为 https，可不填
|
||
|
||
config = CosConfig(Region=region, SecretId=secret_id, SecretKey=secret_key, Token=token, Scheme=scheme)
|
||
client = CosS3Client(config)
|
||
return client
|
||
|
||
|
||
def upload_file_to_cos(local_filename: str, cos_path: str):
|
||
client = get_client()
|
||
# 本地路径 简单上传
|
||
response = client.put_object_from_local_file(
|
||
Bucket=g_cos_bucket,
|
||
LocalFilePath=local_filename,
|
||
Key=cos_path
|
||
)
|
||
print(response['ETag'])
|
||
return response
|
||
|
||
|
||
def get_file_from_cos(cos_path: str, local_filename: str):
|
||
"""
|
||
从cos上下载文件
|
||
:param cos_path:
|
||
:param local_filename:
|
||
:return:
|
||
"""
|
||
client = get_client()
|
||
response = client.get_object(
|
||
Bucket=g_cos_bucket,
|
||
Key=cos_path
|
||
)
|
||
response['Body'].get_stream_to_file(local_filename)
|
||
|
||
|
||
|
||
def upload_order_images_to_cos(idx_to_localfile: dict, orderid: str):
|
||
"""
|
||
将一组本地图片，上传到对应的cos的订单目录下
|
||
:param idx_to_localfile:
|
||
:param orderid:
|
||
:return: idx_to_cosfile
|
||
"""
|
||
xlsyn_cos_host = 'https://xinglu-1324629296.cos.ap-beijing.myqcloud.com'
|
||
cos_order_path = '/user/wave_survey/%s' % orderid
|
||
idx_to_cosfile = {}
|
||
for idx, local_filepath in idx_to_localfile.items():
|
||
image_filename = local_filepath.split('/')[-1]
|
||
cos_path = '%s/%s' % (cos_order_path, image_filename)
|
||
upload_file_to_cos(local_filepath, cos_path)
|
||
cos_file = '%s/%s' % (xlsyn_cos_host, cos_path)
|
||
idx_to_cosfile[idx] = cos_file
|
||
return idx_to_cosfile
|
||
|
||
|
||
if __name__ == '__main__':
|
||
idx_to_localfile = {0:'D:/slgwork/slgcode/wave_survey/1739430848000.jpg',
|
||
2:'D:/slgwork/slgcode/wave_survey/1739430851000.jpg'}
|
||
idx_to_cosfile = upload_order_images_to_cos(idx_to_localfile, 'slg_orderid')
|
||
print(idx_to_cosfile)
|
||
|