U cc!@sddlZddlZddlZddlZddlZddlZddlmZdgZddZ ddZ e j Z d d ZGd d d ZGd ddeejZGdddeZGdddZdS)N) text_encodingPathcCstt|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('')) [] rN) itertoolsislice _ancestry)pathr 1/tmp/pip-unpacked-wheel-1o_9e61b/zipp/__init__.py_parentssr ccs4|tj}|r0|tjkr0|Vt|\}}q dS)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)rtailr r r r!s rcCstt|j|S)zZ Return items in minuend not in subtrahend, retaining order with O(1) lookup. )r filterfalseset __contains__)minuend subtrahendr r r _difference;srcs4eZdZdZfddZddZfddZZS)InitializedStatez? Mix-in to save the initialization state for pickling. cs||_||_tj||dSN)_InitializedState__args_InitializedState__kwargssuper__init__)selfargskwargs __class__r r rHszInitializedState.__init__cCs |j|jfSr)rrrr r r __getstate__MszInitializedState.__getstate__cs|\}}tj||dSr)rr)rstaterrr r r __setstate__PszInitializedState.__setstate__)__name__ __module__ __qualname____doc__rr#r% __classcell__r r r r rCs rcsHeZdZdZeddZfddZddZdd 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|]}|tjVqdSr)r r).0pr r r ^sz-CompleteDirs._implied_dirs..)rchain from_iterablemapr _deduper)namesparentsas_dirsr r r _implied_dirs[szCompleteDirs._implied_dirscs tt|}|t||Sr)rr+namelistlistr6)rr3r r r r7aszCompleteDirs.namelistcCs t|Sr)rr7r"r r r _name_seteszCompleteDirs._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). /)r9)rnamer3dirname dir_matchr r r resolve_dirhszCompleteDirs.resolve_dircCs:t|tr|St|tjs"||Sd|jkr0t}||_|S)zl Given a source (filename or zipfile), return an appropriate CompleteDirs subclass. r) isinstancer+zipfileZipFilemoder!)clssourcer r r makers   zCompleteDirs.make) r&r'r(r) staticmethodr6r7r9r> classmethodrFr*r r r r r+Us   r+cs,eZdZdZfddZfddZZS) FastLookupzV ZipFile subclass to ensure implicit dirs exist and are resolved rapidly. c s:tt|jW5QRSQRXtt||_|jSr) contextlibsuppressAttributeError_FastLookup__namesrrIr7r"r r r r7s zFastLookup.namelistc s:tt|jW5QRSQRXtt||_|jSr)rJrKrL_FastLookup__lookuprrIr9r"r r r r9s zFastLookup._name_set)r&r'r(r)r7r9r*r r r r rIs rIc@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)rIrFrootat)rrPrQr r r rs z Path.__init__r?NpwdcOs|rt||d}|s0|dkr0t||jj|j||d}d|kr`|sT|r\td|St| d|d<t j |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?rRbz*encoding args invalid for binary operationencoding) is_dirIsADirectoryErrorexistsFileNotFoundErrorrPopenrQ ValueErrorrgetio TextIOWrapper)rrCrSrrZzip_modestreamr r r rZsz Path.opencCst|jjp|jjSr)pathlibrrQr;filenamer"r r r r; sz Path.namecCst|jjp|jjSr)r`rrQsuffixrar"r r r rbsz Path.suffixcCst|jjp|jjSr)r`rrQsuffixesrar"r r r rcsz Path.suffixescCst|jjp|jjSr)r`rrQstemrar"r r r rdsz Path.stemcCst|jj|jSr)r`rrPrajoinpathrQr"r r r rasz Path.filenamec OsBt|d|d<|jd||}|W5QRSQRXdS)NrUr?)r?)rr\rZread)rrrstrmr r r read_textszPath.read_textc Cs*|d}|W5QRSQRXdS)Nrb)rZrf)rrgr r r read_bytes#s zPath.read_bytescCst|jd|jdkSNr:)r r<rQr )rrr r r _is_child'szPath._is_childcCs||j|Sr)r!rP)rrQr r r _next*sz Path._nextcCs|j p|jdSrk)rQendswithr"r r r rV-sz Path.is_dircCs|o| Sr)rXrVr"r r r is_file0sz Path.is_filecCs|j|jkSr)rQrPr9r"r r r rX3sz Path.existscCs.|stdt|j|j}t|j|S)NzCan't listdir a file)rVr[r1rmrPr7filterrl)rsubsr r r iterdir6sz Path.iterdircCst|jj|jSr)r joinrPrarQr"r r r __str__<sz Path.__str__cCs|jj|dS)Nr") _Path__reprformatr"r r r __repr__?sz Path.__repr__cGs$tj|jf|}||j|Sr)r rsrQrmrPr>)rothernextr r r reBsz Path.joinpathcCs6|js|jjSt|jd}|r,|d7}||Srk)rQraparentr r<r rm)r parent_atr r r rzHs z Path.parent)rO)r?)r&r'r(r)rurrZpropertyr;rbrcrdrarhrjrlrmrVrorXrrrtrwre __truediv__rzr r r r rs8M      )r]r rArrJr`Z py310compatr__all__r rdictfromkeysr2rrrBr+rIrr r r r s 1