U
    ·mœd´)  ã                   @   s¾   d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZddlm	Z	 dgZ
dd„ Zdd„ ZejZd	d
„ ZG dd„ dƒZG dd„ deejƒZG dd„ deƒZddd„ZG dd„ dƒZdS )é    Né   ©Útext_encodingÚPathc                 C   s   t  t| ƒdd¡S )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)Ú	itertoolsÚisliceÚ	_ancestry)Úpath© r
   úF/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/zipp/__init__.pyÚ_parents   s    r   c                 c   s4   |   tj¡} | r0| tjkr0| V  t | ¡\} }qdS )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Ú	posixpathÚsepÚsplit)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   ÚfilterfalseÚsetÚ__contains__)ÚminuendÚ
subtrahendr
   r
   r   Ú_difference=   s    r   c                       s4   e Zd ZdZ‡ fdd„Zdd„ Z‡ fdd„Z‡  ZS )ÚInitializedStatez?
    Mix-in to save the initialization state for pickling.
    c                    s   || _ || _tƒ j||Ž d S ©N)Ú_InitializedState__argsÚ_InitializedState__kwargsÚsuperÚ__init__)ÚselfÚargsÚkwargs©Ú	__class__r
   r   r   J   s    zInitializedState.__init__c                 C   s   | j | jfS r   )r   r   ©r   r
   r
   r   Ú__getstate__O   s    zInitializedState.__getstate__c                    s   |\}}t ƒ j||Ž d S r   )r   r   )r   Ústater   r    r!   r
   r   Ú__setstate__R   s    zInitializedState.__setstate__)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r$   r&   Ú__classcell__r
   r
   r!   r   r   E   s   r   c                       sT   e Zd ZdZedd„ ƒZ‡ fdd„Zdd„ Zdd	„ Z‡ fd
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/']
    c                 C   s.   t j tt| ƒ¡}dd„ |D ƒ}tt|| ƒƒS )Nc                 s   s   | ]}|t j V  qd S r   )r   r   )Ú.0Úpr
   r
   r   Ú	<genexpr>e   s     z-CompleteDirs._implied_dirs.<locals>.<genexpr>)r   ÚchainÚfrom_iterableÚmapr   Ú_deduper   )ÚnamesÚparentsÚas_dirsr
   r
   r   Ú_implied_dirsb   s    zCompleteDirs._implied_dirsc                    s   t ƒ  ¡ }|t|  |¡ƒ S r   )r   ÚnamelistÚlistr7   )r   r4   r!   r
   r   r8   h   s    
zCompleteDirs.namelistc                 C   s   t |  ¡ ƒS r   )r   r8   r#   r
   r
   r   Ú	_name_setl   s    zCompleteDirs._name_setc                 C   s,   |   ¡ }|d }||ko||k}|r(|S |S )zx
        If the name represents a directory, return that name
        as a directory (with the trailing slash).
        ú/)r:   )r   Únamer4   ÚdirnameÚ	dir_matchr
   r
   r   Úresolve_diro   s    zCompleteDirs.resolve_dirc                    sL   zt ƒ  |¡W S  tk
rF   | d¡r4||  ¡ kr6‚ tj|d� Y S X dS )z6
        Supplement getinfo for implied dirs.
        r;   )ÚfilenameN)r   ÚgetinfoÚKeyErrorÚendswithr:   ÚzipfileÚZipInfo)r   r<   r!   r
   r   rA   y   s    zCompleteDirs.getinfoc                 C   s:   t |tƒr|S t |tjƒs"| |ƒS d|jkr0t} | |_|S )zl
        Given a source (filename or zipfile), return an
        appropriate CompleteDirs subclass.
        Úr)Ú
isinstancer,   rD   ÚZipFileÚmoder"   )ÚclsÚsourcer
   r
   r   Úmake„   s    

zCompleteDirs.make)r'   r(   r)   r*   Ústaticmethodr7   r8   r:   r?   rA   ÚclassmethodrL   r+   r
   r
   r!   r   r,   W   s   


