B
�=�_{H � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z ddl m Z
ddl mZ dd l
mZ dd
l
mZ dd l
mZ dd l
mZ dd
l
mZ ddl
mZ ddl
mZ ddl
mZ ddl
mZ ddl
mZ ddlmZ ddlm Z ddlm Z ddlm Z dgZ e e
d�Z!e e
d�Z"e e
d�Z#e�$� Z%ej&Z&e
j'e
j(e
j)e
j*e
j+e
j,e
j-e
j,e
j.e
j/iZ0e
j1e
j2e
j3e
j4e
j5e
j6e
j7e
j8e
j9e
j:e
j;e
je
j?e
j@e
jAe
jBe
jCe
jDe
jEe
jFe
jGe
jHi ZIeJdddd d d d d d!�ZKed"d#d$g�ZLeLZMed%d&d'd(d)g�ZNed*d+d,d-d.d/g�ZOd0d1� ZPd2d3� ZQd4d5� ZRd6d7� ZSd8d9� ZTd:d;� ZUd� s z cpu_times..)r3 �
per_cpu_timesr) �zip)�retr4 r4 r5 � cpu_times} s rB c C s t �� } dd� | D �S )z5Return system per-CPU times as a list of named tuplesc S s g | ] }t |� �qS r4 )r) )r< r= r4 r4 r5 r> � s z!per_cpu_times..)r3 r? )rA r4 r4 r5 r? � s r? c C s$ y
t �d�S tk
r dS X dS )z0Return the number of logical CPUs in the system.�SC_NPROCESSORS_ONLNN)�os�sysconf�
ValueErrorr4 r4 r4 r5 �cpu_count_logical� s
rG c C sr d} t j| dt jt jd�}|�� \}}tr@dd� ||fD �\}}|jdkrZtd| |f ��|�� �� }t |�ppd S )Nzlsdev -Cc processorT)�shell�stdout�stderrc S s g | ]}|� tjj��qS r4 )�decode�sysrI �encoding)r< r= r4 r4 r5 r> � s z&cpu_count_physical..r z%r command error
%s)
�
subprocess�Popen�PIPE� communicater �
returncode� RuntimeError�strip�
splitlines�len)�cmd�prI rJ Z
processorsr4 r4 r5 �cpu_count_physical� s
rY c C s t �� \} }}}t�| |||�S )z*Return various CPU stats as a named tuple.)r3 � cpu_statsr � scpustats)� ctx_switches�
interrupts�soft_interrupts�syscallsr4 r4 r5 rZ � s rZ Fc C sl g }t �� }xZ|D ]R}|\}}}}|dkr.d}| s>t|�js>qd }} t�|||||| �}
|�|
� qW |S )z Return system disk partitions.Znone� N)r3 �disk_partitions�
disk_usager. r � sdiskpart�append) �all�retlist�
partitions� partition�device�
mountpoint�fstype�opts�maxfile�maxpath�ntupler4 r4 r5 ra � s
ra �����c C s� t j}| |kr.td| d�dd� |D ��f ��t j| \}}t�|�}g }x`|D ]X}|\}} }
} } }
}| |krpqP|
|krzqPt|| |
| | |
t|dkr�|ndd�}|�|� qPW |S )z�Return socket connections. If pid == -1 return system-wide
connections (as opposed to connections opened by one process only).
z+invalid %r kind argument; choose between %sz, c S s g | ] }t |��qS r4 )�repr)r< r= r4 r4 r5 r> � s z#net_connections..rp N)�pid) r � conn_tmaprF �joinr3 �net_connectionsr � TCP_STATUSESrd )�kind�_pidZcmapZfamilies�types�rawlistrA �item�fd�fam�type_�laddr�raddrr& rr �ntr4 r4 r5 ru � s"
ru c C s� t td�} tdd� t� D ��}i }x�|D ]�}t�|�\}}d}d}tjdd|gtjtjd�}|� � \} }
t
r�d d� | |
fD �\} }
|j dkr�t �
d
| �} | d k r�t| �d ��}| �d
�}| �|t�}t�||||�||< q(W |S )z)Get NIC stats (isup, duplex, speed, mtu).)�FullZHalfc S s g | ] }|d �qS )r r4 )r< r= r4 r4 r5 r> � s z net_if_stats..r` r z/usr/bin/entstatz-d)rI rJ c S s g | ]}|� tjj��qS r4 )rK rL rI rM )r< r= r4 r4 r5 r> � s z"Running: (\d+) Mbps.*?(\w+) DuplexNr r )r r
�set� net_if_addrsr3 � net_if_statsrN rO rP rQ r rR �re�search�int�group�getr r � snicstats) Z
duplex_map�namesrA �name�isup�mtu�duplex�speedrX rI rJ Z re_resultr4 r4 r5 r� � s.
r� c C s t �� S )z:The system boot time expressed in seconds since the epoch.)r3 � boot_timer4 r4 r4 r5 r� s r� c C s` g } t �� }d}xJ|D ]B}|\}}}}}} |s0q||kr - s zpids..)rD �listdirr
r4 r4 r4 r5 �pids+ s r� c C s t j�t j�t� t| �d��S )z&Check for the existence of a unix pid.Zpsinfo)rD �path�existsrt r
�str)rr r4 r4 r5 �
pid_exists0 s r� c s t �� �� fdd��}|S )z�Call callable into a try/except clause and translate ENOENT,
EACCES and EPERM in NoSuchProcess or AccessDenied exceptions.
c sz y� | f|�|�S t tfk
rT t| j�s>t| j| j��nt| j| j| j��Y n" tk
rt t | j| j��Y nX d S )N)
r r r� rr r �_namer �_ppidr r )�self�args�kwargs)�funr4 r5 �wrapper9 s
z wrap_exceptions..wrapper)� functools�wraps)r� r� r4 )r� r5 �wrap_exceptions5 s r� c @ sx e Zd ZdZdddddgZdd� Zd d
� Zd d � Zee d
d� ��Z
ee dd� ��Z edd� �Z edd� �Z
edd� �Zedd� �Zedd� �Zedd � �Zer�ed d � �ZedAd d!��Zed"d#� �Zed$d%� �Zed&d'� �Zed(d)� �Zed*d+� �Zed,d-� �Zed.d/� �Zed0d1� �Z ed2d3� �Z e Z ed4d5� �Z d6d7� Z ed8d9� �Z!ed:d;� �Z"edBd=d>��Z#e$�rted?d@� �Z%d � s z&Process.open_files..zno such processz (\d+): S_IFREG.*\s*.*name:(.*)\nz//r zcannot be retrieved)rN rO r� rr rP rQ r �lowerr r� r� �findallrT �
startswithrd r � popenfiler� )r� rX rI rJ Z procfilesrf r| r� r4 r4 r5 �
open_files� s$
zProcess.open_filesc C s( | j dkrdS tt�d| j| j f ��S )Nr z%s/%s/fd)rr rV rD r� r� )r� r4 r4 r5 �num_fds s
zProcess.num_fdsc C s t jt�| j�� S )N)r �pctxswr3 Zproc_num_ctx_switchesrr )r� r4 r4 r5 �num_ctx_switches s zProcess.num_ctx_switchesNc C s t �| j|| j�S )N)r Zwait_pidrr r� )r� �timeoutr4 r4 r5 �wait s z Process.waitc C sX yt �| j�\}}}}W n. tk
rF t| j�s@t| j| j��� Y nX t�||||�S )N) r3 r rr �OSErrorr� r r� r �pio)r� �rcZwc�rb�wbr4 r4 r5 � io_counters s
zProcess.io_counters)r� )N)&�__name__�
__module__� __qualname__�__doc__� __slots__r� r� r� r� r r� r� r� r� r� r� r# r% � HAS_THREADSr� r� r� r� r r� r� rB r� r� r� Zmemory_full_infor& r� r� r� r� �HAS_PROC_IO_COUNTERSr� r4 r4 r4 r5 r� J sJ
r� )F)rp )er� r� r� rD r� rN rL � collectionsr r` r r r r3 r r� r r r
r r r
r r r r �_compatr r r r Z__extra__all__�hasattrr� ZHAS_NET_IO_COUNTERSr� Z getpagesizeZ PAGE_SIZEZAF_LINKZSIDL� STATUS_IDLEZSZOMB�
STATUS_ZOMBIEZSACTIVE�STATUS_RUNNINGZSSWAPZSSTOP�STATUS_STOPPEDr� ZTCPS_ESTABLISHED�CONN_ESTABLISHEDZ
TCPS_SYN_SENT�
CONN_SYN_SENTZ
TCPS_SYN_RCVD�
CONN_SYN_RECVZTCPS_FIN_WAIT_1�CONN_FIN_WAIT1ZTCPS_FIN_WAIT_2�CONN_FIN_WAIT2ZTCPS_TIME_WAIT�CONN_TIME_WAITZ TCPS_CLOSED�
CONN_CLOSEZTCPS_CLOSE_WAIT�CONN_CLOSE_WAITZ
TCPS_LAST_ACK�
CONN_LAST_ACKZ TCPS_LISTEN� CONN_LISTENZ TCPS_CLOSING� CONN_CLOSINGZPSUTIL_CONN_NONE� CONN_NONErv �dictr� r( Zpfullmemr) r- r6 r: rB r? rG rY rZ Zdisk_io_countersrb ra r� r ru r� r� r� r� r� r� �objectr� r4 r4 r4 r5 � s�
%