o
    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   sN    |  tj} | r!| tjkr%| V  t| \} }| r#| tjksdS dS dS 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||}|s't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vS )N>   ...r	   )partr	   r	   r
   allowedf      z)SanitizedNames._sanitize.<locals>.allowedz	^([A-Z]):z\1)flags\/zEmpty filename)	resub
IGNORECASEreplacer   joinfilter
ValueErrorendswith)namer   barecleanparts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 rleZd+S 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zipfileZipFile_pathlib_compatrootat)r   r=   r>   r	   r	   r
   __init__   s
   

zPath.__init__c                 C   s&   z|   W S  ty   t|  Y S w )zu
        For path-like objects, convert to a filename for compatibility
        on Python 3.6.1 and earlier.
        )
__fspath__AttributeErrorstrr   r	   r	   r
   r<      s
   
zPath._pathlib_compatc                 C      t | jj| jS r   )	functoolspartialr=   openr>   r   r	   r	   r
   rF         z	Path.openc                 C   s   t | jdS Nr#   )r   basenamer>   r   r   r	   r	   r
   r,      rG   z	Path.namec                 O   sH   |   }tj|g|R i | W  d    S 1 sw   Y  d S r   )rF   ioTextIOWrapperread)r   argskwargsstrmr	   r	   r
   	read_text   s   
$zPath.read_textc                 C   s4   |   }| W  d    S 1 sw   Y  d S r   )rF   rL   )r   rO   r	   r	   r
   
read_bytes   s   
$zPath.read_bytesc                 C   s   t |jd| jdkS rH   )r   dirnamer>   r   )r   r   r	   r	   r
   	_is_child   s   zPath._is_childc                 C   s   t | j|S r   )r7   r=   )r   r>   r	   r	   r
   _next   r    z
Path._nextc                 C   s   | j  p	| j dS rH   )r>   r+   r   r	   r	   r
   is_dir   s   zPath.is_dirc                 C   s
   |    S r   )rU   r   r	   r	   r
   is_file   s   
zPath.is_filec                 C   s   | j |  v S r   )r>   _namesr   r	   r	   r
   exists      zPath.existsc                 C   s,   |   stdt| j|  }t| j|S )NzCan't listdir a file)rU   r*   r   rT   rW   r)   rS   )r   subsr	   r	   r
   iterdir   s   zPath.iterdirc                 C   rC   r   )r   r(   r=   filenamer>   r   r	   r	   r
   __str__   s   zPath.__str__c                 C   s   | j j| dS )Nr   )_Path__reprformatr   r	   r	   r
   __repr__   rY   zPath.__repr__c                 C   sP   |  |}t| j|}t| j|d}|  }| ||vr%||v r%|S |S )Nr8   )r<   r   r(   r>   rW   rT   )r   addnextnext_dirnamesr	   r	   r
   joinpath   s
   
 zPath.joinpathc                    s   t  fdd D S )Nc                 3   s2    | ]}t |D ]}|d   vr|d  V  qqdS )r#   N)r   ).0r,   parentrd   r	   r
   	<genexpr>   s    z%Path._implied_dirs.<locals>.<genexpr>)more_itertoolsunique_everseenrh   r	   rh   r
   _implied_dirs   s   zPath._implied_dirsc                 C   s   |t | | S r   )r   rl   )clsrd   r	   r	   r
   _add_implied_dirs  rG   zPath._add_implied_dirsc                 C   s(   t | jd}|r|d7 }| |S rH   )r   rR   r>   r   rT   )r   	parent_atr	   r	   r
   rg   
  s   
zPath.parentc                 C   s   |  tttj| j S r   )rn   r   r   r   r   r=   r   r   r	   r	   r
   rW     s   zPath._names)   N)r8   )r1   r2   r3   r4   r^   r?   r5   r<   propertyrF   r,   rP   rQ   rS   rT   rU   rV   rX   r[   r]   r`   re   __truediv__rl   classmethodrn   rg   rW   sysversion_info__div__r	   r	   r	   r
   r7   s   s@    @








r7   )
__future__r   rJ   rt   r   r:   rD   r   r$   rj   type__metaclass__r   r   r   r7   r	   r	   r	   r
   <module>   s   9