More convenient context manager for mmap.mmap

memorymapped.py

"""More convenient context manager for mmap.mmap."""

from collections.abc import Iterator
import contextlib
import mmap
import os


@contextlib.contextmanager
def memorymapped(path: os.PathLike[str] | str)-> Iterator[mmap.mmap]:
    """Return a block context with `path` as memory-mapped file."""
    with (open(path, mode='rb') as f,
          mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_READ) as m):
        yield m


if __name__ == '__main__':
    with memorymapped('memorymapped.py') as data:
        print(data[247:304])