U
    
HaZ#                     @   sN  d 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m	Z	m
Z
 ddlmZ ddlmZ ddlmZmZmZmZ ddlmZ eeZee ZzddlZee7 ZW n ek
r   ed Y nX zddlZee7 ZW n ek
r   ed	 Y nX d
d Zdd Zdd Zdd Z dd Z!dd Z"dddZ#dd Z$dddZ%dS )zUtilities related archives.
    N)IterableListOptional)ZipInfo)InstallationError)BZ2_EXTENSIONSTAR_EXTENSIONSXZ_EXTENSIONSZIP_EXTENSIONS)
ensure_dirzbz2 module is not availablezlzma module is not availablec                  C   s   t d} t |  | S )zBGet the current umask which involves having to set it temporarily.r   )osumask)mask r   A/tmp/pip-unpacked-wheel-tx790h60/pip/_internal/utils/unpacking.pycurrent_umask+   s    

r   c                 C   sh   |  d d} d| krHd| kr4| d| dk s<d| krH| ddS d| kr\| ddS | dgS d S )N/\    )lstripfindsplitpathr   r   r   split_leading_dir3   s    r   c                 C   sD   d}| D ]6}t |\}}|s" dS |dkr0|}q||kr dS qdS )zyReturns true if all the paths have the same leading path name
    (i.e., everything is in one subdirectory in an archive)NFT)r   )pathscommon_prefixr   prefixrestr   r   r   has_leading_dir@   s    r    c                 C   s0   t j| }t j|}t j||g}||kS )zL
    Return true if the absolute path of target is within the directory
    )r   r   abspathcommonprefix)	directorytargetabs_directory
abs_targetr   r   r   r   is_within_directoryP   s    r'   c                 C   s   t | dt  @ dB  dS )zx
    Make file present at path have execute for user/group/world
    (chmod +x) is no-op on windows per python docs
    i  I   N)r   chmodr   r   r   r   r   2set_extracted_file_to_default_mode_plus_executable\   s    r*   c                 C   s$   | j d? }t|o t|o |d@ S )N   r(   )external_attrboolstatS_ISREG)infomoder   r   r   zip_item_is_executablee   s    
r2   Tc              
   C   s  t | t| d}ztj|dd}t| o0|}| D ]}|j}|}|rXt	|d }t
j||}t
j|}	t||sd}
t|
| |||ds|drt | q:t |	 ||}z&t|d}t|| W 5 Q R X W 5 |  t|rt| X q:W 5 |  X d	S )
a  
    Unzip the file (with path `filename`) to the destination `location`.  All
    files are written based on system defaults and umask (i.e. permissions are
    not preserved), except that regular file members with any execute
    permissions (user, group, or world) have "chmod +x" applied after being
    written. Note that for windows, any execute changes using os.chmod are
    no-ops per the python docs.
    rbT)
allowZip64r   zQThe zip file ({}) has a file ({}) trying to install outside target directory ({})r   r   wbN)r   openclosezipfileZipFiler    namelistinfolistfilenamer   r   r   joindirnamer'   r   formatendswithr2   r*   shutilcopyfileobj)r<   locationflattenzipfpzipleadingr0   namefndirmessagefpdestfpr   r   r   
unzip_filem   s6    





rN   c                 C   s.  t | |  ds$|  dr*d}nL|  tr>d}n8|  trRd}n$|  drfd}ntd|  d	}tj| |d
d}zt
dd | D }| D ]p}|j}|rt|d }tj||}t||sd}t|| ||| rt | q| rbz||| W n> tk
r^ }	 ztd| |j|	 W Y qW 5 d}	~	X Y nX qz||}
W nB ttfk
r }	 ztd| |j|	 W Y qW 5 d}	~	X Y nX t tj| |
dk	stt|d}t|
| W 5 Q R X |
	  ||| |j d@ rt!| qW 5 |	  X dS )a  
    Untar the file (with path `filename`) to the destination `location`.
    All files are written based on system defaults and umask (i.e. permissions
    are not preserved), except that regular file members with any execute
    permissions (user, group, or world) have "chmod +x" applied after being
    written.  Note that for windows, any execute changes using os.chmod are
    no-ops per the python docs.
    z.gzz.tgzzr:gzzr:bz2zr:xzz.tarrz-Cannot determine compression type for file %szr:*zutf-8)encodingc                 S   s   g | ]
}|j qS r   )rH   ).0memberr   r   r   
<listcomp>   s     zuntar_file.<locals>.<listcomp>r   zQThe tar file ({}) has a file ({}) trying to install outside target directory ({})z/In the tar file %s the member %s is invalid: %sNr5   r(   )"r   lowerr@   r   r	   loggerwarningtarfiler6   r7   r    
getmembersrH   r   r   r   r=   r'   r   r?   isdirissym_extract_member	ExceptionextractfileKeyErrorAttributeErrorr>   AssertionErrorrA   rB   utimer1   r*   )r<   rC   r1   tarrG   rR   rI   r   rK   excrL   rM   r   r   r   
untar_file   st    





rd   c                 C   s   t j| } |dks,|  ts,t| rDt| || d d nR|dkslt	
| sl|  tt t rxt| | ntd| || td| d S )Nzapplication/zipz.whl)rD   zapplication/x-gzipzZCannot unpack file %s (downloaded from %s, content-type: %s); cannot detect archive formatz#Cannot determine archive format of )r   r   realpathrT   r@   r
   r8   
is_zipfilerN   rW   
is_tarfiler   r   r	   rd   rU   criticalr   )r<   rC   content_typer   r   r   unpack_file   s,    rj   )T)N)&__doc__loggingr   rA   r.   rW   r8   typingr   r   r   r   pip._internal.exceptionsr   pip._internal.utils.filetypesr   r   r	   r
   pip._internal.utils.miscr   	getLogger__name__rU   SUPPORTED_EXTENSIONSbz2ImportErrordebuglzmar   r   r    r'   r*   r2   rN   rd   rj   r   r   r   r   <module>   sB   
	
/U 