mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-10-26 21:26:39 +00:00
fa9fca740c
* fix(browser): publish captured screenshots and fetches atomically _write_screenshot and _write_capture_output wrote the final path with a truncating write and then returned its name, so a failure part-way through left a partial capture that the caller reported as saved - and the leftovers then made the dedupe helper pick a new name for the next capture. Publish through a temporary file plus os.replace. * fix(browser): preserve permissions on atomic capture publication --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
"""Atomic byte publication with normal file-creation permissions."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import stat
|
|
from pathlib import Path
|
|
from uuid import uuid4
|
|
|
|
|
|
def publish_bytes_atomically(path: Path, content: bytes) -> None:
|
|
"""Replace *path* with complete bytes, preserving its existing mode.
|
|
|
|
New files use ``0666 & ~umask``, like ``Path.write_bytes``, so captures
|
|
remain readable through sandbox mounts under the usual ``0022`` umask.
|
|
A hidden, exclusively created sibling keeps partial writes away from the
|
|
final path. Failure before replacement leaves the previous file intact.
|
|
"""
|
|
try:
|
|
mode = stat.S_IMODE(path.stat().st_mode)
|
|
except FileNotFoundError:
|
|
mode = None
|
|
|
|
temporary = path.with_name(f".{path.name}.{uuid4().hex}.tmp")
|
|
flags = os.O_CREAT | os.O_EXCL | os.O_WRONLY | getattr(os, "O_BINARY", 0)
|
|
# Let the kernel apply umask; reading it via os.umask would mutate
|
|
# process-global state while other threads may be creating files.
|
|
fd = os.open(temporary, flags, 0o666)
|
|
try:
|
|
handle = os.fdopen(fd, "wb")
|
|
fd = None # The file object now owns the descriptor.
|
|
with handle:
|
|
handle.write(content)
|
|
handle.flush()
|
|
if mode is not None:
|
|
if os.name == "nt":
|
|
os.chmod(temporary, mode)
|
|
else:
|
|
os.fchmod(handle.fileno(), mode)
|
|
os.fsync(handle.fileno())
|
|
os.replace(temporary, path)
|
|
except BaseException:
|
|
if fd is not None:
|
|
os.close(fd)
|
|
temporary.unlink(missing_ok=True)
|
|
raise
|