File: //var/opt/nydus/ops/primordial/log/__pycache__/__init__.cpython-39.pyc
a
�,�ha$ � @ s� d dl m Z d dlmZ d dlZd dlZd dlZd dlZd dlmZmZm Z m
Z
d dl m Z m
Z
mZmZ d dlmZ d dlZz d dlZW n ey� dZY n0 d dlZd dlmZ d dlmZ d d lmZ d d
l m Z m Z m Z d Z d Z!d
Z"ej#dej$dej%dej&dej'dej(dej)dej*diZ+e�,e-�Z.G dd� dej/�Z0G dd� dej/�Z1G dd� dej2�Z3G dd� de�Z4e-d dej5ej6ddfe7e7ee
e7e f e7e7ee7 ee7 dd �d d �Z8dS ) � )�datetime)�
SysLogHandlerN)�AF_INET�
SOCK_DGRAM� SOCK_STREAM�
SocketType)�Any�Dict�Optional�Tuple)�resource_stream)� iso8601_utc)�Envelope)�
ConfigLogData)� PerfLogExtra� DevLogExtra�
BALogExtrai � z@cee:z<7>z<6>z<4>z<3>z<2>z<1>c sT e Zd ZdZejejejejd�e e e e dd�� fdd�Z
e j e d�dd �Z
� ZS )
�"VerticalServiceStructuredFormatterz�Specialized structured logging formatter for HFS vertical services.
Takes a log record and configuration data and formats it according to the
structure in the Envelope class, returning a json-formatted log message.
)�log_type� log_sub_type�
datacenter� environmentN)r r r r �returnc sF t � j|i |�� t� �tj|��tj|��tj|��tj|�| _d S �N) �super�__init__r �setZ
TYPE_FIELDZSUB_TYPE_FIELDZDATACENTER_FIELDZENVIRONMENT_FIELD�
configLogData)�selfr r r r �args�kwargs�� __class__� �H/opt/nydus/tmp/pip-target-wkfpz8uv/lib/python/primordial/log/__init__.pyr 7 s ����z+VerticalServiceStructuredFormatter.__init__��recordr c C s t �|| j�}t|�� S r )r Z getEnveloper �COOKIEZgetJSON)r r' Z logenveloper$ r$ r% �formatG s z)VerticalServiceStructuredFormatter.format)�__name__�
__module__� __qualname__�__doc__r � DEFAULT_TYPE�DEFAULT_SUB_TYPE�DEFAULT_DATACENTER�DEFAULT_ENVIRONMENT�strr �logging� LogRecordr) �
__classcell__r$ r$ r" r% r 0 s ��r c s* e Zd ZdZejed�� fdd� Z� ZS )� VerticalServiceSystemdFormatterz4Systemd logging formatter for HFS vertical services.r& c s t �|jd�}|t� �|� S )N� )�
PREFIX_MAP�get�levelnor r) )r r' �prefixr" r$ r% r) O s z&VerticalServiceSystemdFormatter.format) r* r+ r, r- r3 r4 r2 r) r5 r$ r$ r" r% r6 L s r6 c @ s" e Zd ZdZejed�dd�ZdS )�#VerticalServiceRequestLoggingFilterz;A log filter for producing only Flask web request messages.r& c C s< t r8t �� r8t|d�s i |_t jj�d�}|r8||jd< dS )N�extraz x-request-id�
request_idT)�flaskZhas_request_context�hasattrr= �request�headersr9 )r r' r> r$ r$ r% �filterW s
z*VerticalServiceRequestLoggingFilter.filterN)r* r+ r, r- r3 r4 �boolrC r$ r$ r$ r% r< T s r< c s� e Zd ZdZddefejefee e
e e f e e dd�dd�Z
dd�� fdd � Zejdd
�d d �Zdd�d
d�Zdd�dd�Z� ZS )�HFSSyslogHandlerz�A customized Log handler for HFS syslogs.
Given a log record, emits a log message formatted according to the
:class:`VerticalServiceStructuredFormatter` class.
N� localhost)�ident�address�facility�socktyper c C sV |t tfvrtd��tj�| � |p0t�d�p0d| _|| _ || _
|| _ d| _ d| _
dS )z�Produce a new HFS syslog handler.
:param ident:
:param address:
:param facility:
:raises TypeError: For socket types other than TCP or UDP
z:HFSSyslogHandler only supports TCP and UDP AF_INET socketsZ SYSLOG_IDENT�pythonNF)r r � TypeErrorr3 �Handlerr �os�getenvrG rH rI rJ �socket�
unixsocket)r rG rH rI rJ r$ r$ r% r h s zHFSSyslogHandler.__init__)r c s | j du r| �� t� �� dS )z0Ensure socket object, as expected by base class.N)rP � _set_socketr �close�r r" r$ r% rS � s
zHFSSyslogHandler.closer& c
C s, �z| � | j| �|j��}tt�|jtj ��}t
� d�}|j durH|j nd}d}d|t
||| j||f }d}| �|�} |d | d }
| d } |
�d�| �d � } z>| jdu r�| �� | jtkr�| j�| | j� n | j�| � W n$ t�y | j�� d| _Y n0 W n t�y& | �|� Y n0 dS )
z�Emit a log record.
The record is formatted, and an RFC-compliant message is sent to the syslog server.
No structured data segment is created and message id is always nil ('-').
:param record: The record to be emitted.
�HOSTNAMENr �-z<%d>%d %s %s %s %d %s� �
�ascii�utf8) �encodePriorityrI � mapPriority� levelnamer
r �
fromtimestamp�created�pytz�utcrN rO �process�SYSLOG_FORMAT_VERSIONrG r) �encoderP �_connect_socketrJ r �sendtorH �sendall�OSErrorrS � Exception� handleError)
r r' �prio� timestamp�hostname�pidZ
message_id�headerZstructured_data�messageZ
ascii_partZ utf8_partZencoded_messager$ r$ r% �emit� s6
�
�
zHFSSyslogHandler.emitc C s0 | � � | j�d� | jtkr,| j�| j� d S )Ng �?)rR rP �
settimeoutrJ r �connectrH rT r$ r$ r% re � s
z HFSSyslogHandler._connect_socketc C s t � t| j�| _ d S r )rP r rJ rT r$ r$ r% rR � s z HFSSyslogHandler._set_socket)r* r+ r, r- �SYSLOG_DEFAULT_PORTr �LOG_USERr r
r2 r �intr r rS r3 r4 rq re rR r5 r$ r$ r" r% rE a s �
� 0rE z logging.yaml)�name�filename�override_loggersr r r r r c C s� |dur |nt j}|dur"|�� nt j}t�t| |��� �� �}|d � � D ]2}|�
d�dkrL||d< ||d< ||d< ||d< qL|dur�|d � |� t j
�|� t �d
� dS ) a+
Initialize logging, updating the existing log config loggers dict with an override_loggers dict.
It's possible, since we have many ways to run the system, that init_logging
be called more than once. We allow this, but warn if so.
:param name: package name where the logging YAML config file is located
:param filename: filename of the logging YAML config
:param override_loggers: a dict of overrides to the default log dictconfig
:param log_type:
:param log_sub_type:
:param datacenter:
:param environment:
N�
formattersz()z1primordial.log.VerticalServiceStructuredFormatterr r r r �loggersT)r r0 �upperr1 �yamlZ safe_loadr �read�decode�valuesr9 �updater3 �config�
dictConfig�captureWarnings) rw rx ry r r r r r� � formatterr$ r$ r% � init_logging� s �
r� )9r �logging.handlersr r3 �logging.configrN rP r r r r �typingr r r
r �
pkg_resourcesr r} r? � ImportErrorr` Zprimordial.timeutilsr
Zprimordial.log.enveloper Zprimordial.log.envelopedatar Zprimordial.log.extrar r r rt rc r( �DEBUG�INFO�NOTSET�WARNING�WARN�ERROR�CRITICAL�FATALr8 � getLoggerr* �LOG� Formatterr r6 �Filterr< rE r. r/ r2 r� r$ r$ r$ r% �<module> sd
�
d��