添加接口,添加方便测试的方法

This commit is contained in:
2025-04-20 11:50:29 +08:00
parent a9043361ec
commit 3d810e5c5b
4 changed files with 41 additions and 2 deletions

View File

@ -231,3 +231,18 @@ def upload_task_file(task_info, ffmpeg_task):
url = data.get('data', "")
logger.info("开始上传文件: %s%s", task_info.get("id"), url)
return oss.upload_to_oss(url, ffmpeg_task.get_output_file())
def get_task_info(id):
try:
response = session.get(os.getenv('API_ENDPOINT') + "/" + id + "/info", params={
'accessKey': os.getenv('ACCESS_KEY'),
}, timeout=10)
response.raise_for_status()
except requests.RequestException as e:
logger.error("请求失败!", e)
return []
data = response.json()
logger.debug("获取任务结果:【%s", data)
if data.get('code', 0) == 200:
return data.get('data', {})