支持redirection

This commit is contained in:
Jerry Yan 2025-04-20 14:32:06 +08:00
parent ec03f8180e
commit 9537f995a1
2 changed files with 9 additions and 0 deletions

1
.env
View File

@ -2,6 +2,7 @@ TEMPLATE_DIR=template/
API_ENDPOINT=https://zhentuai.com/task/v1
ACCESS_KEY=TEST_ACCESS_KEY
TEMP_DIR=tmp/
#REDIRECT_TO_URL=https://worker-renderworker-re-kekuflqjxx.cn-shanghai.fcapp.run/
# QSV
ENCODER_ARGS="-c:v h264_qsv -global_quality 28 -look_ahead 1"
# NVENC

View File

@ -1,6 +1,7 @@
import json
import logging
import os
import threading
import requests
from opentelemetry.trace import Status, StatusCode
@ -44,6 +45,13 @@ def sync_center():
tasks = []
templates = []
logger.warning("获取任务失败")
if os.getenv("REDIRECT_TO_URL", False) != False:
for task in tasks:
_sess = requests.Session()
logger.info("重定向任务【%s】至配置的地址:%s", task.get("id"), os.getenv("REDIRECT_TO_URL"))
url = f"{os.getenv('REDIRECT_TO_URL')}{task.get('id')}"
threading.Thread(target=requests.post, args=(url,)).start()
return []
for template in templates:
template_id = template.get('id', '')
if template_id: