B Mhh)@sddlZddlZddlZddlZddlZddlZddlZddlZddlm Z dgZ ddZ ddZ e jZd d ZGd d d ZGd ddeejZGdddeZdddZGdddZdS)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 //tmp/pip-install-vv81h98y/zipp/zipp/__init__.py_parentssr 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)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__)ZminuendZ subtrahendr r r _difference=srcs4eZdZdZfddZddZfddZZS)InitializedStatez? Mix-in to save the initialization state for pickling. cs||_||_tj||dS)N)_InitializedState__args_InitializedState__kwargssuper__init__)selfargskwargs) __class__r r rJszInitializedState.__init__cCs |j|jfS)N)rr)rr r r __getstate__OszInitializedState.__getstate__cs|\}}tj||dS)N)rr)rstaterr)rr r __setstate__RszInitializedState.__setstate__)__name__ __module__ __qualname____doc__rrr __classcell__r r )rr rEs rcsTeZdZdZeddZfddZddZdd Zfd d Z e d d Z Z S) 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/'] cCs.tjtt|}dd|D}tt||S)Ncss|]}|tjVqdS)N)r r).0pr r r esz-CompleteDirs._implied_dirs..)rchain from_iterablemapr _deduper)namesparentsZas_dirsr r r _implied_dirsbszCompleteDirs._implied_dirscst}|t||S)N)rnamelistlistr0)rr.)rr r r1hs zCompleteDirs.namelistcCs t|S)N)rr1)rr r r _name_setlszCompleteDirs._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). /)r3)rnamer.dirnameZ dir_matchr r r resolve_diroszCompleteDirs.resolve_dircsFy t|Stk r@|dr2||kr4tj|dSXdS)z6 Supplement getinfo for implied dirs. r4)filenameN)rgetinfoKeyErrorendswithr3zipfileZipInfo)rr5)rr r r9ys  zCompleteDirs.getinfocCs:t|tr|St|tjs"||Sd|jkr0t}||_|S)zl Given a source (filename or zipfile), return an appropriate CompleteDirs subclass. r) isinstancer&r<ZipFilemoder)clssourcer r r makes   zCompleteDirs.make) r!r"r#r$ staticmethodr0r1r3r7r9 classmethodrDr%r r )rr r&Ws    r&cs,eZdZdZfddZfddZZS) FastLookupzV ZipFile subclass to ensure implicit dirs exist and are resolved rapidly. c s*tt|jSQRXt|_|jS)N) contextlibsuppressAttributeErrorZ_FastLookup__namesrr1)r)rr r r1s   zFastLookup.namelistc s*tt|jSQRXt|_|jS)N)rHrIrJZ_FastLookup__lookuprr3)r)rr r r3s   zFastLookup._name_set)r!r"r#r$r1r3r%r r )rr rGs rGcOst|d||fS)N)r)encodingrrr r r _extract_text_encodingsrMc@seZdZdZdZd=ddZddZdd Zd>d d d dZe ddZ e ddZ e ddZ e ddZ e ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9d:ZeZ e d;d<Z!d S)?ruD 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(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. 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)rGrDrootat)rrOrPr r r rs z Path.__init__cCs(|j|jk rtS|j|jf|j|jfkS)zU >>> Path(zipfile.ZipFile(io.BytesIO(), 'w')) == 'foo' False )rNotImplementedrOrP)rotherr r r __eq__ s z Path.__eq__cCst|j|jfS)N)hashrOrP)rr r r __hash__sz Path.__hash__r>N)pwdcOs|rt||d}|s0|dkr0t||jj|j||d}d|kr`|sT|r\td|St||\}}}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>)rVbz*encoding args invalid for binary operation) is_dirIsADirectoryErrorexistsFileNotFoundErrorrOopenrP ValueErrorrMio TextIOWrapper)rrArVrrZzip_modestreamrLr r r r\sz Path.opencCst|jjp|jjS)N)pathlibrrPr5r8)rr r r r5-sz Path.namecCst|jjp|jjS)N)rarrPsuffixr8)rr r r rb1sz Path.suffixcCst|jjp|jjS)N)rarrPsuffixesr8)rr r r rc5sz Path.suffixescCst|jjp|jjS)N)rarrPstemr8)rr r r rd9sz Path.stemcCst|jj|jS)N)rarrOr8joinpathrP)rr r r r8=sz Path.filenamec Os8t||\}}}|jd|f|| }|SQRXdS)Nr>)rMr\read)rrrrLstrmr r r read_textAszPath.read_textc Cs|d }|SQRXdS)Nrb)r\rf)rrgr r r read_bytesFs zPath.read_bytescCst|jd|jdkS)Nr4)r r6rPr )rrr r r _is_childJszPath._is_childcCs||j|S)N)rrO)rrPr r r _nextMsz Path._nextcCs|j p|jdS)Nr4)rPr;)rr r r rXPsz Path.is_dircCs|o| S)N)rZrX)rr r r is_fileSsz Path.is_filecCs|j|jkS)N)rPrOr3)rr r r rZVsz Path.existscCs.|stdt|j|j}t|j|S)NzCan't listdir a file)rXr]r,rlrOr1filterrk)rsubsr r r iterdirYsz Path.iterdircCst|j|S)N)rarrPmatch)rZ path_patternr r r rq_sz Path.matchcCsdS)z] Return whether this path is a symlink. Always false (python/cpython#82102). Fr )rr r r is_symlinkbszPath.is_symlinkccs2x,|D] }|V|r |EdHq WdS)N)rprX _descendants)rchildr r r rshszPath._descendantscs<|std|tt|jfddDS)NzUnacceptable pattern: c3s$|]}t|r|VqdS)N)str relative_to)r'rt)matchesrr r r)tszPath.glob..)r]recompilefnmatch translate fullmatchrs)rpatternr )rwrr globns  z Path.globcCs|d|S)Nz**/)r~)rr}r r r rglobysz Path.rglobcGstt|t|j|S)N)r relpathrure)rrRextrar r r rv|szPath.relative_tocCst|jj|jS)N)r joinrOr8rP)rr r r __str__sz Path.__str__cCs|jj|dS)N)r) _Path__reprformat)rr r r __repr__sz Path.__repr__cGs$tj|jf|}||j|S)N)r rrPrlrOr7)rrRnextr r r resz Path.joinpathcCs6|js|jjSt|jd}|r,|d7}||S)Nr4)rPr8parentr r6r rl)rZ parent_atr r r rs z Path.parent)rN)r>)"r!r"r#r$rrrSrUr\propertyr5rbrcrdr8rhrjrkrlrXrmrZrprqrrrsr~rrvrrre __truediv__rr r r r rs<L        )N)r^r r<rrHrarxrzZ py310compatr__all__r rdictfromkeysr-rrr@r&rGrMrr r r r s$ A