File: //opt/alt/python37/lib/python3.7/site-packages/fluent/__pycache__/handler.cpython-37.pyc
B
��f%" � @ sL d dl Z d dlZd dlZd dlmZ G dd� dej�ZG dd� dej�ZdS )� N)�senderc sv e Zd ZdZd� fdd� Z� fdd �Zd
d � Zd d
� Zdd� Z� fdd�Z dd� Z
dd� Z dd� Z e
dd� �Z� ZS )�FluentRecordFormattera6 A structured formatter for Fluent.
Best used with server storing data in an ElasticSearch cluster for example.
:param fmt: a dict or a callable with format string as values to map to provided keys.
If callable, should accept a single argument `LogRecord` and return a dict,
and have a field `usesTime` that is callable and return a bool as would
`FluentRecordFormatter.usesTime`
:param datefmt: strftime()-compatible date/time format string.
:param style: '%', '{' or '$' (used only with Python 3.2 or above)
:param fill_missing_fmt_key: if True, do not raise a KeyError if the format
key is not found. Put None if not found.
:param format_json: if True, will attempt to parse message as json. If not,
will use message as-is. Defaults to True
:param exclude_attrs: switches this formatter into a mode where all attributes
except the ones specified by `exclude_attrs` are logged with the record as is.
If `None`, operates as before, otherwise `fmt` is ignored.
Can be an iterable.
N�%FTc s t � �d |� |dkrHtjdddd�ftjdddd�fd �| \| _}nd | _d
d d d�}|d k r�t|�| _d | _| j | _
t � j | _ nPd | _|s�|| _| j | _
| j
| _ n.t|�r�|| _
|j | _ n|| _| j | _
| j
| _ |r�| j| _n| j| _t�� | _|| _d S )
Nr z
{hostname}z{name}z{module})Zsys_hostZsys_name�
sys_modulez ${hostname}z${name}z ${module})�{�$z %(hostname)sz%(name)sz
%(module)s)�super�__init__�logging�StrFormatStyle�StringTemplateStyle� _FluentRecordFormatter__style�set�
_exc_attrs� _fmt_dict�_format_by_exclusion�
_formatter�usesTime�_format_by_dict�_format_by_dict_uses_time�callable�_format_msg_json� _format_msg�_format_msg_default�socket� gethostname�hostname�fill_missing_fmt_key)�self�fmt�datefmt�styler Z format_jsonZ
exclude_attrsZbasic_fmt_dict)� __class__� �?/opt/alt/python37/lib/python3.7/site-packages/fluent/handler.pyr sH
z FluentRecordFormatter.__init__c s. t � �|� | j|_| �|�}| �||� |S )N)r �formatr r � _structuring)r �record�data)r" r# r$ r% ` s
z FluentRecordFormatter.formatc C s dS )zTThis method is substituted on construction based on settings for performance reasonsNr# )r r# r# r$ r l s z FluentRecordFormatter.usesTimec C sR |j }t|t�r | �||� n0t|t�r>| �|| �||�� n| �|d|i� dS )a| Melds `msg` into `data`.
:param data: dictionary to be sent to fluent server
:param msg: :class:`LogRecord`'s message to add to `data`.
`msg` can be a simple string for backward compatibility with
:mod:`logging` framework, a JSON encoded string or a dictionary
that will be merged into dictionary generated in :meth:`format.
�messageN)�msg�
isinstance�dict�_add_dic�strr )r r( r' r* r# r# r$ r& o s
z"FluentRecordFormatter._structuringc C sN y,t �t|��}t|t�r |S | �||�S W n tk
rH | �||�S X d S )N)�json�loadsr. r+ r, r �
ValueError)r r' r* Zjson_msgr# r# r$ r � s
z&FluentRecordFormatter._format_msg_jsonc s dt � �|�iS )Nr) )r r% )r r' r* )r" r# r$ r � s z)FluentRecordFormatter._format_msg_defaultc C s2 i }x(|j �� D ]\}}|| jkr|||< qW |S )N)�__dict__�itemsr )r r' r( �key�valuer# r# r$ r � s
z*FluentRecordFormatter._format_by_exclusionc C s� i }xv| j �� D ]h\}}y&| jr2| �|��|�}n
||j }W n0 tk
rn } zd }| js^|�W d d }~X Y nX |||< qW |S )N)r r3 r
r% r2 �KeyErrorr )r r' r( r4 r5 �excr# r# r$ r � s z%FluentRecordFormatter._format_by_dictc s0 | j r| j j� nd� t� fdd�| j�� D ��S )Nz
%(asctime)c s g | ]}|� � �d k�qS )r )�find)�.0r5 )�searchr# r$ �
<listcomp>� s zCFluentRecordFormatter._format_by_dict_uses_time.<locals>.<listcomp>)r
�asctime_search�anyr �values)r r# )r: r$ r � s
z/FluentRecordFormatter._format_by_dict_uses_timec C s, x&|� � D ]\}}t|t�r
|| |< q
W d S )N)r3 r+ r. )r( Zdicr4 r5 r# r# r$ r- � s
z FluentRecordFormatter._add_dic)NNr FTN)�__name__�
__module__� __qualname__�__doc__r r% r r& r r r r r � staticmethodr- �
__classcell__r# r# )r" r$ r s <
r c s^ e Zd ZdZddd�Zd d
� Zed d � �Zd
d� Zdd� Z � fdd�Z
dd� Z dd� Z � Z
S )�
FluentHandlerz%
Logging Handler for fluent.
� localhost�^ � @FNc
K sL || _ || _|| _|| _|| _|| _|| _|| _| | _d | _ t
j � | � d S )N)
�tag�_host�_port�_timeout�_verbose�_buffer_overflow_handler�_msgpack_kwargs�_nanosecond_precision�_kwargs�_senderr
�Handlerr )
r rI �host�port�timeout�verbose�buffer_overflow_handler�msgpack_kwargs�nanosecond_precision�kwargsr# r# r$ r � s zFluentHandler.__init__c C s t jS )N)r Z FluentSender)r r# r# r$ �getSenderClass� s z FluentHandler.getSenderClassc C sF | j d kr@| jf | j| j| j| j| j| j| j| j d�| j
��| _ | j S )N)rI rT rU rV rW rX rY rZ ) rR �getSenderInstancerI rJ rK rL rM rN rO rP rQ )r r# r# r$ r � s
zFluentHandler.senderc
K s( | � � }
|
|f|||||||d�| ��S )N)rT rU rV rW rX rY rZ )r\ ) r rI rT rU rV rW rX rY rZ r[ Z sender_classr# r# r$ r] � s z FluentHandler.getSenderInstancec C s6 | � |�}| j}|�d |jr(t�|j�nt|j�|�S )N)r% r Zemit_with_timerZ Z EventTime�created�int)r r' r( rR r# r# r$ �emit� s
zFluentHandler.emitc sH | � � z0z | jd k r&| j�� d | _W d t� �� X W d | �� X d S )N)�acquirerR �closer �release)r )r" r# r$ rb s
zFluentHandler.closec C s | S )Nr# )r r# r# r$ � __enter__ s zFluentHandler.__enter__c C s | � � d S )N)rb )r �exc_type�exc_val�exc_tbr# r# r$ �__exit__ s zFluentHandler.__exit__)rF rG rH FNNF)r? r@ rA rB r r\ �propertyr r] r` rb rd rh rD r# r# )r" r$ rE � s
rE ) r/ r
r Zfluentr � Formatterr rS rE r# r# r# r$ �<module> s ,