|
- # -*- coding: utf-8 -*-
- from json import loads
- from os.path import join
- from traceback import format_exc
-
- import oss2
- import time
-
- from aliyunsdkvod.request.v20170321.GetPlayInfoRequest import GetPlayInfoRequest
- from loguru import logger
-
- from common.YmlConstant import aliyun_yml_path
- from exception.CustomerException import ServiceException
- from enums.ExceptionEnum import ExceptionType
- from aliyunsdkcore.client import AcsClient
- from aliyunsdkvod.request.v20170321 import GetPlayInfoRequest
-
- from util.RWUtils import getConfigs
- from vodsdk.AliyunVodUploader import AliyunVodUploader
- from vodsdk.UploadVideoRequest import UploadVideoRequest
-
-
- class AliyunOssSdk:
-
- __slots__ = ('bucket', '__request_id', '__config')
-
- def __init__(self, *args):
- base_dir, env, self.__request_id = args
- self.bucket = None
- self.__config = getConfigs(join(base_dir, aliyun_yml_path % env))
- self.get_oss_bucket()
-
- def get_oss_bucket(self):
- if self.bucket is None:
- auth = oss2.Auth(self.__config["access_key"], self.__config["access_secret"])
- self.bucket = oss2.Bucket(auth, self.__config["oss"]["endpoint"],
- self.__config["oss"]["bucket"],
- connect_timeout=self.__config["oss"]["connect_timeout"])
-
- def put_object(self, updatePath, fileByte):
- request_id = self.__request_id
- logger.info("开始上传文件到oss, requestId:{}", request_id)
- max_retries = 3
- retry_count = 0
- while True:
- try:
- self.get_oss_bucket()
- self.bucket.put_object(updatePath, fileByte)
- logger.info("上传文件到oss成功! requestId:{}", request_id)
- break
- except Exception as e:
- retry_count += 1
- time.sleep(1)
- self.bucket = None
- logger.info("上传文件到oss失败, 重试次数：{}, requestId:{}", retry_count, request_id)
- if retry_count > max_retries:
- logger.error("上传文件到oss重试失败:{}, requestId:{}", format_exc(), request_id)
- raise e
-
-
- class ThAliyunVodSdk:
-
- __slots__ = ('__config', '__request_id')
-
- def __init__(self, base_dir, env, request_id):
- self.__request_id = request_id
- self.__config = getConfigs(join(base_dir, aliyun_yml_path % env))
-
- def init_vod_client(self, accessKeyId, accessKeySecret):
- regionId = self.__config["vod"]["ecsRegionId"]
- return AcsClient(accessKeyId, accessKeySecret, regionId, auto_retry=True, max_retry_time=3, timeout=30)
-
- def get_play_info(self, videoId):
- logger.info("开始获取视频地址，videoId:{}, requestId:{}", videoId, self.__request_id)
- clt = self.init_vod_client(self.__config["access_key"], self.__config["access_secret"])
- start = time.time()
- retry_num = 0
- while True:
- try:
- request: GetPlayInfoRequest = GetPlayInfoRequest.GetPlayInfoRequest()
- request.set_accept_format('JSON')
- request.set_VideoId(videoId)
- request.set_AuthTimeout(3600 * 5)
- response = loads(clt.do_action_with_exception(request))
- play_url = response["PlayInfoList"]["PlayInfo"][0]["PlayURL"]
- logger.info("获取视频地址成功，视频地址: {}, requestId: {}", play_url, self.__request_id)
- return play_url
- except Exception as e:
- logger.info("获取视频地址失败，5秒后重试, requestId: {}", self.__request_id)
- retry_num += 1
- time.sleep(5)
- current_time = time.time()
- if "HTTP Status: 403" not in str(e) and retry_num > 12:
- logger.error("获取视频地址失败: {}, requestId: {}", format_exc(), self.__request_id)
- 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) or
- "ProduceFail" in str(e)):
- logger.error("获取视频地址失败: {}, requestId: {}", format_exc(), self.__request_id)
- 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 * 5:
- logger.error("获取视频地址失败超时异常: {}，超时时间：{}, requestId: {}", format_exc(),
- diff_time, self.__request_id)
- 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):
- logger.info("开始执行vod视频上传, filePath: {}, requestId: {}", filePath, self.__request_id)
- uploader = AliyunVodUploader(self.__config["access_key"], self.__config["access_secret"], self.__request_id)
- uploadVideoRequest: UploadVideoRequest = UploadVideoRequest(filePath, file_title)
- logger.info("视频分类:{}, requestId:{}", self.__config["vod"]["cateId"], self.__request_id)
- uploadVideoRequest.setCateId(self.__config["vod"]["cateId"])
- # 可以设置视频封面，如果是本地或网络图片可使用UploadImageRequest上传图片到视频点播，获取到ImageURL
- # ImageURL示例：https://example.com/sample-****.jpg
- # uploadVideoRequest.setCoverURL('<your Image URL>')
- # 标签
- # uploadVideoRequest.setTags('tag1,tag2')
- MAX_RETRIES = 3
- retry_count = 0
- while True:
- try:
- result = uploader.uploadLocalVideo(uploadVideoRequest)
- logger.info("vod视频上传成功, videoId:{}, requestId:{}", result.get("VideoId"), self.__request_id)
- return result.get("VideoId")
- except Exception:
- retry_count += 1
- time.sleep(1)
- uploader = AliyunVodUploader(self.__config["access_key"], self.__config["access_secret"],
- self.__request_id)
- uploadVideoRequest: UploadVideoRequest = UploadVideoRequest(filePath, file_title)
- uploadVideoRequest.setCateId(self.__config["vod"]["cateId"])
- logger.error("vod视频上传失败:{}，重试次数：{}, requestId:{}", format_exc(), retry_count,
- self.__request_id)
- if retry_count >= MAX_RETRIES:
- raise ServiceException(ExceptionType.SERVICE_INNER_EXCEPTION.value[0],
- ExceptionType.SERVICE_INNER_EXCEPTION.value[1])
-
- def get_play_url(self, filePath, file_title):
- videoId = self.upload_local_video(filePath, file_title)
- if videoId is None or len(videoId) == 0:
- raise ServiceException(ExceptionType.GET_VIDEO_URL_EXCEPTION.value[0],
- ExceptionType.GET_VIDEO_URL_EXCEPTION.value[1])
- return self.get_play_info(videoId)
-
- # if __name__ == "__main__":
- # aa = ThAliyunVodSdk('/home/th/tuo_heng/dev/tuoheng_alg', 'dev', "1")
- # print(aa.get_play_info('6928821035b171ee9f3b6632b68f0102'))
|
