a
    yjc!                     @   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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	   I/var/www/html/django/DPS/env/lib/python3.9/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|i | d S N)_InitializedState__args_InitializedState__kwargssuper__init__)selfargskwargs	__class__r	   r
   r   H   s    zInitializedState.__init__c                 C   s   | j | jfS r   )r   r   r   r	   r	   r
   __getstate__M   s    zInitializedState.__getstate__c                    s   |\}}t  j|i | d S r   )r   r   )r   stater   r   r    r	   r
   __setstate__P   s    zInitializedState.__setstate__)__name__
__module____qualname____doc__r   r#   r%   __classcell__r	   r	   r    r
   r   C   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>^       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listr7   )r   r4   r    r	   r
   r8   a   s    zCompleteDirs.namelistc                 C   s   t |  S r   )r   r8   r"   r	   r	   r
   	_name_sete   s    zCompleteDirs._name_setc                 C   s,   |   }|d }||vo||v }|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_dirh   s    zCompleteDirs.resolve_dirc                 C   s:   t |tr|S t |tjs"| |S d|jvr0t} | |_|S )zl
        Given a source (filename or zipfile), return an
        appropriate CompleteDirs subclass.
        r)
isinstancer+   zipfileZipFilemoder!   )clssourcer	   r	   r
   maker   s    

zCompleteDirs.make)r&   r'   r(   r)   staticmethodr7   r8   r:   r?   classmethodrG   r*   r	   r	   r    r
   r+   U   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                    sF   t t | jW  d    S 1 s&0    Y  tt|  | _| jS r   )
contextlibsuppressAttributeError_FastLookup__namesr   rJ   r8   r"   r    r	   r
   r8      s    $zFastLookup.namelistc                    sF   t t | jW  d    S 1 s&0    Y  tt|  | _| jS r   )rK   rL   rM   _FastLookup__lookupr   rJ   r:   r"   r    r	   r
   r:      s    $zFastLookup._name_set)r&   r'   r(   r)   r8   r:   r*   r	   r	   r    r
   rJ      s   rJ   c                   @   s   e Zd ZdZdZd-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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)rJ   rG   rootat)r   rQ   rR   r	   r	   r
   r      s    
zPath.__init__r@   Npwdc                O   s   |   rt| |d }|  s0|dkr0t| | jj| j||d}d|v r`|sT|r\td|S t|	d|d< t
j|g|R i |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@   rS   bz*encoding args invalid for binary operationencoding)is_dirIsADirectoryErrorexistsFileNotFoundErrorrQ   openrR   
ValueErrorr   getioTextIOWrapper)r   rD   rT   r   r   zip_modestreamr	   r	   r
   r[      s    z	Path.openc                 C   s   t | jjp| jjS r   )pathlibr   rR   r<   filenamer"   r	   r	   r
   r<   
  s    z	Path.namec                 C   s   t | jjp| jjS r   )rb   r   rR   suffixrc   r"   r	   r	   r
   rd     s    zPath.suffixc                 C   s   t | jjp| jjS r   )rb   r   rR   suffixesrc   r"   r	   r	   r
   re     s    zPath.suffixesc                 C   s   t | jjp| jjS r   )rb   r   rR   stemrc   r"   r	   r	   r
   rf     s    z	Path.stemc                 C   s   t | jj| jS r   )rb   r   rQ   rc   joinpathrR   r"   r	   r	   r
   rc     s    zPath.filenamec                 O   sV   t |d|d< | jdg|R i |}| W  d    S 1 sH0    Y  d S )NrV   r@   )r   r]   r[   read)r   r   r   strmr	   r	   r
   	read_text  s    zPath.read_textc                 C   s6   |  d}| W  d    S 1 s(0    Y  d S )Nrb)r[   rh   )r   ri   r	   r	   r
   
read_bytes#  s    zPath.read_bytesc                 C   s   t |jd| jdkS Nr;   )r   r=   rR   r   )r   r   r	   r	   r
   	_is_child'  s    zPath._is_childc                 C   s   |  | j|S r   )r!   rQ   )r   rR   r	   r	   r
   _next*  s    z
Path._nextc                 C   s   | j  p| j dS rm   )rR   endswithr"   r	   r	   r
   rW   -  s    zPath.is_dirc                 C   s   |   o|   S r   )rY   rW   r"   r	   r	   r
   is_file0  s    zPath.is_filec                 C   s   | j | j v S r   )rR   rQ   r:   r"   r	   r	   r
   rY   3  s    zPath.existsc                 C   s.   |   stdt| j| j }t| j|S )NzCan't listdir a file)rW   r\   r2   ro   rQ   r8   filterrn   )r   subsr	   r	   r
   iterdir6  s    zPath.iterdirc                 C   s   t | jj| jS r   )r   joinrQ   rc   rR   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g|R  }| | j|S r   )r   ru   rR   ro   rQ   r?   )r   othernextr	   r	   r
   rg   B  s    zPath.joinpathc                 C   s6   | j s| jjS t| j d}|r,|d7 }| |S rm   )rR   rc   parentr   r=   r   ro   )r   	parent_atr	   r	   r
   r|   H  s    zPath.parent)rP   )r@   )r&   r'   r(   r)   rw   r   r[   propertyr<   rd   re   rf   rc   rj   rl   rn   ro   rW   rq   rY   rt   rv   ry   rg   __truediv__r|   r	   r	   r	   r
   r      s8   M





)r^   r   rB   r   rK   rb   Zpy310compatr   __all__r   r   dictfromkeysr3   r   r   rC   r+   rJ   r   r	   r	   r	   r
   <module>   s   1