r,   c                       s,   e Zd ZdZ‡ fdd„Z‡ fdd„Z‡  ZS )Ú
FastLookupzV
    ZipFile subclass to ensure implicit
    dirs exist and are resolved rapidly.
    c              
      s6   t  t¡� | jW  5 Q R £ S Q R X tƒ  ¡ | _| jS r   )Ú
contextlibÚsuppressÚAttributeErrorÚ_FastLookup__namesr   r8   r#   r!   r
   r   r8   ž   s    zFastLookup.namelistc              
      s6   t  t¡� | jW  5 Q R £ S Q R X tƒ  ¡ | _| jS r   )rP   rQ   rR   Ú_FastLookup__lookupr   r:   r#   r!   r
   r   r:   ¤   s    zFastLookup._name_set)r'   r(   r)   r*   r8   r:   r+   r
   r
   r!   r   rO   ˜   s   rO   c                 O   s   t | dƒ||fS )Né   r   )Úencodingr   r    r
   r
   r   Ú_extract_text_encoding«   s    rW   c                   @   s  e Zd ZdZdZd=dd„Zdd„ Zdd	„ Zd>ddœdd„Ze	dd„ ƒZ
e	dd„ ƒZe	dd„ ƒZe	dd„ ƒZe	dd„ ƒZdd„ Zdd„ Zdd„ Zdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+d,„ Zd-d.„ Zd/d0„ Zd1d2„ Zd3d4„ Zd5d6„ Zd7d8„ Zd9d:„ ZeZ e	d;d<„ ƒZ!dS )?r   uD  
    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})Ú c                 C   s   t  |¡| _|| _dS )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)rO   rL   ÚrootÚat)r   rY   rZ   r
   r
   r   r      s    
zPath.__init__c                 C   s(   | j |j k	rtS | j| jf|j|jfkS )zU
        >>> Path(zipfile.ZipFile(io.BytesIO(), 'w')) == 'foo'
        False
        )r"   ÚNotImplementedrY   rZ   )r   Úotherr
   r
   r   Ú__eq__  s    zPath.__eq__c                 C   s   t | j| jfƒS r   )ÚhashrY   rZ   r#   r
   r
   r   Ú__hash__  s    zPath.__hash__rF   N©Úpwdc                O   s„   |   ¡ rt| ƒ‚|d }|  ¡ s0|dkr0t| ƒ‚| jj| j||d�}d|kr`|sT|r\tdƒ‚|S t||Ž\}}}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().
        r   rF   r`   Úbz*encoding args invalid for binary operation)Úis_dirÚIsADirectoryErrorÚexistsÚFileNotFoundErrorrY   ÚopenrZ   Ú
ValueErrorrW   ÚioÚTextIOWrapper)r   rI   ra   r   r    Zzip_modeÚstreamrV   r
   r
   r   rg     s    z	Path.openc                 C   s   t  | j¡jp| jjS r   )Úpathlibr   rZ   r<   r@   r#   r
   r
   r   r<   -  s    z	Path.namec                 C   s   t  | j¡jp| jjS r   )rl   r   rZ   Úsuffixr@   r#   r
   r
   r   rm   1  s    zPath.suffixc                 C   s   t  | j¡jp| jjS r   )rl   r   rZ   Úsuffixesr@   r#   r
   r
   r   rn   5  s    zPath.suffixesc                 C   s   t  | j¡jp| jjS r   )rl   r   rZ   Ústemr@   r#   r
   r
   r   ro   9  s    z	Path.stemc                 C   s   t  | jj¡ | j¡S r   )rl   r   rY   r@   ÚjoinpathrZ   r#   r
   r
   r   r@   =  s    zPath.filenamec              
   O   sD   t ||Ž\}}}| jd|f|ž|Ž�}| ¡ W  5 Q R £ S Q R X d S )NrF   )rW   rg   Úread)r   r   r    rV   Ústrmr
   r
   r   Ú	read_textA  s    zPath.read_textc              
   C   s*   |   d¡�}| ¡ W  5 Q R £ S Q R X d S )NÚrb)rg   rq   )r   rr   r
   r
   r   Ú
