U
    ù�f  ã                   @   s€   d dl m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	Z	e
Zdd„ Zdd„ ZG dd„ dƒZG d	d
„ d
ƒZdS )é    )ÚdivisionNc                 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(''))
    []
    é   N)Ú	itertoolsÚisliceÚ	_ancestry©Úpath© r	   ú&/usr/lib/python3/dist-packages/zipp.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                       s,   e Zd ZdZ‡ fdd„Zedd„ ƒZ‡  ZS )ÚSanitizedNamesz7
    ZipFile mix-in to ensure names are sanitized.
    c                    s   t t| jtƒ  ¡ ƒƒS ©N)ÚlistÚmapÚ	_sanitizeÚsuperÚnamelist©Úself©Ú	__class__r	   r
   r   ?   s    zSanitizedNames.namelistc                 C   s`   dd„ }t jdd| t jd�}| dd¡}| d¡}d t||ƒ¡}|sNtdƒ‚|d|  d¡  S )	a]  
        Ensure a relative path with posix separators and no dot names.
        Modeled after
        https://github.com/python/cpython/blob/bcc1be39cb1d04ad9fc0bd1b9193d3972835a57c/Lib/zipfile/__init__.py#L1799-L1813
        but provides consistent cross-platform behavior.
        >>> san = SanitizedNames._sanitize
        >>> san('/foo/bar')
        'foo/bar'
        >>> san('//foo.txt')
        'foo.txt'
        >>> san('foo/.././bar.txt')
        'foo/bar.txt'
        >>> san('foo../.bar.txt')
        'foo../.bar.txt'
        >>> san('\\foo\\bar.txt')
        'foo/bar.txt'
        >>> san('D:\\foo.txt')
        'D/foo.txt'
        >>> san('\\\\server\\share\\file.txt')
        'server/share/file.txt'
        >>> san('\\\\?\\GLOBALROOT\\Volume3')
        '?/GLOBALROOT/Volume3'
        >>> san('\\\\.\\PhysicalDrive1\\root')
        'PhysicalDrive1/root'
        Retain any trailing slash.
        >>> san('abc/')
        'abc/'
        Raises a ValueError if the result is empty.
        >>> san('../..')
        Traceback (most recent call last):
        ...
        ValueError: Empty filename
        c                 S   s   | o
| dkS )N>   ú..Ú.r	   )Úpartr	   r	   r
   Úallowedf   s    z)SanitizedNames._sanitize.<locals>.allowedz	^([A-Z]):z\1)Úflagsú\ú/zEmpty filename)	ÚreÚsubÚ
IGNORECASEÚreplacer   ÚjoinÚfilterÚ
ValueErrorÚendswith)Únamer   ZbareZcleanÚpartsZjoinedr	   r	   r
   r   B   s    $
zSanitizedNames._sanitize)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   Ústaticmethodr   Ú__classcell__r	   r	   r   r
   r   :   s   r   c                   @   sØ   e Zd ZdZdZd,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eZed"d#„ ƒZed$d%„ ƒZed&d'„ ƒZd(d)„ Zejd*k rÔeZd+S )-ÚPathuš  
    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 = '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('abcde.zip', 'a.txt')
    >>> b
    Path('abcde.zip', 'b/')

    name property:

    >>> b.name
    'b'

    join with divide operator:

    >>> c = b / 'c.txt'
    >>> c
    Path('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:

    >>> str(c)
    'abcde.zip/b/c.txt'
    z>{self.__class__.__name__}({self.root.filename!r}, {self.at!r})Ú c                 C   s,   t |tjƒr|nt |  |¡¡| _|| _d S r   )Ú
isinstanceÚzipfileZZipFileÚ_pathlib_compatÚrootÚat)r   r8   r9   r	   r	   r
   Ú__init__¶   s
    
ÿýzPath.__init__c                 C   s,   z
|   ¡ W S  tk
r&   t| ƒ Y S X dS )zu
        For path-like objects, convert to a filename for compatibility
        on Python 3.6.1 and earlier.
        N)Ú
__fspath__ÚAttributeErrorÚstrr   r	   r	   r
   r7   ¾   s    
zPath._pathlib_compatc                 C   s   t  | jj| j¡S r   )Ú	functoolsÚpartialr8   Úopenr9   r   r	   r	   r
   r@   É   s    z	Path.openc                 C   s   t  | j d¡¡S ©Nr"   )r   Úbasenamer9   r   r   r	   r	   r
   r+   Í   s    z	Path.namec              
   O   s6   |   ¡ �$}tj|f|ž|Ž ¡ W  5 Q R £ S Q R X d S r   )r@   ÚioÚTextIOWrapperÚread)r   ÚargsÚkwargsÚstrmr	   r	   r
   Ú	read_textÑ   s    
zPath.read_textc              
   C   s(   |   ¡ �}| ¡ W  5 Q R £ S Q R X d S r   )r@   rE   )r   rH   r	   r	   r
   Ú
read_bytesÕ   s    
zPath.read_bytesc                 C   s   t  |j d¡¡| j d¡kS rA   )r   Údirnamer9   r   )r   r   r	   r	   r
   Ú	_is_childÙ   s    zPath._is_childc                 C   s   t | j|ƒS r   )r3   r8   )r   r9   r	   r	   r
   Ú_nextÜ   s    z
Path._nextc                 C   s   | j  p| j  d¡S rA   )r9   r*   r   r	   r	   r
   Úis_dirß   s    zPath.is_dirc                 C   s
   |   ¡  S r   )rN   r   r	   r	   r
   Úis_fileâ   s    zPath.is_filec                 C   s   | j |  ¡ kS r   )r9   Ú_namesr   r	   r	   r
   Úexistså   s    zPath.existsc                 C   s,   |   ¡ stdƒ‚t| j|  ¡ ƒ}t| j|ƒS )NzCan't listdir a file)rN   r)   r   rM   rP   r(   rL   )r   Zsubsr	   r	   r
   Úiterdirè   s    zPath.iterdirc                 C   s   t  | jj| j¡S r   )r   r'   r8   Úfilenamer9   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                 C   sN   |   |¡}t | j|¡}t | j|d¡}|  ¡ }|  ||krH||krH|n|¡S )Nr4   )r7   r   r'   r9   rP   rM   )r   ÚaddÚnextZnext_dirÚnamesr	   r	   r
   Újoinpathô   s
    
zPath.joinpathc                    s   t  ‡ fdd„ˆ D ƒ¡S )Nc                 3   s0   | ](}t |ƒD ]}|d  ˆ kr|d  V  qqdS )r"   N)r   )Ú.0r+   Úparent©rZ   r	   r
   Ú	<genexpr>ÿ   s
   
 ýz%Path._implied_dirs.<locals>.<genexpr>)Úmore_itertoolsZunique_everseenr^   r	   r^   r
   Ú_implied_dirsý   s    þzPath._implied_dirsc                 C   s   |t |  |¡ƒ S r   )r   ra   )ÚclsrZ   r	   r	   r
   Ú_add_implied_dirs  s    zPath._add_implied_dirsc                 C   s(   t  | j d¡¡}|r|d7 }|  |¡S rA   )r   rK   r9   r   rM   )r   Z	parent_atr	   r	   r
   r]   
  s    zPath.parentc                 C   s   |   tttj| j ¡ ƒƒ¡S r   )rc   r   r   r   r   r8   r   r   r	   r	   r
   rP     s    zPath._names)é   N)r4   )r-   r.   r/   r0   rU   r:   r1   r7   Úpropertyr@   r+   rI   rJ   rL   rM   rN   rO   rQ   rR   rT   rW   r[   Ú__truediv__ra   Úclassmethodrc   r]   rP   ÚsysÚversion_infoZ__div__r	   r	   r	   r
   r3   s   s<   @








r3   )Z
__future__r   rC   rh   r   r6   r>   r   r#   r`   ÚtypeZ__metaclass__r   r   r   r3   r	   r	   r	   r
   Ú<module>   s   9