File: //home/arjun/projects/env/lib/python3.10/site-packages/celery/utils/__pycache__/log.cpython-310.pyc
o
-we5" � @ sJ d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
m Z ddl m
Z
ddl mZ ddlmZ dd lmZ d
Zd Zd aej�d d �Zd
dhZed
� ZZ dd� Z dd� Z edd� �Z d-dd�Z dd� Z!dd� Zed�Z"ed �Z#d d � Z$d d � Z%G d!d"� d"ej&�Z'G d#d$� d$�Z(d%d&� Z)d'd(� Z*d)d*� Z+d.d+d,�Z,dS )/zLogging utilities.� N)�contextmanager)�AnyStr�Sequence)�
LOG_LEVELS)�
get_logger)�safe_str� )�colored) �ColorFormatter� LoggingProxy� base_logger�set_in_sighandler�
in_sighandlerr �get_task_logger�mlevel�get_multiprocessing_logger� reset_multiprocessing_loggerr F�MP_LOG�celeryz celery.taskc C s | a dS )z8Set flag signifiying that we're inside a signal handler.N)�_in_sighandler)�value� r �I/home/arjun/projects/env/lib/python3.10/site-packages/celery/utils/log.pyr
' s r
c
c s� � t � } ttjjj�� �t�d �g }|D ]+}z |jD ]}z|| vr+|j V | �
|� W q t y5 Y qw W q t y@ Y qw d S �N) �set�list�logging�Logger�manager�
loggerDict�values� getLogger�handlers�stream�add�AttributeError)�seen�loggers�l�handlerr r r �iter_open_logger_fds- s* �
�
� �� ��r* c c s* � t d� z
dV W t d� dS t d� w )z5Context that records that we are in a signal handler.TNF)r
r r r r r > s
�r �� c C sj | t � }}t|�D ]#}||kr dS ||v r td| j�d���|�|� |j}|s- dS q
td|� ���)NTzLogger z parents recursivezLogger hierarchy exceeds F)r �range� RuntimeError�namer$ �parent)r( �p�max�thisr& �_r r r �
logger_isaH s �
�r4 c C s t || �s| |_|S r )r4 r/ )�
parent_logger�logger_r r r �_using_logger_parent[ s
r7 c C s. t | �}tj||jfvr|turtt|�}|S )zGet logger by name.)� _get_loggerr �rootr/ r r7 )r. r( r r r r a s
r z
celery.workerc C s& | t v r td| �d���ttt| ��S )z#Get logger for task module by name.z Logger name z
is reserved!)�RESERVED_LOGGER_NAMESr- r7 � task_loggerr )r. r r r r m s r c C s | rt | tj�st| �� S | S )z$Convert level name/int to log level.)�
isinstance�numbers�Integralr �upper)�levelr r r r t s r c s` e Zd ZdZe� jZed ed ed ed d�Zd� fd d
� Z� fd d �Z � fd
d�Z
� Z S )r
z5Logging formatter that adds colors based on severity.�blue�yellow�red�magenta)�DEBUG�WARNING�ERROR�CRITICALNTc s t � �|� || _d S r )�super�__init__� use_color)�self�fmtrK �� __class__r r rJ � s
zColorFormatter.__init__c s& |r t |t�s t�� }t� �|�}|S r )r< �tuple�sys�exc_inforI �formatException)rL �ei�rrN r r rS � s z ColorFormatter.formatExceptionc s t � �|�}| j�|j�}|jdkrt�� n|j}|r| jrz&zt|t �r/t |t
|���W W S t
||��W W S t yD t
|� Y W S w t y~ } z.|j
dd�t|�|�}|_|_
zt � �|�W |||_
|_W Y d }~S |||_
|_w d }~ww t
|�S )Nr z <Unrepresentable {!r}: {!r}>)rI �format�colors�get� levelnamerR rQ rK r< �strr �UnicodeDecodeError� Exception�msg�type)rL �recordr] �color�einfo�exc�prev_msgrN r r rV � s.
�
�� ,�� zColorFormatter.format�NT) �__name__�
__module__� __qualname__�__doc__r �names�COLORSrW rJ rS rV �
__classcell__r r rN r r
{ s � r
c @ sd e Zd ZdZdZdZdZejZ e
� � Z ddd�Z
dd� Zd d
� Zd d � Zd
d� Zdd� Zdd� ZdS )r z�Forward file object to :class:`logging.Logger` instance.
Arguments:
logger (~logging.Logger): Logger instance to forward to.
loglevel (int, str): Log level to use when logging messages.
�wNFc C s* || _ t|p | j jp | j�| _| �� d S r )�loggerr r@ �loglevel�_safewrap_handlers)rL rm rn r r r rJ � s zLoggingProxy.__init__c s dd� � � fdd�| j jD �S )Nc S s G dd� dt j�}|� j| _d S )Nc @ s e Zd Zdd� ZdS )zRLoggingProxy._safewrap_handlers.<locals>.wrap_handler.<locals>.WithSafeHandleErrorc S s* z
t �d tj� W d S ty Y d S w r )� traceback� print_excrQ �
__stderr__�OSError)rL r_ r r r � handleError� s
�z^LoggingProxy._safewrap_handlers.<locals>.wrap_handler.<locals>.WithSafeHandleError.handleErrorN)re rf rg rt r r r r �WithSafeHandleError� s ru )r �Handlerrt )r) ru r r r � wrap_handler� s z5LoggingProxy._safewrap_handlers.<locals>.wrap_handlerc s g | ]}� |��qS r r )�.0�h�rw r r �
<listcomp>� s z3LoggingProxy._safewrap_handlers.<locals>.<listcomp>)rm r" �rL r rz r ro � s z LoggingProxy._safewrap_handlersc C s� t rt|�}t|tjd� t|�S t| jdd�rdS |rJ| jsJd| j_ z!t|��
d�}|r>| j � | j
|� t|�W d| j_ S W d| j_ dS d| j_ w dS )z Write message to logging object.)�file�recurse_protectionFr T�
)r r �printrQ rr �len�getattr�_thread�closedr~ �rstriprm �logrn )rL �data� safe_datar r r �write� s&
�
�zLoggingProxy.writec C s |D ]}| � |� qdS )z�Write list of strings to file.
The sequence can be any iterable object producing strings.
This is equivalent to calling :meth:`write` for each string.
N)r� )rL �sequence�partr r r �
writelines� s �zLoggingProxy.writelinesc C s d S r r r| r r r �flush� s zLoggingProxy.flushc C s
d| _ d S rd )r� r| r r r �close� s
zLoggingProxy.closec C s dS )zHere for file support.Fr r| r r r �isatty s zLoggingProxy.isattyr )re rf rg rh �moder. r� r rG rn � threading�localr� rJ ro r� r� r� r� r� r r r r r � s
r c C s, z ddl m} W | �� S ty Y dS w )z"Return the multiprocessing logger.r ��utilN)�billiardr� � ImportErrorr r� r r r r s ��r c C s> zddl m} W n
ty Y dS w t| d�r d| _dS dS )z$Reset multiprocessing logging setup.r r� �_loggerN)r� r� r� �hasattrr� r� r r r r s �
�r c C s, z ddl m} W | �� S ty Y d S w )Nr ��process)r� r� r� �current_processr� r r r r� s ��r� c C s" t t� dd �}|d ur|| S |S )N�index)r� r� )�baser� r r r �current_process_index% s r� )r+ )r )-rh r r= �osrQ r� rp �
contextlibr �typingr r � kombu.logr r r8 �kombu.utils.encodingr �termr �__all__�_process_awarer �environrX r r: r rm r
r* r r4 r7 r; �
worker_loggerr r � Formatterr
r r r r� r� r r r r �<module> sH
7U