o

�N�gB�@szdZd
dl
Z
d
dlmZmZ
d
dlZdZGdd�de
j�ZGdd�de
j	�Z
d	��D]Ze
e
j	e�Zee
ed
d��
q*dS)aypyzmq logging handlers.

This mainly defines the PUBHandler object for publishing logging messages over
a zmq.PUB socket.

The PUBHandler can be used with the regular logging module, as in::

    >>> import logging
    >>> handler = PUBHandler('tcp://127.0.0.1:12345')
    >>> handler.root_topic = 'foo'
    >>> logger = logging.getLogger('foobar')
    >>> logger.setLevel(logging.DEBUG)
    >>> logger.addHandler(handler)

After this point, all messages logged by ``logger`` will be published on the
PUB socket.

Code adapted from StarCluster:

    https://github.com/jtriley/StarCluster/blob/StarCluster-0.91/starcluster/logger.py
�N)�Optional�Union�::c	@s�eZ
dZUd
Zejed<ejed<		ddee	ejfde
ejde	d	dfd
d�Zed	e	fdd
��
Z
e
jde	fdd
��
Z
de	fdd�Zejf
dd�
Zdd�Zdd�ZdS)�
PUBHandlera|A basic logging handler that emits log messages through a PUB socket.

    Takes a PUB socket already bound to interfaces or an interface to bind to.

    Example::

        sock = context.socket(zmq.PUB)
        sock.bind('inproc://log')
        handler = PUBHandler(sock)

    Or::

        handler = PUBHandler('inproc://loc')

    These are equivalent.

    Log messages handled by this handler are broadcast with ZMQ topics
    ``this.root_topic`` comes first, followed by the log level
    (DEBUG,INFO,etc.), followed by any additional subtopics specified in the
    message by: log.debug("subtopic.subsub::the real message")
    �ctx�socketN��interface_or_socket�context�
root_topic�returncCs�tj
�|�

||_tjt�d
�
tjt�d�
tjt�d
�
tjt�d�
tj	t�d
�
i|_
t|
tj
�r:|
|_|jj|_dS|p?t��|_|j�tj�
|_|j�|
�

dS)Nz4%(levelname)s %(filename)s:%(lineno)d - %(message)s
z%(message)s
zC%(levelname)s %(filename)s:%(lineno)d - %(message)s - %(exc_info)s
)�logging�Handler�__init__r�DEBUG�	Formatter�INFO�WARN�ERROR�CRITICAL�
formatters�
isinstance�zmq�Socketrr
r�ContextZPUB�bind)�selfr	r
r�r�D/opt/saltstack/salt/lib/python3.10/site-packages/zmq/log/handlers.pyr<s,

�

�
�
��



zPUBHandler.__init__c


Cs|jS�
N)
�_root_topic)
rrrrr[szPUBHandler.root_topic�valuecCs|�|
�

dSr)
�setRootTopic)rr!rrrr_scCst|
t
�r
|
�d
�
}
|
|_dS)a|Set the root topic for this handler.

        This value is prepended to all messages published by this handler, and it
        defaults to the empty string ''. When you subscribe to this socket, you must
        set your subscription to an empty string, or to at least the first letter of
        the binary representation of this string to ensure you receive any messages
        from this handler.

        If you use the default empty string root topic, messages will begin with
        the binary representation of the log level string (INFO, WARN, etc.).
        Note that ZMQ SUB sockets can have multiple subscriptions.
        �utf8N)r�bytes�decoder )rrrrrr"cs




zPUBHandler.setRootTopiccCs6|tj
kr|j��D]}|
|j|<q
d
S|
|j|<d
S)z�Set the Formatter for this handler.

        If no level is provided, the same format is used for all levels. This
        will overwrite all selective formatters set in the object constructor.
        N)r
�NOTSETr�keys)r�fmt�levelZ	fmt_levelrrr�setFormatterts



�zPUBHandler.setFormattercCs|j|
j
�|
�
S)
zFormat a record.)r�levelno�format)r�recordrrrr,�szPUBHandler.formatcCs�z|
j�
td
�\}|
_Wnty


d}Yn
wz
|�|
�
�d�
}Wnty1


|�|
�

YdSwg}|jr=|�	|j�

|�	|
j
�

|rJ|�	|�

d�|�
�d�
}|j�
||g�

dS)z Emit a log message on my socket.�
rr#N�
.)�msg�split�TOPIC_DELIM�
ValueErrorr,�encode�	Exception�handleErrorr�append�	levelname�joinrZsend_multipart)rr-�topicZbmsgZ
topic_listZbtopicrrr�emit�s&


�




�


zPUBHandler.emit)Nr)�__name__�
__module__�__qualname__�__doc__rr�__annotations__rr�strrr�propertyr�setterr"r
r&r*r,r;rrrrr"s.





����
�

rc@seZ
dZd
Zdd�ZdS)�TopicLoggera 
A simple wrapper that takes an additional argument to log methods.

    All the regular methods exist, but instead of one msg argument, two
    arguments: topic, msg are passed.

    That is::

        logger.debug('msg')

    Would become::

        logger.debug('topic.sub', 'msg')
    cOs,tj
j||
|�d
|��g|�
Ri|�
�

dS)a
Log 'msg % args' with level and topic.

        To pass exception information, use the keyword argument exc_info
        with a True value::

            logger.log(level, "zmq.fun", "We have a %s",
                    "mysterious problem", exc_info=1)
        rN)r
�Logger�log�rr)r:r0�args�kwargsrrrrF�s,	zTopicLogger.logN)r<r=r>r?rFrrrrrD�s
rDz'debug warn warning error critical fatalcOs"t||
|t
|g|�
Ri|�
�
Sr)�methr2rGrrr�<lambda>�s
�
�
�rK)r?r
�typingrrrr2rrrErDr1�name�getattrrJ�setattrrrrr�<module>
s~




��