a
f:. @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl m
Z
dgZdd Zd d
ZejZd d ZG d
d dZG dd deejZG dd deZdddZG dd dZdS )z
A Path-like interface for zipfiles.
This codebase is shared between zipfile.Path in the stdlib
and zipp in PyPI. See
https://github.com/python/importlib_metadata/wiki/Development-Methodology
for more detail.
N )
text_encoding)
TranslatorPathc C s t t| ddS )a2
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(''))
[]
r N) itertoolsislice _ancestry)path r
1/usr/lib/python3.9/site-packages/zipp/__init__.py_parents s r c c s2 | tj} | tjr.| V t| \} }q dS )a
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(''))
[]
Multiple separators are treated like a single.
>>> list(_ancestry('//b//d///f//'))
['//b//d///f', '//b//d', '//b']
N)rstrip posixpathsepsplit)r tailr
r
r r . s r c C s t t|j| S )zZ
Return items in minuend not in subtrahend, retaining order
with O(1) lookup.
)r filterfalseset __contains__)ZminuendZ
subtrahendr
r
r _differenceM s r c s4 e Zd ZdZ fddZdd Z fddZ ZS )InitializedStatez?
Mix-in to save the initialization state for pickling.
c s" || _ || _t j|i | d S N)_InitializedState__args_InitializedState__kwargssuper__init__)selfargskwargs __class__r
r r Z s zInitializedState.__init__c C s | j | jfS r )r r r r
r
r __getstate___ s z InitializedState.__getstate__c s |\}}t j|i | d S r )r r )r stater r r r
r __setstate__b s z InitializedState.__setstate__)__name__
__module__ __qualname____doc__r r" r$
__classcell__r
r
r r r U s r c sl e Zd ZdZedd Z fddZdd Zdd Z fd
d Z e
d d
Z e
e j
e j
dddZ ZS ) CompleteDirsa8
A ZipFile subclass that ensures that implied directories
are always included in the namelist.
>>> list(CompleteDirs._implied_dirs(['foo/bar.txt', 'foo/bar/baz.txt']))
['foo/', 'foo/bar/']
>>> list(CompleteDirs._implied_dirs(['foo/bar.txt', 'foo/bar/baz.txt', 'foo/bar/']))
['foo/']
c C s. t jtt| }dd |D }tt|| S )Nc s s | ]}|t j V qd S r )r r ).0pr
r
r u z-CompleteDirs._implied_dirs..)r chain
from_iterablemapr _deduper )namesparentsZas_dirsr
r
r
_implied_dirsr s zCompleteDirs._implied_dirsc s t }|t| | S r )r namelistlistr5 )r r3 r r
r r6 x s
zCompleteDirs.namelistc C s t | S r )r r6 r! r
r
r _name_set| s zCompleteDirs._name_setc C s, | }|d }||vo ||v }|r(|S |S )zx
If the name represents a directory, return that name
as a directory (with the trailing slash).
/)r8 )r namer3 dirnameZ dir_matchr
r
r resolve_dir s zCompleteDirs.resolve_dirc sJ zt |W S tyD |dr2|| vr4 tj|d Y S 0 dS )z6
Supplement getinfo for implied dirs.
r9 )filenameN)r getinfoKeyErrorendswithr8 zipfileZZipInfo)r r: r r
r r> s zCompleteDirs.getinfoc C s: t |tr|S t |tjs"| |S d|jvr0t} | |_|S )zl
Given a source (filename or zipfile), return an
appropriate CompleteDirs subclass.
r)
isinstancer* rA ZipFilemoder )clssourcer
r
r make s
zCompleteDirs.make)zfreturnc C s$ | | D ]}||d q|S )z
Given a writable zip file zf, inject directory entries for
any directories implied by the presence of children.
r. )r5 r6 Zwritestr)rF rI r: r
r
r inject s zCompleteDirs.inject)r% r& r' r( staticmethodr5 r6 r8 r< r> classmethodrH rA rD rK r) r
r
r r r* g s
r* c s, e Zd ZdZ fddZ fddZ ZS )
FastLookupzV
ZipFile subclass to ensure implicit
dirs exist and are resolved rapidly.
c sB t t | jW d S 1 s&0 Y t | _| jS r )
contextlibsuppressAttributeErrorZ_FastLookup__namesr r6 r! r r
r r6 s $ zFastLookup.namelistc sB t t | jW d S 1 s&0 Y t | _| jS r )rO rP rQ Z_FastLookup__lookupr r8 r! r r
r r8 s $ zFastLookup._name_set)r% r& r' r( r6 r8 r) r
r
r r rN s rN c O s$ t jjdk}d| }t| |||fS )NZpypy )sysimplementationr: r )encodingr r Zis_pypyZ stack_levelr
r
r _extract_text_encoding s rV c @ s e Zd ZdZdZd=ddZdd Zdd Zd>d d d
dZdd Z e
dd Z e
dd Z e
dd Z
e
dd Ze
dd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Z d5d6 Z d7d8 Z d9d: Z e Z e
d;d< Z!d S )?r u'
A :class:`importlib.resources.abc.Traversable` interface for zip files.
Implements many of the features users enjoy from
:class:`pathlib.Path`.
Consider a zip file with this structure::
.
├── a.txt
└── b
├── c.txt
└── d
└── e.txt
>>> data = io.BytesIO()
>>> zf = zipfile.ZipFile(data, 'w')
>>> zf.writestr('a.txt', 'content of a')
>>> zf.writestr('b/c.txt', 'content of c')
>>> zf.writestr('b/d/e.txt', 'content of e')
>>> zf.filename = 'mem/abcde.zip'
Path accepts the zipfile object itself or a filename
>>> path = Path(zf)
From there, several path operations are available.
Directory iteration (including the zip file itself):
>>> a, b = path.iterdir()
>>> a
Path('mem/abcde.zip', 'a.txt')
>>> b
Path('mem/abcde.zip', 'b/')
name property:
>>> b.name
'b'
join with divide operator:
>>> c = b / 'c.txt'
>>> c
Path('mem/abcde.zip', 'b/c.txt')
>>> c.name
'c.txt'
Read text:
>>> c.read_text(encoding='utf-8')
'content of c'
existence:
>>> c.exists()
True
>>> (b / 'missing.txt').exists()
False
Coercion to string:
>>> import os
>>> str(c).replace(os.sep, posixpath.sep)
'mem/abcde.zip/b/c.txt'
At the root, ``name``, ``filename``, and ``parent``
resolve to the zipfile.
>>> str(path)
'mem/abcde.zip/'
>>> path.name
'abcde.zip'
>>> path.filename == pathlib.Path('mem/abcde.zip')
True
>>> str(path.parent)
'mem'
If the zipfile has no filename, such attributes are not
valid and accessing them will raise an Exception.
>>> zf.filename = None
>>> path.name
Traceback (most recent call last):
...
TypeError: ...
>>> path.filename
Traceback (most recent call last):
...
TypeError: ...
>>> path.parent
Traceback (most recent call last):
...
TypeError: ...
# workaround python/cpython#106763
>>> pass
z>{self.__class__.__name__}({self.root.filename!r}, {self.at!r}) c C s t || _|| _dS )aX
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.
N)rN rH rootat)r rX rY r
r
r r 5 s
z
Path.__init__c C s( | j |j urtS | j| jf|j|jfkS )zU
>>> Path(zipfile.ZipFile(io.BytesIO(), 'w')) == 'foo'
False
)r NotImplementedrX rY )r otherr
r
r __eq__B s z Path.__eq__c C s t | j| jfS r )hashrX rY r! r
r
r __hash__K s z
Path.__hash__rB Npwdc O s | rt| |d }| s0|dkr0t| | jj| j||d}d|v r`|sT|r\td|S t|i |\}}}t j
||g|R i |S )z
Open this entry as text or binary following the semantics
of ``pathlib.Path.open()`` by passing arguments through
to io.TextIOWrapper().
r rB r_ bz*encoding args invalid for binary operation) is_dirIsADirectoryErrorexistsFileNotFoundErrorrX openrY
ValueErrorrV io
TextIOWrapper)r rE r` r r Zzip_modestreamrU r
r
r rf N s z Path.openc C s t | jp| jjS r )pathlib
PurePosixPathrY rX r= r! r
r
r _baseb s z
Path._basec C s
| jS r )rm r: r! r
r
r r: e s z Path.namec C s
| jS r )rm suffixr! r
r
r rn i s z Path.suffixc C s
| jS r )rm suffixesr! r
r
r ro m s z
Path.suffixesc C s
| jS r )rm stemr! r
r
r rp q s z Path.stemc C s t | jj| jS r )rk r rX r= joinpathrY r! r
r
r r= u s z
Path.filenamec O sZ t |i |\}}}| jd|g|R i |}| W d S 1 sL0 Y d S )NrB )rV rf read)r r r rU strmr
r
r read_texty s zPath.read_textc C s6 | d}| W d S 1 s(0 Y d S )Nrb)rf rr )r rs r
r
r
read_bytes~ s zPath.read_bytesc C s t |jd| jdkS Nr9 )r r; rY r
)r r r
r
r _is_child s zPath._is_childc C s | | j|S r )r rX )r rY r
r
r _next s z
Path._nextc C s | j p| j dS rw )rY r@ r! r
r
r rb s z Path.is_dirc C s | o| S r )rd rb r! r
r
r is_file s z Path.is_filec C s | j | j v S r )rY rX r8 r! r
r
r rd s z Path.existsc C s. | stdt| j| j }t| j|S )NzCan't listdir a file)rb rg r1 ry rX r6 filterrx )r Zsubsr
r
r iterdir s z Path.iterdirc C s t | j|S r )rk rl rY match)r path_patternr
r
r r} s z
Path.matchc C s" | j | j}|jd? }t|S )z8
Return whether this path is a symlink.
)rX r> rY Z
external_attrstatS_ISLNK)r inforE r
r
r
is_symlink s
zPath.is_symlinkc C sV |st d|t| j}tdd}t||| j}t| j t
|| j S )NzUnacceptable pattern: r9 )seps)
rg reescaperY r compile translate fullmatchr1 ry r{ rX r6 )r patternprefixtrmatchesr
r
r glob s
z Path.globc C s | d| S )Nz**/)r )r r r
r
r rglob s z
Path.rglobc G s t t| t|j| S r )r relpathstrrq )r r[ Zextrar
r
r relative_to s zPath.relative_toc C s t | jj| jS r )r joinrX r= rY r! r
r
r __str__ s z Path.__str__c C s | j j| dS )Nr! ) _Path__reprformatr! r
r
r __repr__ s z
Path.__repr__c G s& t j| jg|R }| | j|S r )r r rY ry rX r< )r r[ nextr
r
r rq s z
Path.joinpathc C s6 | j s| jjS t| j d}|r,|d7 }| |S rw )rY r= parentr r; r
ry )r Z parent_atr
r
r r s z Path.parent)rW )rB )"r% r& r' r( r r r\ r^ rf rm propertyr: rn ro rp r= rt rv rx ry rb rz rd r| r} r r r r r r rq __truediv__r r
r
r
r r sH f
)N)r( rh r rA r rO rk r r rS Z compat.py310r r r __all__r r dictfromkeysr2 r r rD r* rN rV r r
r
r
r s* K