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
|
|||
|
|
|