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

add download filename rule for supabase (#2432)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-05-22 10:53:59 -07:00
committed by GitHub
parent 7c1240bfef
commit e5f52b8c9d
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
skyvern/forge/sdk/api/files.py
View File

@@ -7,7 +7,7 @@ import shutil
import tempfile import tempfile
import zipfile import zipfile
from pathlib import Path from pathlib import Path
from urllib.parse import unquote, urlparse from urllib.parse import parse_qsl, unquote, urlparse
import aiohttp import aiohttp
import structlog import structlog
@@ -98,7 +98,16 @@ async def download_file(url: str, max_size_mb: int | None = None) -> str:
# Get the file name # Get the file name
temp_dir = make_temp_directory(prefix="skyvern_downloads_") temp_dir = make_temp_directory(prefix="skyvern_downloads_")
# Check for download parameter in Supabase URLs
file_name = os.path.basename(a.path) file_name = os.path.basename(a.path)
if "supabase.co" in a.netloc.lower():
query_params = dict(parse_qsl(a.query))
if "download" in query_params:
file_name = query_params["download"]
else:
file_name = os.path.basename(a.path)
file_name = sanitize_filename(file_name)
# if no suffix in the URL, we need to parse it from HTTP headers # if no suffix in the URL, we need to parse it from HTTP headers
if not Path(file_name).suffix: if not Path(file_name).suffix:
LOG.info("No file extension detected, trying to retrieve it from HTTP headers") LOG.info("No file extension detected, trying to retrieve it from HTTP headers")