U
    EZhE.                     @   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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mZmZmZmZ d dlmZmZmZ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$m%Z% d dl&m'Z' d dl(m)Z) d dl*m+Z+ e, Z-e%dZ.G dd de/Z0ee1 e1e2dddZ3e j4d.e5ed dddZ6e5dddZ7G dd dej8Z9eG dd  d Z:G d!d" d"eZ;G d#d$ d$eZ<G d%d& d&ej=j>Z?G d'd( d(eZ@G d)d* d*eZAe5e2eeB e5d+d,d-ZCdS )/    N)	dataclass)TextIOWrapper)Filter)AnyClassVar	GeneratorListOptionalTextIOType)ConsoleConsoleOptionsConsoleRenderableRenderableTypeRenderResultRichCast)NullHighlighter)RichHandler)Segment)Style)VERBOSE	getLogger)WINDOWS)DEPRECATION_MSG_PREFIX)
ensure_dirzpip.subprocessorc                   @   s   e Zd ZdZdS )BrokenStdoutLoggingErrorzO
    Raised if BrokenPipeError occurs for the stdout stream while logging.
    N)__name__
__module____qualname____doc__ r    r    ?/tmp/pip-unpacked-wheel-36xvkur8/pip/_internal/utils/logging.pyr   #   s   r   )	exc_classexcreturnc                 C   s0   | t krdS tsdS t|to.|jtjtjfkS )NTF)BrokenPipeErrorr   
isinstanceOSErrorerrnoEINVALEPIPE)r"   r#   r    r    r!   _is_broken_pipe_error)   s
    r+      )NNN)numr$   c              	   c   s6   t  t_t j| 7  _z
dV  W 5 t j| 8  _X dS )zv
    A context manager which will cause the log output to be indented for any
    log messages emitted inside it.
    N)get_indentation
_log_stateindentation)r-   r    r    r!   
indent_log6   s
    
r1   r$   c                   C   s   t tddS )Nr0   r   )getattrr/   r    r    r    r!   r.   E   s    r.   c                       sZ   e Zd ZdZddeeedd fddZeeedd	d
Z	e
jed fddZ  ZS )IndentingFormatterz%Y-%m-%dT%H:%M:%SF)add_timestampN)argsr5   kwargsr$   c                   s   || _ t j|| dS )z
        A logging.Formatter that obeys the indent_log() context manager.

        :param add_timestamp: A bool indicating output lines should be prefixed
            with their record's timestamp.
        N)r5   super__init__)selfr5   r6   r7   	__class__r    r!   r9   L   s    zIndentingFormatter.__init__)	formattedlevelnor$   c                 C   s.   |t jk rdS |trdS |t jk r*dS dS )zv
        Return the start of the formatted log message (not counting the
        prefix to add to each line).
         z	WARNING: zERROR: )loggingWARNING
startswithr   ERROR)r:   r=   r>   r    r    r!   get_message_start[   s    


z$IndentingFormatter.get_message_startrecordr$   c                    sl   t  |}| ||j}|| }d | jr<| | d  dt  7  d fdd|dD }|S )z
        Calls the standard formatter, but will indent all of the log message
        lines by our current indentation level.
        r?    c                    s   g | ]} | qS r    r    ).0lineprefixr    r!   
<listcomp>x   s     z-IndentingFormatter.format.<locals>.<listcomp>T)	r8   formatrD   r>   r5   
formatTimer.   join
splitlines)r:   rF   r=   message_startr;   rJ   r!   rM   k   s    zIndentingFormatter.format)r   r   r   default_time_formatr   boolr9   strintrD   r@   	LogRecordrM   __classcell__r    r    r;   r!   r4   I   s   r4   c                   @   s0   e Zd ZU eed< eed< eeedddZ	dS )IndentedRenderable
renderableindent)consoleoptionsr$   c                 c   sJ   | | j|}t|}|D ](}td| j V  |E d H  tdV  qd S )NrG   
)renderrY   r   Zsplit_linesrZ   )r:   r[   r\   segmentslinesrI   r    r    r!   __rich_console__   s    

z#IndentedRenderable.__rich_console__N)
r   r   r   r   __annotations__rU   r   r   r   ra   r    r    r    r!   rX   |   s   
 rX   c                   @   s   e Zd ZddddZdS )
PipConsoleNr2   c                 C   s   t  d d S N)r%   r:   r    r    r!   on_broken_pipe   s    zPipConsole.on_broken_pipe)r   r   r   rf   r    r    r    r!   rc      s   rc   c                       sl   e Zd ZU g Zeeee   ed< ee	 e
dd fddZejddddZejdd fd	d
Z  ZS )RichPipStreamHandlerKEYWORDSN)streamno_colorr$   c                    s&   t  jt||dddddt d d S )NT)filerj   Z	soft_wrapF)r[   Z	show_timeZ
show_levelZ	show_pathZhighlighter)r8   r9   rc   r   )r:   ri   rj   r;   r    r!   r9      s    zRichPipStreamHandler.__init__rE   c                 C   s   d }t |ddrVt|jts t|j\}t|tttfsFt| dt|t	 d}nN| 
|}| ||}|jd k	r|jtjkrtdd}n|jtjkrtdd}z| jj|dd|d	 W n tk
r   | | Y nX d S )
NZrichFz is not rich-console-renderable)rZ   Zred)colorZyellowignore)ZoverflowZcropstyle)r3   r&   r6   tupleAssertionErrorr   r   rT   rX   r.   rM   Zrender_messager>   r@   rC   r   rA   r[   print	ExceptionhandleError)r:   rF   rn   Zrich_renderablerY   messager    r    r!   emit   s2      


