File: //var/opt/nydus/ops/primordial/log/__pycache__/analyticslogger.cpython-39.pyc
a
�,�h� � @ sl d dl mZmZ d dlmZmZmZmZmZ d dl m
Z
d dl m Z m
Z
mZ d dlmZ G dd� de�ZdS ) � )�Logger�
LoggerAdapter)�Any�Dict�MutableMapping�Optional�Tuple)�LogExtra)�EnvelopeVersion� EnvelopeRole�EnvelopeDataSchema)�Envelopec s& e Zd ZdZd&eee ee dd�� fdd�
Zed d�dd��Z ed d�d d
��Z
ed d�d d ��Z ed d�d
d��Z ee
dd�dd�Zeeee
f eeeee
f f d�dd�Zed d�dd�Zed d�dd�Zed d�dd �Zed d�d d �Zedd �d d!�Zed�d"d#�Zdd�d$d%�Z� ZS )'�AnalyticsLoggerzmA LoggerAdapter which specifies an interface for log calls and adds default/initial log data into the record.N)�logger�envelope_version� data_schema�returnc s� i | _ t� �|| j � |d u r(t� | _n t�|�| _|d u rFt� | _n t�|�| _| �t j
t | j�� | �t j t | j�� t
� | _| �t
jt | j�� d| _d S �NF)�extra�super�__init__r
�_envelope_version�
fromStringr � _data_schema� setExtraAttrr
�ENVELOPE_VERSION_FIELD�str�DATA_SCHEMA_FIELDr �role�
ROLE_FIELD�validate_data_schema)�selfr r r �� __class__� �O/opt/nydus/tmp/pip-target-wkfpz8uv/lib/python/primordial/log/analyticslogger.pyr s
zAnalyticsLogger.__init__)r c C s | � tj� | S �N)�_roler ZROLE_PERFORMANCE�r! r$ r$ r% � performance& s zAnalyticsLogger.performancec C s | � tj� | S r& )r' r ZROLE_DEVELOPMENTr( r$ r$ r% � development+ s zAnalyticsLogger.developmentc C s | � tj� | S r& )r' r ZROLE_BUSINESS_ANALYTICSr( r$ r$ r% �business_analytics0 s z"AnalyticsLogger.business_analyticsc C s
d| _ | S )NT)r r( r$ r$ r% �validate5 s zAnalyticsLogger.validate)�field�valuer c C s || j |< d S r& )r )r! r- r. r$ r$ r% r : s z AnalyticsLogger.setExtraAttr)�msg�kwargsr c C s� d|v r| � |d � d|v r,| �|d � d|v rB| �|d � | �� tjf i | j��}|jf i |�� i }|�� |d< | � � ||fS )Nr r r r )
�v�sr' �_validate_schemar Z getLogExtrar �set�get�_reset)r! r/ r0 ZlogextraZ
new_kwargsr$ r$ r% �process> s zAnalyticsLogger.process)r r c C s$ t �|�| _| �tjt| j�� | S )zAPermanently set envelope_version for the lifetime of this logger.)r
r r r r
r r �r! r r$ r$ r% r Q s z AnalyticsLogger.envelope_version)r r c C s$ t �|�| _| �tjt| j�� | S )z<Permanently set data_schema for the lifetime of this logger.)r r r r r
r r �r! r r$ r$ r% r W s zAnalyticsLogger.data_schemac C s | � tj|� | S )z9Temporary override envelope_version for one logging call.)r r
r r8 r$ r$ r% r1 ] s zAnalyticsLogger.vc C s | � tj|� | S )z4Temporary override data_schema for one logging call.)r r
r r9 r$ r$ r% r2 b s zAnalyticsLogger.s)r r c C s* t � �t j|�| _| �t jt| j�� d S r& )r r4 r r r r )r! r r$ r$ r% r' g s zAnalyticsLogger._rolec C s | j r| jd }dS dS d S )Nr T)r r r9 r$ r$ r% r3 l s
z AnalyticsLogger._validate_schemac C sT t � | _d| _i | _| �t jt| j�� | �tjt| j �� | �tj
t| j �� d S r ) r r r r r r r r
r r r r r( r$ r$ r% r6 t s zAnalyticsLogger._reset)NN)�__name__�
__module__� __qualname__�__doc__r r r r �propertyr) r* r+ r, r r r r r7 r r r1 r2 r' �boolr3 r6 �
__classcell__r$ r$ r" r% r s2 ��*r N)�loggingr r �typingr r r r r Zprimordial.log.extrar Zprimordial.log.envelopedatar
r r Zprimordial.log.enveloper
r r$ r$ r$ r% �<module> s