read_bytesF  s    zPath.read_bytesc                 C   s   t  |j d¡¡| j d¡kS ©Nr;   )r   r=   rZ   r   )r   r	   r
   r
   r   Ú	_is_childJ  s    zPath._is_childc                 C   s   |   | j|¡S r   )r"   rY   )r   rZ   r
   r
   r   Ú_nextM  s    z
Path._nextc                 C   s   | j  p| j  d¡S rv   )rZ   rC   r#   r
   r
   r   rc   P  s    zPath.is_dirc                 C   s   |   ¡ o|  ¡  S r   )re   rc   r#   r
   r
   r   Úis_fileS  s    zPath.is_filec                 C   s   | j | j ¡ kS r   )rZ   rY   r:   r#   r
   r
   r   re   V  s    zPath.existsc                 C   s.   |   ¡ stdƒ‚t| j| j ¡ ƒ}t| j|ƒS )NzCan't listdir a file)rc   rh   r2   rx   rY   r8   Úfilterrw   )r   Úsubsr
   r
   r   ÚiterdirY  s    zPath.iterdirc                 C   s   t  | j¡ |¡S r   )rl   r   rZ   Úmatch)r   Úpath_patternr
   r
   r   r}   _  s    z
Path.matchc                 C   s   dS )z]
        Return whether this path is a symlink. Always false (python/cpython#82102).
        Fr
   r#   r
   r
   r   Ú
is_symlinkb  s    zPath.is_symlinkc                 c   s.   |   ¡ D ] }|V  | ¡ r| ¡ E d H  qd S r   )r|   rc   Ú_descendants)r   Úchildr
   r
   r   r€   h  s    zPath._descendantsc                    s<   |st d|›�ƒ‚t t |¡¡j‰ ‡ ‡fdd„ˆ ¡ D ƒS )NzUnacceptable pattern: c                 3   s$   | ]}ˆ t | ˆ¡ƒƒr|V  qd S r   )ÚstrÚrelative_to)r-   r�   ©Úmatchesr   r
   r   r/   s  s   þzPath.glob.<locals>.<genexpr>)rh   ÚreÚcompileÚfnmatchÚ	translateÚ	fullmatchr€   ©r   Úpatternr
   r„   r   Úglobn  s    þz	Path.globc                 C   s   |   d|› �¡S )Nz**/)r�   r‹   r
   r
   r   Úrgloby  s    z
Path.rglobc                 G   s   t  t| ƒt|j|Ž ƒ¡S r   )r   Úrelpathr‚   rp   )r   r\   Úextrar
   r
   r   rƒ   |  s    zPath.relative_toc                 C   s   t  | jj| j¡S r   )r   ÚjoinrY   r@   rZ   r#   r
   r
   r   Ú__str__  s    zPath.__str__c                 C   s   | j j| d�S )Nr#   )Ú_Path__reprÚformatr#   r
   r
   r   Ú__repr__‚  s    zPath.__repr__c                 G   s$   t j| jf|žŽ }|  | j |¡¡S r   )r   r‘   rZ   rx   rY   r?   )r   r\   Únextr
   r
   r   rp   …  s    zPath.joinpathc                 C   s6   | j s| jjS t | j  d¡¡}|r,|d7 }|  |¡S rv   )rZ   r@   Úparentr   r=   r   rx   )r   Ú	parent_atr
   r
   r   r—   ‹  s    zPath.parent)rX   )rF   )"r'   r(   r)   r*   r“   r   r]   r_   rg   Úpropertyr<   rm   rn   ro   r@   rs   ru   rw   rx   rc   ry   re   r|   r}   r   r€   r�   rŽ   rƒ   r’   r•   rp   Ú__truediv__r—   r
   r
   r
   r   r   °   sH   M
	




)N)ri   r   rD   r   rP   rl   r†   rˆ   Zpy310compatr   Ú__all__r   r   ÚdictÚfromkeysr3   r   r   rH   r,   rO   rW   r   r
   r
   r
   r   Ú<module>   s$   A
