
    h              	          d 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 ddlmZ ddlmZ dd	lmZ dd
lmZ ddlmZmZmZ ddlmZmZmZ  ej@                  e!      Z"dedee#   fdZ$dedede%dee&   fdZ'de%de%fdZ(de%de%de%fdZ)dedede%fdZ*dededefdZ+ G d d      Z, G d d      Z-y)z)Download files with progress indicators.
    N)IterableOptionalTuple)Response)get_download_progress_renderer)NetworkConnectionError)PyPI)Link)is_from_cache)
PipSession)HEADERSraise_for_statusresponse_chunks)format_sizeredact_auth_from_urlsplitextrespreturnc                 h    	 t        | j                  d         S # t        t        t        f$ r Y y w xY w)Nzcontent-length)intheaders
ValueErrorKeyError	TypeError)r   s    GD:\EasyAligner\venv\Lib\site-packages\pip/_internal/network/download.py_get_http_response_sizer      s5    4<< 0122), s    11linkprogress_barc                    t        |       }|j                  t        j                  k(  r|j                  }n|j
                  }t        |      }|r| dt        |       d}t        |       rt        j                  d|       nt        j                  d|       t        j                         t        j                  kD  rd}nt        |       rd}n|sd}n
|dkD  rd}nd}t        |       }|s|S t        ||      } ||      S )	Nz ()zUsing cached %szDownloading %sFTi   )bar_typesize)r   netlocr	   file_storage_domainshow_urlurl_without_fragmentr   r   r   loggerinfogetEffectiveLevelloggingINFOr   r   )	r   r   r   total_lengthurl
logged_urlshow_progresschunksrenderers	            r   _prepare_downloadr2      s    
 +40L{{d...mm''%c*J"|2k,&?%@B
T%z2$j1!GLL0	t			$T"F-|,WHF    filenamec                 @    t         j                  j                  |       S )zJ
    Sanitize the "filename" value from a Content-Disposition header.
    )ospathbasename)r4   s    r   sanitize_content_filenamer9   I   s     77H%%r3   content_dispositiondefault_filenamec                     t         j                  j                         }| |d<   |j                  d      }|rt	        t        |            }|xs |S )z
    Parse the "filename" value from a Content-Disposition header, and
    return the default filename if the result is empty.
    content-typer4   )emailmessageMessage	get_paramr9   str)r:   r;   mr4   s       r   parse_content_dispositionrD   P   sM    
 	A+An{{:&H -S];'''r3   c                    |j                   }| j                  j                  d      }|rt        ||      }t	        |      d   }|s6t        j                  | j                  j                  dd            }|r||z  }|sL|j                  | j                  k7  r3t        j                  j	                  | j                        d   }|r||z  }|S )zoGet an ideal filename from the given HTTP response, falling back to
    the link filename if not provided.
    zcontent-disposition   r=    )
r4   r   getrD   r   	mimetypesguess_extensionr-   r6   r7   )r   r   r4   r:   exts        r   _get_http_response_filenamerL   _   s     }}H,,**+@A,-@(K!(+A.C''(8(8(LMOH488txx'ggtxx(+OHOr3   sessionc                     |j                   j                  dd      d   }| j                  |t        d      }t	        |       |S )N#rF   r   T)r   stream)r-   splitrH   r   r   )rM   r   
target_urlr   s       r   _http_get_downloadrS   t   s<    Q'*J;;z74;@DTKr3   c                   >    e Zd ZdededdfdZdededeeef   fdZy)	
DownloaderrM   r   r   Nc                      || _         || _        y N_session_progress_barselfrM   r   s      r   __init__zDownloader.__init__|       
  )r3   r   locationc                     	 t        | j                  |      }t        ||      }t        j                  j                  ||      }t        ||| j                        }t        |d      5 }|D ]  }	|j                  |	        	 ddd       |j                  j!                  dd      }
||
fS # t        $ r?}|j                  J t        j                  d|j                  j                  |        d}~ww xY w# 1 sw Y   txY w)z.Download the file given by link into location.NHTTP error %s while getting %swbContent-TyperG   rS   rY   r   responser'   criticalstatus_coderL   r6   r7   joinr2   rZ   openwriter   rH   )r\   r   r_   r   er4   filepathr0   content_filechunkcontent_types              r   __call__zDownloader.__call__   s    	%dmmT:D /tT:77<<(3"4t/A/AB(D!\""5)   " ||'';%% & 	::)))OO0!**2H2H$ 	 "!s#   B) 'C4)	C12:C,,C14C=)	__name__
__module____qualname__r   rB   r]   r
   r   rp    r3   r   rU   rU   {   sE    ** * 
	*&T &S &U38_ &r3   rU   c                   T    e Zd ZdededdfdZdee   dedeeeeeef   f      fdZ	y)	BatchDownloaderrM   r   r   Nc                      || _         || _        y rW   rX   r[   s      r   r]   zBatchDownloader.__init__   r^   r3   linksr_   c              #      K   |D ]  }	 t        | j                  |      }t        ||      }t        j                  j                  ||      }t        ||| j                        }t        |d      5 }	|D ]  }
|	j                  |
        	 ddd       |j                  j!                  dd      }|||ff  y# t        $ r?}|j                  J t        j                  d|j                  j                  |        d}~ww xY w# 1 sw Y   {xY ww)z0Download the files given by links into location.Nra   rb   rc   rG   rd   )r\   rx   r_   r   r   rk   r4   rl   r0   rm   rn   ro   s               r   rp   zBatchDownloader.__call__   s      D	)$--> 34>Hww||Hh7H&tT43E3EFFh%#E &&u- $ &  <<++NB?L<000)  * zz---4JJ**
  &%s@   DB7AD.D/D7	C? :C::C??DD	D)
rq   rr   rs   r   rB   r]   r   r
   r   rp   rt   r3   r   rv   rv      s[    ** * 
	*1d^1/21	%eCHo-.	/1r3   rv   ).__doc__email.messager>   r*   rI   r6   typingr   r   r   pip._vendor.requests.modelsr   pip._internal.cli.progress_barsr   pip._internal.exceptionsr   pip._internal.models.indexr	   pip._internal.models.linkr
   pip._internal.network.cacher   pip._internal.network.sessionr   pip._internal.network.utilsr   r   r   pip._internal.utils.miscr   r   r   	getLoggerrq   r'   r   r   rB   bytesr2   r9   rD   rL   rS   rU   rv   rt   r3   r   <module>r      s      	 , , 0 J ; + * 5 4 R R P P			8	$( x} '
'
' ' e_	'T& & &(3 (# (RU (h d s *
 $ 8 & &>!1 !1r3   