This commit is contained in:
2024-12-09 09:25:58 +08:00
parent 8d6159d302
commit 515bf156ab
6 changed files with 69 additions and 7 deletions

View File

@ -43,7 +43,6 @@ def get_template_info(template_id):
try:
response = session.post('{0}/template/{1}'.format(os.getenv('API_ENDPOINT'), template_id), json={
'accessKey': os.getenv('ACCESS_KEY'),
'clientStatus': util.system.get_sys_info()
}, timeout=10)
except requests.RequestException as e:
logger.error("请求失败!", e)
@ -96,3 +95,42 @@ def get_template_info(template_id):
template['video_parts'].append(parts)
template['local_path'] = os.path.join(os.getenv('TEMPLATE_DIR'), str(template_id))
return template
def report_task_success(task_info):
try:
response = session.post('{0}/{1}/success'.format(os.getenv('API_ENDPOINT'), task_info.get("id")), json={
'accessKey': os.getenv('ACCESS_KEY'),
}, timeout=10)
except requests.RequestException as e:
logger.error("请求失败!", e)
return None
def report_task_failed(task_info):
try:
response = session.post('{0}/{1}/fail'.format(os.getenv('API_ENDPOINT'), task_info.get("id")), json={
'accessKey': os.getenv('ACCESS_KEY'),
}, timeout=10)
except requests.RequestException as e:
logger.error("请求失败!", e)
return None
def upload_task_file(task_info, ffmpeg_task):
try:
response = session.post('{0}/{1}/uploadUrl'.format(os.getenv('API_ENDPOINT'), task_info.get("id")), json={
'accessKey': os.getenv('ACCESS_KEY'),
}, timeout=10)
except requests.RequestException as e:
logger.error("请求失败!", e)
return False
data = response.json()
url = data.get('data', "")
try:
with open(ffmpeg_task.get_output_file(), 'rb') as f:
requests.put(url, data=f)
except requests.RequestException as e:
logger.error("上传失败!", e)
return False
return True