B xtb]@sddlZddlZddlZddlZddlZddlZdgZddZddZe j Z ddZ Gd d d ej ZGd d d eZGd ddZdS)NPathcCstt|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('')) [] N) itertoolsislice _ancestry)pathr$/tmp/pip-build-xxqi51j6/zipp/zipp.py_parents sr ccs8|tj}x&|r2|tjkr2|Vt|\}}qWdS)aR 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('')) [] N)rstrip posixpathsepsplit)rtailrrr rs rcCstt|j|S)zZ Return items in minuend not in subtrahend, retaining order with O(1) lookup. )r filterfalseset __contains__)ZminuendZ subtrahendrrr _difference9srcsHeZdZdZeddZfddZddZdd Ze d d Z Z S) CompleteDirszk A ZipFile subclass that ensures that implied directories are always included in the namelist. cCs.tjtt|}dd|D}tt||S)Ncss|]}|tjVqdS)N)r r ).0prrr Jsz-CompleteDirs._implied_dirs..)rchain from_iterablemapr _deduper)namesparentsZas_dirsrrr _implied_dirsGszCompleteDirs._implied_dirscs tt|}|t||S)N)superrnamelistlistr)selfr) __class__rr r MszCompleteDirs.namelistcCs t|S)N)rr )r"rrr _name_setQszCompleteDirs._name_setcCs,|}|d}||ko||k}|r(|S|S)zx If the name represents a directory, return that name as a directory (with the trailing slash). /)r$)r"namerdirnameZ dir_matchrrr resolve_dirTszCompleteDirs.resolve_dircCs:t|tr|St|tjs"||Sd|jkr0t}||_|S)zl Given a source (filename or zipfile), return an appropriate CompleteDirs subclass. r) isinstancerzipfileZipFilemoder#)clssourcerrr make^s   zCompleteDirs.make) __name__ __module__ __qualname____doc__ staticmethodrr r$r( classmethodr0 __classcell__rr)r#r rAs    rcs,eZdZdZfddZfddZZS) FastLookupzV ZipFile subclass to ensure implicit dirs exist and are resolved rapidly. c s.tt|jSQRXtt||_|jS)N) contextlibsuppressAttributeErrorZ_FastLookup__namesrr8r )r")r#rr r xs  zFastLookup.namelistc s.tt|jSQRXtt||_|jS)N)r9r:r;Z_FastLookup__lookuprr8r$)r")r#rr r$~s  zFastLookup._name_set)r1r2r3r4r r$r7rr)r#r r8rs r8c@seZdZdZdZd-ddZd.ddd d Zed d Zed dZ eddZ eddZ eddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*ZeZed+d,ZdS)/ru4 A pathlib-compatible interface for zip files. 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 >>> root = Path(zf) From there, several path operations are available. Directory iteration (including the zip file itself): >>> a, b = root.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() '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. Note these attributes are not valid and will raise a ``ValueError`` if the zipfile has no filename. >>> root.name 'abcde.zip' >>> str(root.filename).replace(os.sep, posixpath.sep) 'mem/abcde.zip' >>> str(root.parent) 'mem' z>{self.__class__.__name__}({self.root.filename!r}, {self.at!r})cCst||_||_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)r8r0rootat)r"r=r>rrr __init__s z Path.__init__r)N)pwdcOsr|rt||d}|s0|dkr0t||jj|j||d}d|kr`|sT|r\td|Stj |f||S)z Open this entry as text or binary following the semantics of ``pathlib.Path.open()`` by passing arguments through to io.TextIOWrapper(). rr))r@bz*encoding args invalid for binary operation) is_dirIsADirectoryErrorexistsFileNotFoundErrorr=openr> ValueErrorio TextIOWrapper)r"r-r@argskwargsZzip_modestreamrrr rFsz Path.opencCst|jjp|jjS)N)pathlibrr>r&filename)r"rrr r&sz Path.namecCst|jjp|jjS)N)rMrr>suffixrN)r"rrr rOsz Path.suffixcCst|jjp|jjS)N)rMrr>suffixesrN)r"rrr rPsz Path.suffixescCst|jjp|jjS)N)rMrr>stemrN)r"rrr rQsz Path.stemcCst|jj|jS)N)rMrr=rNjoinpathr>)r"rrr rNsz Path.filenamec Os$|jd|| }|SQRXdS)Nr))r))rFread)r"rJrKstrmrrr read_textszPath.read_textc Cs|d }|SQRXdS)Nrb)rFrS)r"rTrrr read_bytes s zPath.read_bytescCst|jd|jdkS)Nr%)r r'r>r )r"rrrr _is_childszPath._is_childcCs||j|S)N)r#r=)r"r>rrr _nextsz Path._nextcCs|j p|jdS)Nr%)r>endswith)r"rrr rBsz Path.is_dircCs|o| S)N)rDrB)r"rrr is_filesz Path.is_filecCs|j|jkS)N)r>r=r$)r"rrr rDsz Path.existscCs.|stdt|j|j}t|j|S)NzCan't listdir a file)rBrGrrYr=r filterrX)r"subsrrr iterdirsz Path.iterdircCst|jj|jS)N)r joinr=rNr>)r"rrr __str__%sz Path.__str__cCs|jj|dS)N)r") _Path__reprformat)r"rrr __repr__(sz Path.__repr__cGs$tj|jf|}||j|S)N)r r_r>rYr=r()r"othernextrrr rR+sz Path.joinpathcCs6|js|jjSt|jd}|r,|d7}||S)Nr%)r>rNparentr r'r rY)r"Z parent_atrrr rf1s z Path.parent)r<)r))r1r2r3r4rar?rFpropertyr&rOrPrQrNrUrWrXrYrBr[rDr^r`rcrR __truediv__rfrrrr rs,L      )rHr r+rr9rM__all__r rdictfromkeysrrr,rr8rrrrr s1