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

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

2
.env
View File

@ -5,4 +5,4 @@ TEMP_DIR=tmp/
# QSV
ENCODER_ARGS="-c:v h264_qsv -global_quality 28 -look_ahead 1"
# NVENC
#ENCODER_ARGS="-c:v h264_nvenc -rc cbr_ld_hq -b:v 3000k -preset p7 -profile high -bf 0 -bf-strategy 0 -rc-lookahead 0 -qp 28 -spatial-aq 1 -temporal-aq 1 -aq-strength 8 -cq 28"
#ENCODER_ARGS="-c:v h264_nvenc -cq:v 24 -preset:v p7 -tune:v hq -profile:v high -bf 0 -bf-strategy 0 -rc-lookahead 0 -qp 28 -spatial-aq 1 -temporal-aq 1 -aq-strength 8 -cq 28"

23
app.py Normal file
View File

@ -0,0 +1,23 @@
import flask
import config
import biz.task
from telemetry import init_opentelemetry
from template import load_local_template
from util import api
load_local_template()
import logging
LOGGER = logging.getLogger(__name__)
init_opentelemetry()
app = flask.Flask(__name__)
@app.post('/:id')
def do_task(id):
task_info = api.get_task_info(id)
biz.task.start_task(task_info)
if __name__ == '__main__':
app.run(host="0.0.0.0", port=9998)

View File

@ -4,3 +4,4 @@ python-dotenv~=1.0.1
opentelemetry-api~=1.30.0
opentelemetry-sdk~=1.30.0
opentelemetry-exporter-otlp~=1.30.0
flask~=3.1.0

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', {})