File: //usr/local/lib/python3.9/site-packages/agent360/plugins/__pycache__/vms.cpython-39.pyc
a
q�dg� � @ st d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z G dd� dej
�Z e dkrpe � �
� dS )� )�print_functionNc @ s@ e Zd ZdZ dd� Zdd� Zdd� Zdd � Zd
d � Zd d
� ZdS )�PluginZvmsc C s� i }i }| � � }t�dd�}| �|�}i }|�� D ]�\}} i ||< | �� D ]t\}
} |
dksr|
dksr|
dksr|
dkr�| |
|| |
< qJ| �d||
f t| �|�|| |
< t| |
�|d||
f < qJq2t�� |d< | �|� |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 �>/usr/local/lib/python3.9/site-packages/agent360/plugins/vms.py�run s
"
z
Plugin.runc C s t �dd|�S )Nz
[^a-zA-Z0-9_]�_)�re�sub)r �namer r r �canon% s z Plugin.canonc
C s� |� d�}d }zt�|�}W n g Y S 0 |�� }g }zp|�d�}|D ]<}d }|jD ]} | jdkrX| �d�}qX|d kr|qJ|�|� qJW |d kr�|� � |d kr�|�
� n"|d kr�|� � |d kr�|�
� 0 |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( s6
�
zPlugin.get_ifacesc C s� dddd�}|� d�}zt�|�}W n g Y S 0 |�� }zv|D ]L}|�d| �}z&|d jD ]}t|j�||< qzqbW qD ty� Y qD0 qDW |d kr�|� � |d kr�|�
� n"|d kr�|� � |d kr�|�
� 0 |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/ r0 r r2 r5 r r r � get_memtuneB s2
�
zPlugin.get_memtunec C s
t �|�}|�� }i }|D �]�}i }d|d< d|d< z|�|�}|�� }W nH t jy� } z.td|| f tjd� W Y d } ~ qW Y d } ~ n
d } ~ 0 0 |dkr�q| � |�}
|
D ]^} z6|�
| �} |d | d 7 < |d | d 7 < W q� ttj? d| |f f Y q�0 q�t |� � d �}
d |
}||d
< z|t
�� |d < W n& t�yf } z W Y d }~n
d }~0 0 |� � d d
� \}}|d9 }|d9 }||d< | �|�}|d d |d< |d d |d< |d d |d< d|d< d|d< d|d< d|d< z|�|�}|�� }W nJ t j�yN } z.td|| f tjd� W Y d } ~ qW Y d } ~ n
d } ~ 0 0 |dk�r\q| �|�}|D ]�}zX|�|�\}}}}}|d |7 < |d |7 < |d |7 < |d |7 < W n* t�y� ttj? d||f f Y n0 �qj||| �|�< q|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 r7 r r9 r r8 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�stderrr6 ZinterfaceStatsr �info�psutil� cpu_count� Exceptionr> � get_disksZ
blockStats� TypeErrorr )r r �conn�idsr �id�datar- r �errr1 Ziface�statsZcputimeZcputime_percentage�eZmaxmemZmemr= �disksZdiskZrd_reqZrd_bytesZwr_reqZwr_bytesZerrsr r r r ] s|
"
"
zPlugin.fetch_valuesc
C s� |� d�}d }zt�|�}W n g Y S 0 |�� }g }zp|�d�}|D ]<}d }|jD ]} | jdkrX| �d�}qX|d kr|qJ|�|� qJW |d kr�|� � |d kr�|�
� n"|d kr�|� � |d kr�|�
� 0 |S )Nr z/domain/devices/diskr r! r" )
r r- r. r/ r0 rU r2 r3 r4 r5 r r r rL � s6
�
zPlugin.get_disksN) �__name__�
__module__� __qualname__r r r6 r> r rL r r r r r s Dr �__main__)�
__future__r r rF r
rD r$ r ZpluginsrI Z
BasePluginr rV �executer r r r �<module> s 4