vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix download file extension (#1157)

Browse Source
This commit is contained in:
LawyZheng
2024-11-08 01:09:01 +08:00
committed by GitHub
parent 2252148c5a
commit 45477ab327
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
skyvern/forge/agent.py
View File

@@ -5,6 +5,7 @@ import random
import string import string
from asyncio.exceptions import CancelledError from asyncio.exceptions import CancelledError
from datetime import datetime from datetime import datetime
from pathlib import Path
from typing import Any, Tuple from typing import Any, Tuple
import httpx import httpx
@@ -293,11 +294,12 @@ class ForgeAgent:
if len(list_files_after) > len(list_files_before): if len(list_files_after) > len(list_files_before):
files_to_rename = list(set(list_files_after) - set(list_files_before)) files_to_rename = list(set(list_files_after) - set(list_files_before))
for file in files_to_rename: for file in files_to_rename:
file_extension = Path(file).suffix
random_file_id = "".join(random.choices(string.ascii_uppercase + string.digits, k=4)) random_file_id = "".join(random.choices(string.ascii_uppercase + string.digits, k=4))
random_file_name = f"download-{datetime.now().strftime('%Y%m%d%H%M%S%f')}-{random_file_id}" random_file_name = f"download-{datetime.now().strftime('%Y%m%d%H%M%S%f')}-{random_file_id}"
if task_block.download_suffix: if task_block.download_suffix:
random_file_name = f"{random_file_name}-{task_block.download_suffix}" random_file_name = f"{random_file_name}-{task_block.download_suffix}"
rename_file(os.path.join(workflow_download_directory, file), random_file_name) rename_file(os.path.join(workflow_download_directory, file), random_file_name + file_extension)
LOG.info( LOG.info(
"Task marked as completed due to download", "Task marked as completed due to download",