zRichPipStreamHandler.emitc                    sF   t  dd \}}|r:|r:| jjt jkr:t||r:t t |S )z1Called when logging is unable to log some output.Nr,   )	sysexc_infor[   rk   stdoutr+   r   r8   rs   )r:   rF   r"   r#   r;   r    r!   rs      s    z RichPipStreamHandler.handleError)r   r   r   rh   r   r	   r   rT   rb   r
   rS   r9   r@   rV   ru   rs   rW   r    r    r;   r!   rg      s   

rg   c                       s"   e Zd Zed fddZ  ZS )BetterRotatingFileHandlerr2   c                    s   t tj| j t  S rd   )r   ospathdirnamebaseFilenamer8   _openre   r;   r    r!   r~      s    zBetterRotatingFileHandler._open)r   r   r   r   r~   rW   r    r    r;   r!   ry      s   ry   c                   @   s.   e Zd ZeddddZejedddZdS )MaxLevelFilterN)levelr$   c                 C   s
   || _ d S rd   )r   )r:   r   r    r    r!   r9      s    zMaxLevelFilter.__init__rE   c                 C   s   |j | jk S rd   )r>   r   r:   rF   r    r    r!   filter   s    zMaxLevelFilter.filter)	r   r   r   rU   r9   r@   rV   rS   r   r    r    r    r!   r      s   r   c                       s*   e Zd ZdZejed fddZ  ZS )ExcludeLoggerFilterzQ
    A logging Filter that excludes records from a logger (or its children).
    rE   c                    s   t  | S rd   )r8   r   r   r;   r    r!   r      s    zExcludeLoggerFilter.filter)	r   r   r   r   r@   rV   rS   r   rW   r    r    r;   r!   r      s   r   )	verbosityrj   user_log_filer$   c                 C   s  | dkrt j}nD| dkrt}n6| dkr.t j}n&| dkr>t j}n| dkrNt j}nt j}t |}|dk	}|rt|}d}nd}|}|d	krd
nd}ddd}	ddd}
dddg|rdgng  }t j	dddt jddt
jddt
jddtddtdddd ||
d! ||	d" d#d$gd%d&d
|
d! ||	d' d#gd%d&||
d! |	d' |d(gd%d)d|
d* |d+dd,d-d.||d/d0d1|iid2 |S )3znConfigures and sets up all of the logging

    Returns the requested logging level, as its integer value.
    r,      NDEBUGz	/dev/null)INFOrC   rA   zext://sys.stdoutzext://sys.stderr)rx   stderrz0pip._internal.utils.logging.RichPipStreamHandlerz5pip._internal.utils.logging.BetterRotatingFileHandler)ri   rk   r[   console_errorsconsole_subprocessuser_logFz*pip._internal.utils.logging.MaxLevelFilter)()r   zlogging.Filter)r   namez/pip._internal.utils.logging.ExcludeLoggerFilter)exclude_warningsrestrict_to_subprocessexclude_subprocessz%(message)s)r   rM   T)r   rM   r5   )rZ   indent_with_timestampri   rx   r   r   rZ   )r   classrj   ri   filters	formatterr   r   )r   r   ri   rj   r   r   rk   zutf-8r   )r   r   filenameencodingdelayr   )r[   r   r   r   )r   handlerszpip._vendorr   )versiondisable_existing_loggersr   
formattersr   rootloggers)r@   r   r   rA   rC   CRITICALr   getLevelNameconfig
dictConfigsubprocess_loggerr   r4   )r   rj   r   level_numberr   include_user_logadditional_log_file
root_levelvendored_log_levellog_streamshandler_classesr   r    r    r!   setup_logging   s    
		%
Ir   )r,   )D
contextlibr(   r@   logging.handlersrz   rv   	threadingZdataclassesr   ior   r   typingr   r   r   r   r	   r
   r   Zpip._vendor.rich.consoler   r   r   r   r   r   Zpip._vendor.rich.highlighterr   Zpip._vendor.rich.loggingr   Zpip._vendor.rich.segmentr   Zpip._vendor.rich.styler   Zpip._internal.utils._logr   r   pip._internal.utils.compatr   pip._internal.utils.deprecationr   pip._internal.utils.miscr   localr/   r   rr   r   BaseExceptionrS   r+   contextmanagerrU   r1   r.   	Formatterr4   rX   rc   rg   r   RotatingFileHandlerry   r   r   rT   r   r    r    r    r!   <module>   sF   $ 3<