hzhang/MoneyPrinterTurbo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed: issue 183

Browse Source
This commit is contained in:
harry
2024-04-07 18:08:51 +08:00
parent c10b5696fc
commit b8103e98f1
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
app/controllers/v1/video.py
View File

@@ -48,21 +48,27 @@ def get_task(request: Request, task_id: str = Path(..., description="Task ID"),
request_id = base.get_task_id(request) request_id = base.get_task_id(request)
task = sm.get_task(task_id) task = sm.get_task(task_id)
if task: if task:
task_dir = utils.task_dir()
def file_to_uri(file):
if not file.startswith(endpoint):
_uri_path = v.replace(task_dir, "tasks").replace("\\", "/")
_uri_path = f"{endpoint}/{_uri_path}"
else:
_uri_path = file
return _uri_path
if "videos" in task: if "videos" in task:
videos = task["videos"] videos = task["videos"]
task_dir = utils.task_dir()
urls = [] urls = []
for v in videos: for v in videos:
uri_path = v.replace(task_dir, "tasks") urls.append(file_to_uri(v))
urls.append(f"{endpoint}/{uri_path}")
task["videos"] = urls task["videos"] = urls
if "combined_videos" in task: if "combined_videos" in task:
combined_videos = task["combined_videos"] combined_videos = task["combined_videos"]
task_dir = utils.task_dir()
urls = [] urls = []
for v in combined_videos: for v in combined_videos:
uri_path = v.replace(task_dir, "tasks") urls.append(file_to_uri(v))
urls.append(f"{endpoint}/{uri_path}")
task["combined_videos"] = urls task["combined_videos"] = urls
return utils.get_response(200, task) return utils.get_response(200, task)