File: //usr/lib/python3.6/site-packages/glances/exports/__pycache__/glances_influxdb2.cpython-36.pyc
3
Eo�cL � @ sT d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z m
Z
G dd� de�Z dS ) z1InfluxDB (from to InfluxDB 1.8+) interface class.� N)�node)�logger)�
GlancesExport)�InfluxDBClient� WriteOptionsc s: e Zd ZdZd � fdd� Zdd� Zdd� Zd d
� Z� ZS ) �Exportz.This class manages the InfluxDB export module.Nc s� t t| �j||d� d| _d| _d| _d| _d| _d| _d| _ | j
dddddd d
d gd d
dgd�| _ | j srt d� t
� jd�d | _ | j� | _dS )z Init the InfluxDB export IF.)�config�argsNZhttpZ influxdb2�host�port�userZpassword�org�bucket�token�protocol�prefix�tags)Z mandatoriesZoptionsz!Missing INFLUXDB version 1 config�.r )�superr �__init__r
r r r r r �hostnameZ load_conf�
export_enable�exitr �split�init�client)�selfr r )� __class__� �'/usr/lib/python3.6/glances_influxdb2.pyr s zExport.__init__c C s� | j s
dS dj| j| j| j�}yt|dd| j| jd�}W n> tk
rv } z"t j
d||f � t j d� W Y dd}~X n X t j
dj|j� j|j� j�� |jtdd d
d d d
dd�d�}|S )z+Init the connection to the InfluxDB server.Nz
{}://{}:{}F)�urlZ enable_gzipZ
verify_sslr
r z+Cannot connect to InfluxDB server '%s' (%s)� z,Connected to InfluxDB server version {} ({})i� i' i� i� � i0u )Z
batch_sizeZflush_intervalZjitter_intervalZretry_intervalZ max_retriesZmax_retry_delayZexponential_base)Z
write_options)r �formatr r
r r r
r � Exceptionr Zcritical�sysr �infoZhealth�version�messageZ write_apir )r r r �eZ write_clientr r r r 6 s( z Export.initc sF g }t t||��� dd� |D �}t|�dkr2dg}�x |D �]��dk r^� �fdd�� D �}n� }xv|D ]n}|| dkrzqhyt|| �||< W qh ttfk
r� yt|| �||< W n ttfk
r� Y nX Y qhX qhW | j| j�}d|ko�|d |k�r"t||d �||d < |j |d � | j
|d< |j |||d �� q:W |S )
z`Normalize data for the InfluxDB's data model.
:return: a list of measurements.
c S s$ g | ] }|j d �r|jd�d �qS )z.keyr r )�endswithr )�.0�kr r r �
<listcomp>a s z%Export._normalize.<locals>.<listcomp>r Nc s4 i | ],}|j d j���r� | |jd j��d��qS )z{}.� )�
startswithr# �replace)r+ r, )� data_dict� measurementr r �
<dictcomp>h s z%Export._normalize.<locals>.<dictcomp>�keyr )r2 r �fields) �dict�zip�len�float� TypeError�
ValueError�strZ
parse_tagsr �popr �append) r �name�columns�points�retZ keys_listr5 r, r r )r1 r2 r �
_normalizeU s6
zExport._normalizec C s� | j dk r| j d | }t|�dkr6tjdj|�� nly&| jj| j| j| j |||�dd� W n4 t
k
r� } ztjdj||�� W Y dd}~X nX tjdj|�� dS ) z(Write the points to the InfluxDB server.Nr r z(Cannot export empty {} stats to InfluxDB�s)Ztime_precisionz'Cannot export {} stats to InfluxDB ({})zExport {} stats to InfluxDB) r r8 r �debugr# r �writer r
rC r$ )r r? r@ rA r) r r r �export� s
&$z
Export.export)NN) �__name__�
__module__� __qualname__�__doc__r r rC rG �
__classcell__r r )r r r s
7r ) rK r% �platformr Zglances.loggerr Z glances.exports.glances_exportr Zinfluxdb_clientr r r r r r r �<module>
s