241 lines
8.6 KiB
Python
241 lines
8.6 KiB
Python
# uncompyle6 version 3.9.2
|
|
# Python bytecode version base 3.7.0 (3394)
|
|
# Decompiled from: Python 3.8.19 (default, Mar 20 2024, 15:27:52)
|
|
# [Clang 14.0.6 ]
|
|
# Embedded file name: /var/user/app/device_supervisorbak/device_supervisor/lib/zipp.py
|
|
# Compiled at: 2024-04-18 03:12:57
|
|
# Size of source mod 2**32: 8404 bytes
|
|
import io, posixpath, zipfile, itertools, contextlib, sys, pathlib
|
|
if sys.version_info < (3, 7):
|
|
from collections import OrderedDict
|
|
else:
|
|
OrderedDict = dict
|
|
|
|
def _parents(path):
|
|
"""
|
|
Given a path with elements separated by
|
|
posixpath.sep, generate all parents of that path.
|
|
|
|
>>> list(_parents('b/d'))
|
|
['b']
|
|
>>> list(_parents('/b/d/'))
|
|
['/b']
|
|
>>> list(_parents('b/d/f/'))
|
|
['b/d', 'b']
|
|
>>> list(_parents('b'))
|
|
[]
|
|
>>> list(_parents(''))
|
|
[]
|
|
"""
|
|
return itertools.islice(_ancestry(path), 1, None)
|
|
|
|
|
|
def _ancestry(path):
|
|
"""
|
|
Given a path with elements separated by
|
|
posixpath.sep, generate all elements of that path
|
|
|
|
>>> list(_ancestry('b/d'))
|
|
['b/d', 'b']
|
|
>>> list(_ancestry('/b/d/'))
|
|
['/b/d', '/b']
|
|
>>> list(_ancestry('b/d/f/'))
|
|
['b/d/f', 'b/d', 'b']
|
|
>>> list(_ancestry('b'))
|
|
['b']
|
|
>>> list(_ancestry(''))
|
|
[]
|
|
"""
|
|
path = path.rstrip(posixpath.sep)
|
|
while path and path != posixpath.sep:
|
|
yield path
|
|
path, tail = posixpath.split(path)
|
|
|
|
|
|
_dedupe = OrderedDict.fromkeys
|
|
|
|
def _difference(minuend, subtrahend):
|
|
"""
|
|
Return items in minuend not in subtrahend, retaining order
|
|
with O(1) lookup.
|
|
"""
|
|
return itertools.filterfalse(set(subtrahend).__contains__, minuend)
|
|
|
|
|
|
class CompleteDirs(zipfile.ZipFile):
|
|
__doc__ = "\n A ZipFile subclass that ensures that implied directories\n are always included in the namelist.\n "
|
|
|
|
@staticmethod
|
|
def _implied_dirs(names):
|
|
parents = itertools.chain.from_iterable(map(_parents, names))
|
|
as_dirs = (p + posixpath.sep for p in parents)
|
|
return _dedupe(_difference(as_dirs, names))
|
|
|
|
def namelist(self):
|
|
names = super(CompleteDirs, self).namelist()
|
|
return names + list(self._implied_dirs(names))
|
|
|
|
def _name_set(self):
|
|
return set(self.namelist())
|
|
|
|
def resolve_dir(self, name):
|
|
"""
|
|
If the name represents a directory, return that name
|
|
as a directory (with the trailing slash).
|
|
"""
|
|
names = self._name_set()
|
|
dirname = name + "/"
|
|
dir_match = name not in names and dirname in names
|
|
if dir_match:
|
|
return dirname
|
|
return name
|
|
|
|
@classmethod
|
|
def make(cls, source):
|
|
"""
|
|
Given a source (filename or zipfile), return an
|
|
appropriate CompleteDirs subclass.
|
|
"""
|
|
if isinstance(source, CompleteDirs):
|
|
return source
|
|
else:
|
|
return isinstance(source, zipfile.ZipFile) or cls(_pathlib_compat(source))
|
|
if "r" not in source.mode:
|
|
cls = CompleteDirs
|
|
source.__class__ = cls
|
|
return source
|
|
|
|
|
|
class FastLookup(CompleteDirs):
|
|
__doc__ = "\n ZipFile subclass to ensure implicit\n dirs exist and are resolved rapidly.\n "
|
|
|
|
def namelist(self):
|
|
with contextlib.suppress(AttributeError):
|
|
return self._FastLookup__names
|
|
self._FastLookup__names = super(FastLookup, self).namelist()
|
|
return self._FastLookup__names
|
|
|
|
def _name_set(self):
|
|
with contextlib.suppress(AttributeError):
|
|
return self._FastLookup__lookup
|
|
self._FastLookup__lookup = super(FastLookup, self)._name_set()
|
|
return self._FastLookup__lookup
|
|
|
|
|
|
def _pathlib_compat(path):
|
|
"""
|
|
For path-like objects, convert to a filename for compatibility
|
|
on Python 3.6.1 and earlier.
|
|
"""
|
|
try:
|
|
return path.__fspath__()
|
|
except AttributeError:
|
|
return str(path)
|
|
|
|
|
|
class Path:
|
|
__doc__ = "\n A pathlib-compatible interface for zip files.\n\n Consider a zip file with this structure::\n\n .\n ├── a.txt\n └── b\n ├── c.txt\n └── d\n └── e.txt\n\n >>> data = io.BytesIO()\n >>> zf = zipfile.ZipFile(data, 'w')\n >>> zf.writestr('a.txt', 'content of a')\n >>> zf.writestr('b/c.txt', 'content of c')\n >>> zf.writestr('b/d/e.txt', 'content of e')\n >>> zf.filename = 'mem/abcde.zip'\n\n Path accepts the zipfile object itself or a filename\n\n >>> root = Path(zf)\n\n From there, several path operations are available.\n\n Directory iteration (including the zip file itself):\n\n >>> a, b = root.iterdir()\n >>> a\n Path('mem/abcde.zip', 'a.txt')\n >>> b\n Path('mem/abcde.zip', 'b/')\n\n name property:\n\n >>> b.name\n 'b'\n\n join with divide operator:\n\n >>> c = b / 'c.txt'\n >>> c\n Path('mem/abcde.zip', 'b/c.txt')\n >>> c.name\n 'c.txt'\n\n Read text:\n\n >>> c.read_text()\n 'content of c'\n\n existence:\n\n >>> c.exists()\n True\n >>> (b / 'missing.txt').exists()\n False\n\n Coercion to string:\n\n >>> import os\n >>> str(c).replace(os.sep, posixpath.sep)\n 'mem/abcde.zip/b/c.txt'\n\n At the root, ``name``, ``filename``, and ``parent``\n resolve to the zipfile. Note these attributes are not\n valid and will raise a ``ValueError`` if the zipfile\n has no filename.\n\n >>> root.name\n 'abcde.zip'\n >>> str(root.filename).replace(os.sep, posixpath.sep)\n 'mem/abcde.zip'\n >>> str(root.parent)\n 'mem'\n "
|
|
_Path__repr = "{self.__class__.__name__}({self.root.filename!r}, {self.at!r})"
|
|
|
|
def __init__(self, root, at=''):
|
|
"""
|
|
Construct a Path from a ZipFile or filename.
|
|
|
|
Note: When the source is an existing ZipFile object,
|
|
its type (__class__) will be mutated to a
|
|
specialized type. If the caller wishes to retain the
|
|
original type, the caller should either create a
|
|
separate ZipFile object or pass a filename.
|
|
"""
|
|
self.root = FastLookup.make(root)
|
|
self.at = at
|
|
|
|
def open(self, mode='r', *args, pwd=None, **kwargs):
|
|
"""
|
|
Open this entry as text or binary following the semantics
|
|
of ``pathlib.Path.open()`` by passing arguments through
|
|
to io.TextIOWrapper().
|
|
"""
|
|
if self.is_dir():
|
|
raise IsADirectoryError(self)
|
|
zip_mode = mode[0]
|
|
if not self.exists():
|
|
if zip_mode == "r":
|
|
raise FileNotFoundError(self)
|
|
stream = self.root.open((self.at), zip_mode, pwd=pwd)
|
|
if "b" in mode:
|
|
if args or kwargs:
|
|
raise ValueError("encoding args invalid for binary operation")
|
|
return stream
|
|
return (io.TextIOWrapper)(stream, *args, **kwargs)
|
|
|
|
@property
|
|
def name(self):
|
|
return pathlib.Path(self.at).name or self.filename.name
|
|
|
|
@property
|
|
def suffix(self):
|
|
return pathlib.Path(self.at).suffix or self.filename.suffix
|
|
|
|
@property
|
|
def suffixes(self):
|
|
return pathlib.Path(self.at).suffixes or self.filename.suffixes
|
|
|
|
@property
|
|
def stem(self):
|
|
return pathlib.Path(self.at).stem or self.filename.stem
|
|
|
|
@property
|
|
def filename(self):
|
|
return pathlib.Path(self.root.filename).joinpath(self.at)
|
|
|
|
def read_text(self, *args, **kwargs):
|
|
with (self.open)('r', *args, **kwargs) as strm:
|
|
return strm.read()
|
|
|
|
def read_bytes(self):
|
|
with self.open("rb") as strm:
|
|
return strm.read()
|
|
|
|
def _is_child(self, path):
|
|
return posixpath.dirname(path.at.rstrip("/")) == self.at.rstrip("/")
|
|
|
|
def _next(self, at):
|
|
return self.__class__(self.root, at)
|
|
|
|
def is_dir(self):
|
|
return not self.at or self.at.endswith("/")
|
|
|
|
def is_file(self):
|
|
return self.exists() and not self.is_dir()
|
|
|
|
def exists(self):
|
|
return self.at in self.root._name_set()
|
|
|
|
def iterdir(self):
|
|
if not self.is_dir():
|
|
raise ValueError("Can't listdir a file")
|
|
subs = map(self._next, self.root.namelist())
|
|
return filter(self._is_child, subs)
|
|
|
|
def __str__(self):
|
|
return posixpath.join(self.root.filename, self.at)
|
|
|
|
def __repr__(self):
|
|
return self._Path__repr.format(self=self)
|
|
|
|
def joinpath(self, *other):
|
|
next = (posixpath.join)(self.at, *map(_pathlib_compat, other))
|
|
return self._next(self.root.resolve_dir(next))
|
|
|
|
__truediv__ = joinpath
|
|
|
|
@property
|
|
def parent(self):
|
|
if not self.at:
|
|
return self.filename.parent
|
|
parent_at = posixpath.dirname(self.at.rstrip("/"))
|
|
if parent_at:
|
|
parent_at += "/"
|
|
return self._next(parent_at)
|