3
)��hR � @ sP d dl Z d dlZd dlZd dlZd dlZG dd� dej�ZedkrLe� j� dS )� Nc @ s e Zd ZdZ dd� ZdS )�Pluginzdiskstatus-nvmec C s i }y>t jdt jt jdd�j� d }tj|jd��}|d d}W n tk
rZ d}dS X |dk�rx�|d D ]�}i }tj d j
|d
��j � }ytj|�}W n tk
r� Y nX x:|j � D ].\} }
| j
d �r�t|
d d�|| < q�|
|| < q�W |||d
jd
d�< qpW |S )z�
Monitor nvme disk status
For NVME drives install nvme-cli (https://github.com/linux-nvme/nvme-cli#distro-support)
This plugin requires the agent to be run under the root user.
z nvme --list --output-format=jsonT)�stdout�stderr�shellr zutf-8ZDevicesFz'Could not fetch nvme status informationz&nvme smart-log {} --output-format=jsonZ
DevicePathZ temperaturegfffffq@z/dev/� )�
subprocess�Popen�PIPE� communicate�json�loads�decode� Exception�os�popen�format�read�items�
startswith�round�replace) �self�config�results�dataZnvme�valueZdeviceZ disk_dataZ data_diskZdisk_keyZ
disk_value� r �I/root/tmp/pip-build-yeh9mt8d/agent360/agent360/plugins/diskstatus-nvme.py�run s.
z
Plugin.runN)�__name__�
__module__� __qualname__r r r r r r s r �__main__) r r Zpluginsr �reZ
BasePluginr r �executer r r r �<module> s % |