File: //usr/lib/python3.6/site-packages/glances/exports/__pycache__/glances_csv.cpython-36.pyc
3
Eo�c� � @ sl d Z ddlZddlZddlZddlZddlmZmZm Z ddl
m Z ddl m
Z
G dd� de
�Zdd � ZdS )
zCSV interface class.� N)�PY3�iterkeys�
itervalues)�logger)�
GlancesExportc s2 e Zd ZdZd � fdd� Zdd� Zdd� Z� ZS )
�Exportz)This class manages the CSV export module.Nc s6 t t| �j||d� |j| _tjj| j� s2|jr>d}d| _ nvd}y t
| jd�| _ t j
| j �}W n< tk
r� } z tjdj|�� tjd� W Y dd}~X nX t|d�| _ | j j� y t
| j|�| _ t j| j �| _W n> tk
�r } z tjdj|�� tjd� W Y dd}~X nX tjd j| j�� d
| _d
| _dS ) zInit the CSV export IF.)�config�args�wN�a�rz!Cannot open existing CSV file: {}� z Cannot create the CSV file: {}z Stats exported to CSV file: {}T)�superr �__init__Zexport_csv_fileZ csv_filename�os�path�isfileZexport_csv_overwrite�
old_header�
open_csv_file�csv_file�csv�reader�IOErrorr Zcritical�format�sys�exit�next�close�writer�infoZ
export_enable�
first_line)�selfr r � file_moder �e)� __class__� �!/usr/lib/python3.6/glances_csv.pyr s.
zExport.__init__c C s t jd| j � | jj� dS )zClose the CSV file.z Finalise export interface %sN)r �debugZ export_namer r )r! r% r% r& r B s z Export.exitc st |j �j|�d�}�jr dg}tjd�g}x��j|�D ]�� t|� t�r�x�t|� dd� d�D ]0��jr�|� ��fdd��D �7 }|t��7 }q^W q6t|� t �r6�jr�t
|� �}|� fd d
�|D �7 }|t|� �7 }q6W �j�rN�j d kr��j j
|� �j |k�rB�j d k �rBtjd � tjd
j�j �� tjdj|�� nd �_ d�_�j d k�rp�j j
|� �jj� d S )z$Update stats in the CSV output file.)Z plugin_listZ timestampz%Y-%m-%d %H:%M:%Sc S s | d S )N�keyr% )�xr% r% r& �<lambda>T s z Export.update.<locals>.<lambda>)r( c s g | ]}d j � �j��|��qS )z{}_{}_{})r Z get_item_key)�.0�item)�pluginr! �statr% r& �
<listcomp>W s z!Export.update.<locals>.<listcomp>c 3 s | ]}d j � |�V qdS )z{}_{}N)r )r+ Z fieldname)r- r% r& � <genexpr>^ s z Export.update.<locals>.<genexpr>Nz?Cannot append data to existing CSV file. Headers are different.zOld header: {}zNew header: {}F)ZgetAllExportsAsDictZplugins_to_exportr �timeZstrftime�
isinstance�list�sortedr �dictr r r Zwriterowr �errorr' r r �flush)r! ZstatsZ all_statsZ
csv_headerZcsv_dataZ
fieldnamesr% )r- r! r. r& �updateG s6
z
Export.update)NN)�__name__�
__module__� __qualname__�__doc__r r r8 �
__classcell__r% r% )r$ r& r s (r c C s& t rt| |dd�}nt| |d �}|S )N� )�newline�b)r �open)� file_namer" r r% r% r& r y s r )r< Zos.pathr r r r1 Zglances.compatr r r Zglances.loggerr Z glances.exports.glances_exportr r r r% r% r% r& �<module>
s c