Function read_file_bytes

src/utils/image_processor.py:123–134

Read up to ``max_bytes`` from ``path``. Caps disk I/O so a symlink to /dev/zero or a corrupted stat'd size can't OOM the process. When ``max_bytes`` is None, behaves like ``read_bytes``.

(path: Path, max_bytes: int | None = None)

# ---------------------------------------------------------------------------

def read_file_bytes(path: Path, max_bytes: int | None = None) -> bytes:
    """Read up to ``max_bytes`` from ``path``.

    Caps disk I/O so a symlink to /dev/zero or a corrupted stat'd size can't
    OOM the process. When ``max_bytes`` is None, behaves like ``read_bytes``.
    """
    if max_bytes is None:
        return path.read_bytes()
    size = path.stat().st_size
    read_size = min(size, max_bytes)
    with path.open("rb") as fh:
        return fh.read(read_size)


# ---------------------------------------------------------------------------