|
- import oss2
- import time
- from exception.CustomerException import ServiceException
- from enums.ExceptionEnum import ExceptionType
- import json
- from aliyunsdkcore.client import AcsClient
- from aliyunsdkvod.request.v20170321 import GetPlayInfoRequest
- from voduploadsdk.AliyunVodUtils import *
- from voduploadsdk.AliyunVodUploader import AliyunVodUploader
- from voduploadsdk.UploadVideoRequest import UploadVideoRequest
-
-
- class AliyunOssSdk():
-
- def __init__(self, content, logger, requestId):
- self.content = content
- self.bucket = None
- self.logger = logger
- self.requestId = requestId
-
- def get_oss_bucket(self):
- if self.bucket is None:
- self.logger.info("初始化oss桶, requestId:{}", self.requestId)
- auth = oss2.Auth(self.content["aliyun"]["access_key"], self.content["aliyun"]["access_secret"])
- self.bucket = oss2.Bucket(auth, self.content["aliyun"]["oss"]["endpoint"],
- self.content["aliyun"]["oss"]["bucket"],
- connect_timeout=self.content["aliyun"]["oss"]["connect_timeout"])
-
- def sync_upload_file(self, updatePath, fileByte):
- self.logger.info("开始上传文件到oss, requestId:{}", self.requestId)
- self.get_oss_bucket()
- MAX_RETRIES = 3
- retry_count = 0
- while True:
- try:
- self.bucket.put_object(updatePath, fileByte)
- self.logger.info("上传文件到oss成功! requestId:{}", self.requestId)
- break
- except Exception as e:
- retry_count += 1
- time.sleep(1)
- self.logger.info("上传文件到oss失败, 重试次数：{}, requestId:{}", retry_count, self.requestId)
- if retry_count > MAX_RETRIES:
- self.logger.exception("上传文件到oss重试失败:{}, requestId:{}", e, self.requestId)
- raise e
-
-
- async def put_object(self, updatePath, fileByte):
- self.bucket.put_object(updatePath, fileByte)
-
- async def upload_file(self, updatePath, fileByte):
- self.logger.info("开始上传文件到oss, requestId:{}", self.requestId)
- self.get_oss_bucket()
- MAX_RETRIES = 3
- retry_count = 0
- while True:
- try:
- await self.put_object(updatePath, fileByte)
- self.logger.info("上传文件到oss成功! requestId:{}", self.requestId)
- break
- except Exception as e:
- retry_count += 1
- time.sleep(1)
- self.logger.info("上传文件到oss失败, 重试次数：{}, requestId:{}", retry_count, self.requestId)
- if retry_count > MAX_RETRIES:
- self.logger.exception("上传文件到oss重试失败:{}, requestId:{}", e, self.requestId)
- raise e
-
-
- class ThAliyunVodSdk():
-
- def __init__(self, content, logger, requestId):
- self.content = content
- self.logger = logger
- self.requestId = requestId
-
- def init_vod_client(self, accessKeyId, accessKeySecret):
- regionId = self.content["aliyun"]["vod"]["ecsRegionId"]
- return AcsClient(accessKeyId, accessKeySecret, regionId, auto_retry=True, max_retry_time=3, timeout=5)
-
- def get_play_info(self, videoId):
- self.logger.info("开始获取视频地址，videoId:{}, requestId:{}", videoId, self.requestId)
- start = time.time()
- while True:
- try:
- clt = self.init_vod_client(self.content["aliyun"]["access_key"],
- self.content["aliyun"]["access_secret"])
- request = GetPlayInfoRequest.GetPlayInfoRequest()
- request.set_accept_format('JSON')
- request.set_VideoId(videoId)
- request.set_AuthTimeout(3600 * 5)
- response = json.loads(clt.do_action_with_exception(request))
- play_url = response["PlayInfoList"]["PlayInfo"][0]["PlayURL"]
- self.logger.info("获取视频地址成功，视频地址: {}, requestId: {}", play_url, self.requestId)
- return play_url
- except Exception as e:
- self.logger.error("获取视频地址失败，5秒后重试, requestId: {}", self.requestId)
- time.sleep(5)
- current_time = time.time()
- if "HTTP Status: 403" not in str(e):
- self.logger.exception("获取视频地址失败: {}, requestId: {}", e, self.requestId)
- raise ServiceException(ExceptionType.GET_VIDEO_URL_EXCEPTION.value[0],
- ExceptionType.GET_VIDEO_URL_EXCEPTION.value[1])
- if "HTTP Status: 403" in str(e) and ("UploadFail" in str(e) or "TranscodeFail" in str(e)):
- self.logger.exception("获取视频地址失败: {}, requestId: {}", e, self.requestId)
- raise ServiceException(ExceptionType.GET_VIDEO_URL_EXCEPTION.value[0],
- ExceptionType.GET_VIDEO_URL_EXCEPTION.value[1])
- diff_time = current_time - start
- if diff_time > 60 * 60 * 2:
- self.logger.exception("获取视频地址失败超时异常: {}，超时时间：{}, requestId: {}", e, diff_time, self.requestId)
- raise ServiceException(ExceptionType.GET_VIDEO_URL_TIMEOUT_EXCEPTION.value[0],
- ExceptionType.GET_VIDEO_URL_TIMEOUT_EXCEPTION.value[1])
-
- def upload_local_video(self, filePath, file_title, storageLocation=None):
- self.logger.info("开始执行vod视频上传, filePath: {}, requestId: {}", filePath, self.requestId)
- uploader = AliyunVodUploader(self.content["aliyun"]["access_key"], self.content["aliyun"]["access_secret"])
- uploadVideoRequest = UploadVideoRequest(filePath, file_title)
- self.logger.info("视频分类:{}", self.content["aliyun"]["vod"][self.content["dsp"]["active"]]["CateId"])
- uploadVideoRequest.setCateId(self.content["aliyun"]["vod"][self.content["dsp"]["active"]]["CateId"])
- # 可以设置视频封面，如果是本地或网络图片可使用UploadImageRequest上传图片到视频点播，获取到ImageURL
- # ImageURL示例：https://example.com/sample-****.jpg
- # uploadVideoRequest.setCoverURL('<your Image URL>')
- # 标签
- # uploadVideoRequest.setTags('tag1,tag2')
- if storageLocation:
- uploadVideoRequest.setStorageLocation(storageLocation)
- MAX_RETRIES = 3
- retry_count = 0
- while True:
- try:
- result = uploader.uploadLocalVideo(uploadVideoRequest)
- self.logger.info("vod视频上传成功, videoId:{}, requestId:{}", result.get("VideoId"), self.requestId)
- return result.get("VideoId")
- except AliyunVodException as e:
- retry_count += 1
- time.sleep(3)
- self.logger.error("vod视频上传失败，重试次数：{}, requestId:{}", retry_count, self.requestId)
- if retry_count >= MAX_RETRIES:
- self.logger.exception("vod视频上传重试失败: {}, requestId:{}", e, self.requestId)
- raise ServiceException(ExceptionType.SERVICE_INNER_EXCEPTION.value[0],
- ExceptionType.SERVICE_INNER_EXCEPTION.value[1])
-
-
- def get_play_url(args):
- thAliyunVodSdk = ThAliyunVodSdk(args[2], args[3], args[4])
- videoId = thAliyunVodSdk.upload_local_video(args[0], args[1])
- if videoId is None or len(videoId) == 0:
- return None
- return thAliyunVodSdk.get_play_info(videoId)
|
