suguo
/
dify
mirror of https://github.com/langgenius/dify.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(sandbox): add file extension attribute to SandboxFileNode and update related logic

This commit is contained in:
Harry 2026-01-27 15:58:07 +08:00
parent 046aff93f6
commit 85ecf1a198
3 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/controllers/console/sandbox_files.py
View File

@ -36,6 +36,7 @@ SANDBOX_FILE_NODE_FIELDS = {
"is_dir": fields.Boolean,
"size": fields.Raw,
"mtime": fields.Raw,
"extension": fields.String,
}
@ -47,9 +48,7 @@ SANDBOX_FILE_DOWNLOAD_TICKET_FIELDS = {
sandbox_file_node_model = console_ns.model("SandboxFileNode", SANDBOX_FILE_NODE_FIELDS)
sandbox_file_download_ticket_model = console_ns.model(
"SandboxFileDownloadTicket", SANDBOX_FILE_DOWNLOAD_TICKET_FIELDS
)
sandbox_file_download_ticket_model = console_ns.model("SandboxFileDownloadTicket", SANDBOX_FILE_DOWNLOAD_TICKET_FIELDS)
@console_ns.route("/sandboxes/<string:sandbox_id>/files")

1
api/core/sandbox/entities/files.py
View File

@ -9,6 +9,7 @@ class SandboxFileNode:
is_dir: bool
size: int | None
mtime: int | None
extension: str | None
@dataclass(frozen=True)

20
api/core/sandbox/inspector.py
View File

@ -115,12 +115,19 @@ print(json.dumps(entries))
entries: list[SandboxFileNode] = []
for item in raw:
item_path = str(item.get("path"))
item_is_dir = bool(item.get("is_dir"))
extension = None
if not item_is_dir:
ext = os.path.splitext(item_path)[1]
extension = ext or None
entries.append(
SandboxFileNode(
path=str(item.get("path")),
is_dir=bool(item.get("is_dir")),
path=item_path,
is_dir=item_is_dir,
size=item.get("size"),
mtime=item.get("mtime"),
extension=extension,
)
)
return entries
@ -243,7 +250,9 @@ class SandboxFileArchiveSource(SandboxFileSource):
if dir_path in ("", "."):
return
if dir_path not in entries_by_path:
entries_by_path[dir_path] = SandboxFileNode(path=dir_path, is_dir=True, size=None, mtime=None)
entries_by_path[dir_path] = SandboxFileNode(
path=dir_path, is_dir=True, size=None, mtime=None, extension=None
)
def clean(member_name: str) -> str:
name = member_name.lstrip("./")
@ -283,11 +292,16 @@ class SandboxFileArchiveSource(SandboxFileSource):
parent = os.path.dirname(parent)
is_dir = m.isdir()
extension = None
if not is_dir:
ext = os.path.splitext(mp)[1]
extension = ext or None
entries_by_path[mp] = SandboxFileNode(
path=mp,
is_dir=is_dir,
size=None if is_dir else int(m.size),
mtime=int(m.mtime) if m.mtime else None,
extension=extension,
)
return sorted(entries_by_path.values(), key=lambda e: e.path)