import os import pathlib import zipfile import tempfile import functools import contextlib def from_package(package): """ Return a Traversable object for the given package. """ return fallback_resources(package.__spec__) def fallback_resources(spec): package_directory = pathlib.Path(spec.origin).parent try: archive_path = spec.loader.archive rel_path = package_directory.relative_to(archive_path) return zipfile.Path(archive_path, str(rel_path) + '/') except Exception: pass return package_directory @contextlib.contextmanager def _tempfile(reader, suffix=''): # Not using tempfile.NamedTemporaryFile as it leads to deeper 'try' # blocks due to the need to close the temporary file to work on Windows # properly. fd, raw_path = tempfile.mkstemp(suffix=suffix) try: os.write(fd, reader()) os.close(fd) yield pathlib.Path(raw_path) finally: try: os.remove(raw_path) except FileNotFoundError: pass @functools.singledispatch @contextlib.contextmanager def as_file(path): """ Given a Traversable object, return that object as a path on the local file system in a context manager. """ with _tempfile(path.read_bytes, suffix=path.name) as local: yield local @as_file.register(pathlib.Path) @contextlib.contextmanager def _(path): """ Degenerate behavior for pathlib.Path objects. """ yield path
Name | Type | Size | Permission | Actions |
---|---|---|---|---|
__pycache__ | Folder | 0755 |
|
|
__init__.py | File | 5.92 KB | 0644 |
|
_bootstrap.py | File | 39.38 KB | 0644 |
|
_bootstrap_external.py | File | 63.42 KB | 0644 |
|
_common.py | File | 1.46 KB | 0644 |
|
abc.py | File | 14.57 KB | 0644 |
|
machinery.py | File | 844 B | 0644 |
|
metadata.py | File | 18.31 KB | 0644 |
|
resources.py | File | 7.04 KB | 0644 |
|
util.py | File | 11.06 KB | 0644 |
|