File: //usr/local/lib/python3.6/site-packages/agent360/plugins/__pycache__/vms.cpython-36.pyc
3
3 Eg� � @ st d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z G dd� dej
�Z e dkrpe � j
� dS )� )�print_functionNc @ s@ e Zd ZdZ dd� Zdd� Zdd� Zdd � Zd
d � Zd d
� ZdS )�PluginZvmsc C s� i }i }| j � }tjdd�}| j|�}i }x�|j� D ]�\}} i ||< x�| j� D ]t\}
} |
dksv|
dksv|
dksv|
dkr�| |
|| |
< qN| jd||
f t| �|�|| |
< t| |
�|d||
f < qNW q4W tj� |d< | j|� |S ) z�
Using the libvirt API to fetch statistics from guests
running KVM, QEMU, Xen, Virtuozzo, VMWare ESX, LXC,
BHyve and more
�urizqemu:///system� mem_bytes�soft_limit_bytes�min_guarantee_bytes�hard_limit_bytesz%s_%s�ts) Zget_agent_cache�os�getenv� fetch_values�itemsZabsolute_to_per_second�float�timeZset_agent_cache) �self�config�results�
last_valueZ
prev_cacher �valuesZdeltas�key�value�subkeyZsubvalue� r �=/root/tmp/pip-build-f6rmvwrq/agent360/agent360/plugins/vms.py�run s
"
z
Plugin.runc C s t jdd|�S )Nz
[^a-zA-Z0-9_]�_)�re�sub)r �namer r r �canon% s z Plugin.canonc
C s� |j d�}d }ytj|�}W n
g S |j� }g }zX|jd�}xH|D ]@}d }x"|jD ]} | jdkrV| jd�}qVW |d kr|qF|j|� qFW W d |d kr�|j � |d kr�|j
� X |S )Nr z/domain/devices/interface�target�dev) �XMLDesc�libxml2�parseDoc�xpathNewContext� xpathEval�childrenr �prop�append�xpathFreeContext�freeDoc)
r �dom�xml�doc�ctx�ifaces�ret�node�devdst�childr r r �
get_ifaces( s.
zPlugin.get_ifacesc C s� dddd�}|j d�}ytj|�}W n
g S |j� }z^xX|D ]P}|jd| �}y(x"|d jD ]}t|j�||< P q`W W q@ tk
r� Y q@X q@W W d |d kr�|j � |d kr�|j
� X |S )Nr )�
min_guarantee�
soft_limit�
hard_limitz/domain/memtune/%s) r" r# r$ r% r& r' �int�content�
IndexErrorr* r+ ) r r, �memtuner- r. r/ r r1 r4 r r r � get_memtuneB s*
zPlugin.get_memtunec 7 C s t j|�}|j� }i }�x�|D �]�}i }d|d< d|d< y|j|�}|j� }W n< t jk
r� } z td|| f tjd� w W Y d d } ~ X nX |dkr�q | j |�}
xf|
D ]^} y6|j
| �} |d | d 7 < |d | d 7 < W q� ttj? d| |f f Y q�X q�W t |j � d �}
d |
}||d
< y|t
j� |d < W n$ tk
�r` } zW Y d d }~X nX |j � d d
� \}}|d9 }|d9 }||d< | j|�}|d d |d< |d d |d< |d d |d< d|d< d|d< d|d< d|d< y|j|�}|j� }W n> t jk
�r< } z td|| f tjd� w W Y d d } ~ X nX |dk�rJq | j|�}x�|D ]�}yX|j|�\}}}}}|d |7 < |d |7 < |d |7 < |d |7 < W n, tk
�r� ttj? d||f f Y nX �qZW ||| j|�< q W |S )Nr Z net_rx_bytesZ net_tx_bytesz
Id: %s: %s)�filezDomain-0� z#Cannot get ifstats for '%s' on '%s'gH�����z>�cpuZcpu_percentage� � i r r6 r r8 r r7 r Z
disk_rd_bytesZ
disk_wr_bytesZ disk_wr_reqZ disk_rd_reqz&Cannot get blockstats for '%s' on '%s')�libvirtZ openReadOnlyZ
listDomainsIDZ
lookupByIDr Z libvirtError�print�sys�stderrr5 ZinterfaceStatsr �info�psutil� cpu_count� Exceptionr= � get_disksZ
blockStats� TypeErrorr )r r �conn�idsr �id�datar, r �errr0 Ziface�statsZcputimeZcputime_percentage�eZmaxmemZmemr< �disksZdiskZrd_reqZrd_bytesZwr_reqZwr_bytesZerrsr r r r ] s|
"zPlugin.fetch_valuesc
C s� |j d�}d }ytj|�}W n
g S |j� }g }zX|jd�}xH|D ]@}d }x"|jD ]} | jdkrV| jd�}qVW |d kr|qF|j|� qFW W d |d kr�|j � |d kr�|j
� X |S )Nr z/domain/devices/diskr r! ) r" r# r$ r% r& r' r r( r) r* r+ )
r r, r- r. r/ rT r1 r2 r3 r4 r r r rK � s.
zPlugin.get_disksN) �__name__�
__module__� __qualname__r r r5 r= r rK r r r r r s Dr �__main__)�
__future__r r rE r
rC r# r ZpluginsrH Z
BasePluginr rU �executer r r r �<module> s 4