File: //usr/lib/python3.6/site-packages/glances/exports/__pycache__/glances_influxdb.cpython-36.pyc
3
Eo�c� � @ s\ d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z ddl
m Z G dd � d e�Z dS )
z0InfluxDB (up to InfluxDB 1.7.x) interface class.� N)�node)�logger)�
GlancesExport)�InfluxDBClient)�InfluxDBClientErrorc 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 gd
d d gd
�| _ | j snt d� t
� jd�d | _ | j� | _dS )z Init the InfluxDB export IF.)�config�argsNZhttp�influxdb�host�port�user�password�db�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_influxdb.pyr s zExport.__init__c C s� | j s
dS | jdk r(| jj� dkr(d}nd}y6t| j| j|d| j| j| jd�}dd� |j � D �}W n@ t
k
r� } z$t j d| j|f � t
jd � W Y dd}~X nX | j|kr�t jd
j|j�� nt j d | j � t
jd � |S ) z+Init the connection to the InfluxDB server.NZhttpsTF)r r �sslZ
verify_sslZusernamer Zdatabasec S s g | ] }|d �qS )�namer )�.0�ir r r �
<listcomp>J s z Export.init.<locals>.<listcomp>z-Cannot connect to InfluxDB database '%s' (%s)� z-Stats will be exported to InfluxDB server: {}z6InfluxDB database '%s' did not exist. Please create it)r r �lowerr r r r
r r Zget_list_databaser r Zcritical�sysr �info�formatZ_baseurl)r r r Z
get_all_db�er r r r 5 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 )r" �kr r r r$ c 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>j s z%Export._normalize.<locals>.<dictcomp>�keyr )r1 r �fields) �dict�zip�len�float� TypeError�
ValueError�strZ
parse_tagsr �popr �append) r r! �columns�points�retZ keys_listr4 r, r r )r0 r1 r �
_normalizeW s6
zExport._normalizec C s� | j dk r| j d | }t|�dkr6tjdj|�� ndy | jj| j|||�dd� W n4 tk
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 r7 r �debugr) r Z write_pointsrA � Exception)r r! r> r? r* r r r �export� s
$z
Export.export)NN) �__name__�
__module__� __qualname__�__doc__r r rA rE �
__classcell__r r )r r r s
"7r )
rI r' �platformr Zglances.loggerr Z glances.exports.glances_exportr r
r Zinfluxdb.clientr r r r r r �<module>
s