U
    Tc&                     @   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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	   1/tmp/pip-unpacked-wheel-xxnzqn00/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                       sH   e Zd ZdZedd Z fddZdd Zdd	 Ze	d
d Z
  ZS )CompleteDirszk
    A ZipFile subclass that ensures that implied directories
    are always included in the namelist.
    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>`   s     z-CompleteDirs._implied_dirs.<locals>.<genexpr>)r   chainfrom_iterablemapr   _deduper   )namesparentsas_dirsr	   r	   r
   _implied_dirs]   s    zCompleteDirs._implied_dirsc                    s    t t|  }|t| | S r   )r   r+   namelistlistr6   )r   r3   r    r	   r
   r7   c   s    zCompleteDirs.namelistc                 C   s   t |  S r   )r   r7   r"   r	   r	   r
   	_name_setg   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).
        /)r9   )r   namer3   dirname	dir_matchr	   r	   r
   resolve_dirj   s    zCompleteDirs.resolve_dirc                 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+   zipfileZipFilemoder!   )clssourcer	   r	   r
   maket   s    

zCompleteDirs.make)r&   r'   r(   r)   staticmethodr6   r7   r9   r>   classmethodrF   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              
      s:   t t | jW  5 Q R  S Q R X tt|  | _| jS r   )
contextlibsuppressAttributeError_FastLookup__namesr   rI   r7   r"   r    r	   r
   r7      s    zFastLookup.namelistc              
      s:   t t | jW  5 Q R  S Q R X tt|  | _| jS r   )rJ   rK   rL   _FastLookup__lookupr   rI   r9   r"   r    r	   r
   r9      s    zFastLookup._name_set)r&   r'   r(   r)   r7   r9   r*   r	   r	   r    r
   rI      s   rI   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   u4  
    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}) 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)rI   rF   rootat)r   rP   rQ   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!   NotImplementedrP   rQ   )r   otherr	   r	   r
   __eq__   s    zPath.__eq__c                 C   s   t | j| jfS r   )hashrP   rQ   r"   r	   r	   r
   __hash__  s    zPath.__hash__r?   Npwdc                O   s   |   rt| |d }|  s0|dkr0t| | jj| j||d}d|kr`|sT|r\td|S t|	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().
        r   r?   rW   bz*encoding args invalid for binary operationencoding)is_dirIsADirectoryErrorexistsFileNotFoundErrorrP   openrQ   
ValueErrorr   getioTextIOWrapper)r   rC   rX   r   r   Zzip_modestreamr	   r	   r
   r_     s    z	Path.openc                 C   s   t | jjp| jjS r   )pathlibr   rQ   r;   filenamer"   r	   r	   r
   r;     s    z	Path.namec                 C   s   t | jjp| jjS r   )re   r   rQ   suffixrf   r"   r	   r	   r
   rg     s    zPath.suffixc                 C   s   t | jjp| jjS r   )re   r   rQ   suffixesrf   r"   r	   r	   r
   rh      s    zPath.suffixesc                 C   s   t | jjp| jjS r   )re   r   rQ   stemrf   r"   r	   r	   r
   ri   $  s    z	Path.stemc                 C   s   t | jj| jS r   )re   r   rP   rf   joinpathrQ   r"   r	   r	   r
   rf   (  s    zPath.filenamec              
   O   sB   t |d|d< | jd||}| W  5 Q R  S Q R X d S )NrZ   r?   )r?   )r   ra   r_   read)r   r   r   strmr	   r	   r
   	read_text,  s    zPath.read_textc              
   C   s*   |  d}| W  5 Q R  S Q R X d S )Nrb)r_   rk   )r   rl   r	   r	   r
   
read_bytes1  s    zPath.read_bytesc                 C   s   t |jd| jdkS Nr:   )r   r<   rQ   r   )r   r   r	   r	   r
   	_is_child5  s    zPath._is_childc                 C   s   |  | j|S r   )r!   rP   )r   rQ   r	   r	   r
   _next8  s    z
Path._nextc                 C   s   | j  p| j dS rp   )rQ   endswithr"   r	   r	   r
   r[   ;  s    zPath.is_dirc                 C   s   |   o|   S r   )r]   r[   r"   r	   r	   r
   is_file>  s    zPath.is_filec                 C   s   | j | j kS r   )rQ   rP   r9   r"   r	   r	   r
   r]   A  s    zPath.existsc                 C   s.   |   stdt| j| j }t| j|S )NzCan't listdir a file)r[   r`   r1   rr   rP   r7   filterrq   )r   subsr	   r	   r
   iterdirD  s    zPath.iterdirc                 C   s   t | j|S r   )re   r   rQ   match)r   path_patternr	   r	   r
   rx   J  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_symlinkM  s    zPath.is_symlinkc                 c   s.   |   D ] }|V  | r| E d H  qd S r   )rw   r[   _descendants)r   childr	   r	   r
   r{   S  s    zPath._descendantsc                    s<   |st d|tt|j  fdd D S )NzUnacceptable pattern: {!r}c                 3   s$   | ]} t |r|V  qd S r   )strrelative_to)r,   r|   matchesr   r	   r
   r.   ^  s   zPath.glob.<locals>.<genexpr>)r`   formatrecompilefnmatch	translate	fullmatchr{   r   patternr	   r   r
   globY  s    z	Path.globc                 C   s   |  d| S )Nz**/)r   r   r	   r	   r
   rglobd  s    z
Path.rglobc                 G   s   t t| t|j| S r   )r   relpathr}   rj   )r   rS   extrar	   r	   r
   r~   g  s    zPath.relative_toc                 C   s   t | jj| jS r   )r   joinrP   rf   rQ   r"   r	   r	   r
   __str__j  s    zPath.__str__c                 C   s   | j j| dS )Nr"   )_Path__reprr   r"   r	   r	   r
   __repr__m  s    zPath.__repr__c                 G   s$   t j| jf| }| | j|S r   )r   r   rQ   rr   rP   r>   )r   rS   nextr	   r	   r
   rj   p  s    zPath.joinpathc                 C   s6   | j s| jjS t| j d}|r,|d7 }| |S rp   )rQ   rf   parentr   r<   r   rr   )r   	parent_atr	   r	   r
   r   v  s    zPath.parent)rO   )r?   )"r&   r'   r(   r)   r   r   rT   rV   r_   propertyr;   rg   rh   ri   rf   rm   ro   rq   rr   r[   rt   r]   rw   rx   rz   r{   r   r   r~   r   r   rj   __truediv__r   r	   r	   r	   r
   r      sH   M
	




)rb   r   rA   r   rJ   re   r   r   Zpy310compatr   __all__r   r   dictfromkeysr2   r   r   rB   r+   rI   r   r	   r	   r	   r
   <module>   s"   1