SWAMPy

✎ CARVE FLESH

EDITING: file.py

"""For internal use only. It provides a slice-like file reader."""

import os
from typing import Union

try:
    # pylint: disable=no-name-in-module
    from multiprocessing import Lock
except ImportError:
    from threading import Lock  # type: ignore

class FileBuffer:
    """A slice-able file reader"""

def __init__(self, database: str) -> None:
        # pylint: disable=consider-using-with
        self._handle = open(database, "rb")
        self._size = os.fstat(self._handle.fileno()).st_size
        if not hasattr(os, "pread"):
            self._lock = Lock()

def __getitem__(self, key: Union[slice, int]):
        if isinstance(key, slice):
            return self._read(key.stop - key.start, key.start)
        if isinstance(key, int):
            return self._read(1, key)[0]
        raise TypeError("Invalid argument type.")

def rfind(self, needle: bytes, start: int) -> int:
        """Reverse find needle from start"""
        pos = self._read(self._size - start - 1, start).rfind(needle)
        if pos == -1:
            return pos
        return start + pos

def size(self) -> int:
        """Size of file"""
        return self._size

def close(self) -> None:
        """Close file"""
        self._handle.close()

if hasattr(os, "pread"):

def _read(self, buffersize: int, offset: int) -> bytes:
            """read that uses pread"""
            # pylint: disable=no-member
            return os.pread(self._handle.fileno(), buffersize, offset)

else:

def _read(self, buffersize: int, offset: int) -> bytes:
            """read with a lock

This lock is necessary as after a fork, the different processes
            will share the same file table entry, even if we dup the fd, and
            as such the same offsets. There does not appear to be a way to
            duplicate the file table entry and we cannot re-open based on the
            original path as that file may have replaced with another or
            unlinked.
            """
            with self._lock:
                self._handle.seek(offset)
                return self._handle.read(buffersize)

CANCEL

Name	Type	Size	Modified	Actions
↩ ..	DIR	—	—
📁 __pycache__/	DIR	—	2026-08-13 05:35
📄 __init__.py	PY	2.5 KB	2026-07-20 08:43	EDIT
📄 const.py	PY	122 B	2026-07-20 08:43	EDIT
📄 decoder.py	PY	6.8 KB	2026-07-20 08:43	EDIT
📄 errors.py	PY	226 B	2026-07-20 08:43	EDIT
📄 extension.pyi	PYI	1.4 KB	2026-07-20 08:43	EDIT
📄 file.py	PY	2.1 KB	2026-07-20 08:43	EDIT
📄 py.typed	TYPED	0 B	2026-07-20 08:43	EDIT
📄 reader.py	PY	10.7 KB	2026-07-20 08:43	EDIT
📄 types.py	PY	558 B	2026-07-20 08:43	EDIT