120 lines
5.1 KiB
Python
120 lines
5.1 KiB
Python
# -*- coding: utf-8 -*-
|
||
import datetime
|
||
|
||
import cv2
|
||
import oss2
|
||
import time
|
||
|
||
from loguru import logger
|
||
|
||
'''
|
||
图片上传使用OSS
|
||
1. 阿里云对象存储OSS官网地址：https://help.aliyun.com/product/31815.html?spm=a2c4g.32006.0.0.8c546cf0BpkAQ2
|
||
2. 阿里云对象存储OSS SDK示例地址：https://help.aliyun.com/document_detail/32006.html?spm=a2c4g.32006.0.0.66874b78q1pwLa
|
||
3. python安装SDK地址： https://help.aliyun.com/document_detail/85288.html?spm=a2c4g.32026.0.0.3f24417coCphWj
|
||
4. 安装SDK： pip install oss2
|
||
5. 安装python-devel
|
||
安装python-devel
|
||
由于SDK需要crcmod库计算CRC校验码，而crcmod依赖Python.h文件，如果系统缺少这个头文件，安装SDK不会失败，但crcmod的C扩展模式安装会失败，因此导致上传、下载等操作效率非常低下。
|
||
如果python-devel包不存在，则首先要安装这个包。
|
||
对于Windows系统和Mac OS X系统，由于安装Python的时候会将Python依赖的头文件一并安装，因此您无需安装python-devel。
|
||
对于CentOS、RHEL、Fedora系统，请执行以下命令安装python-devel。
|
||
sudo yum install python-devel
|
||
对于Debian，Ubuntu系统，请执行以下命令安装python-devel。
|
||
sudo apt-get install python-dev
|
||
6、图片域名地址：https://image.t-aaron.com/
|
||
'''
|
||
|
||
|
||
class AliyunOssSdk:
|
||
|
||
def __init__(self):
|
||
self.__client = None
|
||
self.__access_key = 'LTAI5tMiefafZ6br4zmrQWv9'
|
||
self.__access_secret = 'JgzQjSCkwZ7lefZO6egOArw38YH1Tk'
|
||
self.__endpoint = 'http://oss-cn-shanghai.aliyuncs.com'
|
||
self.__bucket = 'ta-tech-image'
|
||
|
||
def get_oss_bucket(self):
|
||
if not self.__client:
|
||
auth = oss2.Auth(self.__access_key, self.__access_secret)
|
||
self.__client = oss2.Bucket(auth, self.__endpoint, self.__bucket, connect_timeout=30)
|
||
|
||
def upload_file(self, updatePath, fileByte):
|
||
logger.info("开始上传文件到oss!")
|
||
MAX_RETRIES = 3
|
||
retry_count = 0
|
||
while True:
|
||
try:
|
||
self.get_oss_bucket()
|
||
result = self.__client.put_object(updatePath, fileByte)
|
||
return result
|
||
logger.info("上传文件到oss成功!")
|
||
break
|
||
except Exception as e:
|
||
self.__client = None
|
||
retry_count += 1
|
||
time.sleep(1)
|
||
logger.info("上传文件到oss失败, 重试次数：{}", retry_count)
|
||
if retry_count > MAX_RETRIES:
|
||
logger.exception("上传文件到oss重试失败:{}", e)
|
||
raise e
|
||
|
||
|
||
YY_MM_DD_HH_MM_SS = "%Y-%m-%d %H:%M:%S"
|
||
YMDHMSF = "%Y%m%d%H%M%S%f"
|
||
|
||
def generate_timestamp():
|
||
"""根据当前时间获取时间戳，返回整数"""
|
||
return int(time.time())
|
||
|
||
def now_date_to_str(fmt=None):
|
||
if fmt is None:
|
||
fmt = YY_MM_DD_HH_MM_SS
|
||
return datetime.datetime.now().strftime(fmt)
|
||
|
||
if __name__ == "__main__":
|
||
# 初始化oss对象
|
||
ossClient = AliyunOssSdk()
|
||
# 读取本地图片
|
||
image_frame = cv2.imread('aaa.jpeg')
|
||
or_result, or_image = cv2.imencode(".jpg", image_frame)
|
||
# 图片名称命名规则
|
||
# 1、base_dir 基本文件夹名称，由拓恒公司传参
|
||
# 2、time_now 现在的时间
|
||
# 3、current_frame 当前视频的帧数
|
||
# 4、last_frame 如果有跳帧操作, 填写跳帧的步长，如果没有，和current_frame参数保持一致
|
||
# 5、random_num 随机时间字符串
|
||
# 6、mode_type 类型：实时视频直播的方式用（online） 离线视频直播（填写视频地址识别）用（offline）
|
||
# 7、requestId 请求id, 拓恒公司传参
|
||
# 8、image_type 原图用（OR） AI识别后的图片用（AI）
|
||
random_num = now_date_to_str(YMDHMSF)
|
||
time_now = now_date_to_str("%Y-%m-%d-%H-%M-%S")
|
||
image_format = "{base_dir}/{time_now}_frame-{current_frame}-{last_frame}_type_{random_num}-{mode_type}-{base_dir}" \
|
||
"-{requestId}_{image_type}.jpg"
|
||
image_name = image_format.format(
|
||
base_dir='PWL202304141639429276',
|
||
time_now=time_now,
|
||
current_frame='0',
|
||
last_frame='0',
|
||
random_num=random_num,
|
||
mode_type='offline',
|
||
requestId='111111111111111111',
|
||
image_type='OR')
|
||
result = ossClient.upload_file(image_name, or_image.tobytes())
|
||
# print('http status: {0}'.format(result.status))
|
||
# # 请求ID。请求ID是本次请求的唯一标识，强烈建议在程序日志中添加此参数。
|
||
# print('request_id: {0}'.format(result.request_id))
|
||
# # ETag是put_object方法返回值特有的属性，用于标识一个Object的内容。
|
||
# print('ETag: {0}'.format(result.etag))
|
||
# # HTTP响应头部。
|
||
# print('date: {0}'.format(result.headers['date']))
|
||
# print(result.__reduce__())
|
||
# 对于图片上传, 上传成功后，直接将image_name给拓恒公司就可以了
|
||
# 如果测试查看图片是否上传成功
|
||
# 可以使用域名拼接
|
||
image_url = 'https://image.t-aaron.com/' + image_name
|
||
print(image_url)
|
||
# 拓恒公司只需要image_name
|
||
|