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 re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