U
    (cz                     @   s  d Z ddlZddlZddlZddlZddlm  mZ ddl	m
Z
 ddl	mZ ddlmZ ddlmZ ddlmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z  ddlm!Z! ddlm"Z" g Z#erzej$ej%ej&ej'ej(ej)ej*ej+ej,ej-ej.ej/ej0ej1iZ2n~erej$ej%ej(ej)ej*ej+ej3ej-ej,ej-ej&ej4ej5ej'iZ2n:erej$ej%ej(ej)ej*ej+ej,ej-ej&ej4ej5ej'iZ2ej6ej7ej8ej9ej:ej;ej<ej=ej>ej?ej@ejAejBejCejDejEejFejGejHejIejJejKejLejMiZNeO ZPejQZQeRedZSeRedZTeRedZUeRedZVeWdddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4ZXed5d6d7d8d9d:d;d<d=d>d?d@gZYedAdBdCdDdEdFgZZedGdHdIdJdKdLgZ[e[Z\edMdBdDdNdOgZ]edPdQZ^edRdSZ_er^edTdUdVdWdXdYdZd[gZ`nedTdUdVdWdXgZ`d\d] Zad^d_ Zbd`da ZceSrdbd Zdndcd Zddded_ededf Zfeserdgdh Zgndidh Zgdjdk Zherdldm Zinerdndm ZiddodpZjejkZkejlZlejmZmejnZndqdr Zodsdt Zper:dudv Zqdwdx Zrdydz Zsd{d| Zted}d~ Zudd Zvesjertdd ZwnejwZwdd Zxdd Zyejzdd Z{G dd de|Z}dS )z5FreeBSD, OpenBSD and NetBSD platforms implementation.    N)defaultdict)
namedtuple   )_common)_psposix)_psutil_bsd)_psutil_posix)FREEBSD)NETBSD)OPENBSD)AccessDenied)NoSuchProcess)ZombieProcess)	conn_tmap)conn_to_ntuple)memoize)memoize_when_activated)usage_percent)FileNotFoundError)PermissionError)ProcessLookupError)whichper_cpu_timesproc_num_threadsproc_open_filesproc_num_fds                     	   
                                             )ppidstatusreal_uideffective_uid	saved_uidreal_gideffective_gid	saved_gidttynrcreate_timectx_switches_volctx_switches_unvolread_io_countwrite_io_count	user_timesys_timech_user_timech_sys_timerssvmsmemtextmemdatamemstackcpunumnamesvmemtotal	availablepercentusedfreeactiveinactivebufferscachedsharedwired	scputimesusernicesystemidleirqpmemrE   rF   textdatastack	pcputimeschildren_userchildren_systempmmap_grouped*path rss, private, ref_count, shadow_count	pmmap_ext6addr, perms path rss, private, ref_count, shadow_countsdiskioZ
read_countZwrite_count
read_byteswrite_bytesZ	read_timeZ
write_timeZ	busy_timec                  C   s   t  } | \}}}}}}}}trtddn}	|	D ]b}
|
drTt|
 d d }q0|
drtt|
 d d }q0|
dr0t|
 d d }q0W 5 Q R X || | }|| | }t|| |dd}t|||||||||||S )	z&System virtual memory as a namedtuple.z/proc/meminforbs   Buffers:r   i   s
   MemShared:s   Cached:Zround_)	cextZvirtual_memr
   open
startswithintsplitr   rL   )ZmemrM   rQ   rR   rS   rW   rU   rT   rV   flineZavailrP   rO    ru   1/tmp/pip-unpacked-wheel-l83rk7k5/psutil/_psbsd.pyvirtual_memory   s,    


      rw   c                  C   s4   t  \} }}}}t|| dd}t| |||||S )z@System swap memory as (total, used, free, sin, sout) namedtuple.r   rm   )rn   Zswap_memr   r   Zsswap)rM   rP   rQ   sinZsoutrO   ru   ru   rv   swap_memory   s    ry   c                  C   s"   t  \} }}}}t| ||||S )z+Return system per-CPU times as a namedtuple)rn   	cpu_timesrX   )rY   rZ   r[   r\   r]   ru   ru   rv   rz      s    rz   c                  C   s>   g } t  D ],}|\}}}}}t|||||}| | q| S )'Return system CPU times as a namedtuple)rn   r   rX   append)retZcpu_trY   rZ   r[   r\   r]   itemru   ru   rv   r      s    c                   C   s.   t  dkrt gS tjr tddt_t gS )r{   r   &supported only starting from FreeBSD 8T)cpu_count_logicalrz   r   
__called__NotImplementedErrorru   ru   ru   rv   r      s    
Fc                   C   s   t  S )z0Return the number of logical CPUs in the system.)rn   r   ru   ru   ru   rv   r      s    r   c                   C   s   t  dkrdS d S )Nr   )r   ru   ru   ru   rv   cpu_count_cores   s    r   c                  C   sx   d} t  }|dk	rb|d}|dkrb|d|d  }t|}zt|dpRd} W 5 |  X | stt dkrtdS | S )z-Return the number of CPU cores in the system.Nz	</groups>r#   zgroup/children/group/cpur   )	rn   Zcpu_topologyrfindET
fromstringclearlenfindallr   )r}   sindexrootru   ru   rv   r     s    



c            	   	   C   s   t rt \} }}}}nptrnt \} }}}}}}tdd*}|D ]}|drBt| d }qBW 5 Q R X ntrt \} }}}}}}t	
| |||S )z*Return various CPU stats as a named tuple.z
/proc/statrl   s   intrr   )r	   rn   	cpu_statsr
   ro   rp   rq   rr   r   r   Z	scpustats)	ZctxswZintrsZ
soft_intrsZsyscallsZtrapsZfaultsZforksrs   rt   ru   ru   rv   r     s    
r   c               
   C   s   g } t  }t|D ]}zt|\}}W n tk
r@   Y qY nX |rz t|dd dd }W n ttfk
r   d}Y nX z t|dd dd }W n ttfk
r   d}Y nX | 	t
||| q| S )zReturn frequency metrics for CPUs. As of Dec 2018 only
        CPU 0 appears to be supported by FreeBSD and all other cores
        match the frequency of CPU 0.
         r   /r   N)r   rangern   cpu_freqr   rq   rr   
IndexError
ValueErrorr|   r   scpufreq)r}   num_cpuscpucurrentZavailable_freqZmin_freqZmax_freqru   ru   rv   r   <  s$    
 
 
r   c                  C   s   t t } t| ddgS )Ng        )floatrn   r   r   r   )currru   ru   rv   r   T  s    c              	   C   sL   g }t  }|D ]6}|\}}}}d }}	t||||||	}
||
 q|S )zReturn mounted disk partitions as a list of namedtuples.
    'all' argument is ignored, see:
    https://github.com/giampaolo/psutil/issues/906
    N)rn   disk_partitionsr   Z	sdiskpartr|   )allretlistZ
partitions	partitionZdeviceZ
mountpointZfstypeoptsZmaxfileZmaxpathntupleru   ru   rv   r   ^  s     r   c            
      C   s   t   } i }| D ]}z&t|}t|}t|\}}W n0 tk
rl } z|jtjkr\ W 5 d}~X Y qX t	t
drt
|}d|}d|k}	t
|	||||||< q|S )z)Get NIC stats (isup, duplex, speed, mtu).N	NicDuplex,Zrunning)net_io_counterskeys
cext_posixZ
net_if_mtuZnet_if_flagsZnet_if_duplex_speedOSErrorerrnoZENODEVhasattrr   r   joinZ	snicstats)
namesr}   rK   ZmtuflagsZduplexspeederrZoutput_flagsZisupru   ru   rv   net_if_stats{  s$    






r   c              
   C   s  t rtg }t D ]`}zt|| }W n ttfk
r@   Y qY qX |D ]&}t|}|| |tj	|  qFq|S | tj
krtd| ddd t
D f t
|  \}}t }trtd}nt }|D ]F}|\}	}
}}}}}|
|kr||krt|	|
||||t|}|| qt|S )z System-wide network connections.+invalid %r kind argument; choose between %s, c                 S   s   g | ]}t |qS ru   repr.0xru   ru   rv   
<listcomp>  s     z#net_connections.<locals>.<listcomp>r   )r   pidsProcessconnectionsr   r   listr|   r   Zsconnr   r   r   setr
   rn   net_connectionsr   TCP_STATUSESadd)kindr}   pidZconsconnfamiliestypesrawlistr~   fdfamtypeladdrraddrr4   ntru   ru   rv   r     s<    



 r   c                  C   sd   zt  \} }}W n tk
r(   Y dS X |dk}|r>tj}n|dkrNtj}n|d }t| ||S )zReturn battery info.Nr   r   <   )rn   sensors_batteryr   r   ZPOWER_TIME_UNLIMITEDZPOWER_TIME_UNKNOWNZsbattery)rO   ZminsleftZpower_pluggedZsecsleftru   ru   rv   r     s    r   c               	   C   sv   t t} t }t|D ]Z}z@t|\}}|dkr6d}d| }| d t|||| W q t	k
rn   Y qX q| S )z?Return CPU cores temperatures if available, else an empty dict.r   NzCore %sZcoretemp)
r   r   r   r   rn   Zsensors_cpu_temperaturer|   r   Zshwtempr   )r}   r   r   r   highrK   ru   ru   rv   sensors_temperatures  s    r   c                   C   s   t  S )z:The system boot time expressed in seconds since the epoch.)rn   	boot_timeru   ru   ru   rv   r     s    r   c            	      C   sf   g } t  }|D ]P}|\}}}}}|dkr6ts2td}|dkr@qt||pLd|||}| | q| S )z:Return currently connected users as a list of namedtuples.r   N~)rn   usersr   AssertionErrorr   Zsuserr|   )	r   r   r~   rY   ttyhostnameZtstampr   r   ru   ru   rv   r     s    r   c                   C   sD   zt d  W n* tk
r&   Y dS  tk
r:   Y dS X dS d S )Nr   FT)r   rK   r   r   ru   ru   ru   rv   _pid_0_exists  s    r   c                  C   s*   t  } tr&d| kr&t r&| dd | S )z7Returns a list of PIDs currently running on the system.r   )rn   r   r   r   insert)r}   ru   ru   rv   r   	  s    r   c                 C   s    t | }|s| t kS dS dS )zReturn True if pid exists.TN)r   
pid_existsr   )r   existsru   ru   rv   r     s    

r   c                 C   s:   zt | td  }|t jkW S  tk
r4   Y dS X d S )Nr4   F)rn   proc_oneshot_infokinfo_proc_mapSZOMB	Exception)r   stru   ru   rv   	is_zombie!  s
    r   c                    s   t   fdd}|S )z`Decorator which translates bare OSError exceptions into
    NoSuchProcess and AccessDenied.
    c                    s   z | f||W S  t k
rR   t| jr@t| j| j| jnt| j| jY n\ tk
rr   t| j| jY n< t	k
r   | jdkrdt
 krt| j| jn  Y nX d S Nr   )r   r   r   r   _name_ppidr   r   r   r   r   )selfargskwargsfunru   rv   wrapper-  s    


z wrap_exceptions.<locals>.wrapper)	functoolswraps)r   r   ru   r   rv   wrap_exceptions)  s    r   c              	   c   st   z
dV  W nd t tfk
rN   t| jr<t| j| j| jnt| j| jY n" tk
rn   t	| j| jY nX dS )z8Same as above, for routines relying on reading /proc fs.N)
r   r   r   r   r   r   r   r   r   r   )instru   ru   rv   wrap_exceptions_procfsB  s    

r   c                   @   s  e Zd ZdZddddgZdd Zdd	 Zeed
d Z	dd Z
dd Zedd Zedd Zedd Zedd Zedd Zedd Zedd Zedd Zed d! Zered"d# Zed$d% ZeZed&d' Zed(d) Zed*d+ Zed,d- ZedOd/d0ZedPd2d3Zed4d5 Zed6d7 Z ed8d9 Z!ed:d; Z"ed<d= Z#e$d>d?Z%e$d>d@Z&dAdB Z'e(redCdD Z)ne'Z)e*redEdF Z+ne'Z+eredGdH Z,edIdJ Z-edKdL Z.edQdMdNZ/d1S )Rr   z1Wrapper class around underlying C implementation.r   r   r   _cachec                 C   s   || _ d | _d | _d S N)r   r   r   )r   r   ru   ru   rv   __init__X  s    zProcess.__init__c                 C   s   t | j dS )z+Raise NSP if the process disappeared on us.N)rn   	proc_namer   r   ru   ru   rv   _assert_alive]  s    zProcess._assert_alivec                 C   s$   t | j}t|ttks t|S )z;Retrieves multiple process info in one shot as a raw tuple.)rn   r   r   r   r   r   r   r}   ru   ru   rv   oneshotc  s    zProcess.oneshotc                 C   s   | j |  d S r   )r   Zcache_activater   ru   ru   rv   oneshot_enterk  s    zProcess.oneshot_enterc                 C   s   | j |  d S r   )r   Zcache_deactivater   ru   ru   rv   oneshot_exitn  s    zProcess.oneshot_exitc                 C   s(   |   td  }|d k	r|S t| jS )NrK   )r   r   rn   r   r   )r   rK   ru   ru   rv   rK   q  s    zProcess.namec              
   C   s   t r| jdkrdS t| jS tr^| jdkr0dS t|  td| j W  5 Q R  S Q R X n |  }|rzt	|d pxdS dS d S )Nr    z/proc/%s/exe)
r	   r   rn   Zproc_exer
   r   osreadlinkcmdliner   )r   r   ru   ru   rv   exev  s    


$zProcess.exec              
   C   s   t r| jdkrg S trzt| jW S  tk
r } z\|jtjkrt| jr`t	| j| j
| jqt| js~t| j| j
| jqg  W Y 
S n W 5 d }~X Y qX nt| jS d S r   )r   r   r
   rn   Zproc_cmdliner   r   EINVALr   r   r   r   r   r   )r   r   ru   ru   rv   r     s    

zProcess.cmdlinec                 C   s   t | jS r   )rn   Zproc_environr   r   ru   ru   rv   environ  s    zProcess.environc                 C   s>   |   td  }t }z
|| W S  tk
r8   Y d S X d S )Nr;   )r   r   r   Zget_terminal_mapKeyError)r   Ztty_nrZtmapru   ru   rv   terminal  s    
zProcess.terminalc                 C   s   |   td  | _| jS )Nr3   )r   r   r   r   ru   ru   rv   r3     s    zProcess.ppidc                 C   s.   |   }t|td  |td  |td  S )Nr5   r6   r7   )r   r   Zpuidsr   r   Zrawtupleru   ru   rv   uids  s    


zProcess.uidsc                 C   s.   |   }t|td  |td  |td  S )Nr8   r9   r:   )r   r   Zpgidsr   r   ru   ru   rv   gids  s    


zProcess.gidsc                 C   s8   |   }t|td  |td  |td  |td  S )NrA   rB   rC   rD   )r   r   rb   r   r   ru   ru   rv   rz     s    



zProcess.cpu_timesc                 C   s   |   td  S )NrJ   r   r   r   ru   ru   rv   cpu_num  s    zProcess.cpu_numc                 C   s@   |   }t|td  |td  |td  |td  |td  S )NrE   rF   rG   rH   rI   )r   r^   r   r   ru   ru   rv   memory_info  s    




zProcess.memory_infoc                 C   s   |   td  S )Nr<   r  r   ru   ru   rv   r<     s    zProcess.create_timec                 C   s    t rt| jS t|  S d S r   )HAS_PROC_NUM_THREADSrn   r   r   r   threadsr   ru   ru   rv   num_threads  s    zProcess.num_threadsc                 C   s$   |   }t|td  |td  S )Nr=   r>   )r   r   Zpctxswr   r   ru   ru   rv   num_ctx_switches  s
    

zProcess.num_ctx_switchesc                 C   sH   t | j}g }|D ]"\}}}t|||}|| qtrD|   |S r   )rn   Zproc_threadsr   r   Zpthreadr|   r   r   )r   r   r   Z	thread_idutimeZstimer   ru   ru   rv   r    s    zProcess.threadsinetc              	   C   s  |t kr(td|ddd t D f trt | \}}g }t| j}|D ]R}|\}}}	}
}}}|| jkspt||krL|	|krLt|||	|
||t	}|
| qL|   t|S t | \}}t| j||}g }|D ]2}|\}}}	}
}}t|||	|
||t	}|
| qtr|   |S )Nr   r   c                 S   s   g | ]}t |qS ru   r   r   ru   ru   rv   r     s     z'Process.connections.<locals>.<listcomp>)r   r   r   r
   rn   r   r   r   r   r   r|   r   r   Zproc_connectionsr   )r   r   r   r   r}   r   r~   r   r   r   r   r   r4   r   r   ru   ru   rv   r     s<    zProcess.connectionsNc                 C   s   t | j|| jS r   )r   Zwait_pidr   r   )r   timeoutru   ru   rv   wait%  s    zProcess.waitc                 C   s   t | jS r   )r   getpriorityr   r   ru   ru   rv   nice_get)  s    zProcess.nice_getc                 C   s   t | j|S r   )r   setpriorityr   )r   valueru   ru   rv   nice_set-  s    zProcess.nice_setc                 C   s   |   td  }t|dS )Nr4   ?)r   r   PROC_STATUSESget)r   coderu   ru   rv   r4   1  s    zProcess.statusc                 C   s(   |   }t|td  |td  ddS )Nr?   r@   r   )r   r   Zpior   r   ru   ru   rv   io_counters7  s    

zProcess.io_countersc                 C   s>   t r| jdkrdS tstr*t| jp(dS ttr4dnddS )z)Return process current working directory.r   Nr   r   )r   r   r
   HAS_PROC_OPEN_FILESrn   Zproc_cwdr   r	   r   ru   ru   rv   cwd@  s    zProcess.cwdZmmaprf   rh   c                 C   s   t d S r   )r   r   ru   ru   rv   _not_implementedU  s    zProcess._not_implementedc                 C   s   t | j}dd |D S )z8Return files opened by process as a list of namedtuples.c                 S   s   g | ]\}}t ||qS ru   )r   Z	popenfile)r   pathr   ru   ru   rv   r   _  s     z&Process.open_files.<locals>.<listcomp>)rn   r   r   )r   r   ru   ru   rv   
open_files[  s    zProcess.open_filesc                 C   s   t | j}tr|   |S )z=Return the number of file descriptors opened by this process.)rn   r   r   r
   r   r   ru   ru   rv   num_fdsf  s    zProcess.num_fdsc                 C   s   t | jS r   )rn   Zproc_cpu_affinity_getr   r   ru   ru   rv   cpu_affinity_gett  s    zProcess.cpu_affinity_getc              
   C   s   t ttt }|D ]}||krtd||f qzt| j| W nX tk
r } z:|j	t	j
t	jfkr|D ]}||krntd||f qn W 5 d }~X Y nX d S )Nz#invalid CPU #%i (choose between %s))tupler   r   r   r   rn   Zproc_cpu_affinity_setr   r   r   r   ZEDEADLK)r   ZcpusZallcpusr   r   ru   ru   rv   cpu_affinity_setx  s&     zProcess.cpu_affinity_setc                 C   s   t | jS r   )rn   Zproc_memory_mapsr   r   ru   ru   rv   memory_maps  s    zProcess.memory_mapsc                 C   sP   |d krt | j|S t|dkr2tdt| |\}}t | j|||S d S )Nr   z4second argument must be a (soft, hard) tuple, got %s)rn   Zproc_getrlimitr   r   r   r   Zproc_setrlimit)r   resourceZlimitsZsofthardru   ru   rv   rlimit  s    zProcess.rlimit)r  )N)N)0__name__
__module____qualname____doc__	__slots__r   r   r   r   r   r   r   rK   r   r   r   r   r3   r  r  rz   r	   r  r  Zmemory_full_infor<   r  r	  r  r   r  r  r  r4   r  r  r   Znt_mmap_groupedZnt_mmap_extr  r  r  HAS_PROC_NUM_FDSr  r  r   r!  r$  ru   ru   ru   rv   r   S  s   










	



!




  


r   )F)~r(  
contextlibr   r   r   xml.etree.ElementTreeetreeElementTreer   collectionsr   r   r   r   r   r   rn   r   r   r	   r
   r   r   r   r   r   r   r   r   r   _compatr   r   r   r   Z__extra__all__ZSIDLZSTATUS_IDLEZSRUNZSTATUS_RUNNINGZSSLEEPZSTATUS_SLEEPINGZSSTOPZSTATUS_STOPPEDr   ZSTATUS_ZOMBIEZSWAITZSTATUS_WAITINGZSLOCKZSTATUS_LOCKEDr  ZSDEADZSTATUS_WAKINGZSONPROCZTCPS_ESTABLISHEDZCONN_ESTABLISHEDZTCPS_SYN_SENTZCONN_SYN_SENTZTCPS_SYN_RECEIVEDZCONN_SYN_RECVZTCPS_FIN_WAIT_1ZCONN_FIN_WAIT1ZTCPS_FIN_WAIT_2ZCONN_FIN_WAIT2ZTCPS_TIME_WAITZCONN_TIME_WAITZTCPS_CLOSEDZ
CONN_CLOSEZTCPS_CLOSE_WAITZCONN_CLOSE_WAITZTCPS_LAST_ACKZCONN_LAST_ACKZTCPS_LISTENZCONN_LISTENZTCPS_CLOSINGZCONN_CLOSINGZPSUTIL_CONN_NONEZ	CONN_NONEr   ZgetpagesizeZPAGESIZEZAF_LINKr   ZHAS_PER_CPU_TIMESr  r  r*  dictr   rL   rX   r^   Zpfullmemrb   re   rg   ri   rw   ry   rz   r   r   r   r   r   r   r   
disk_usageZdisk_io_countersr   Znet_if_addrsr   r   r   r   r   r   r   r   r   r   r   contextmanagerr   objectr   ru   ru   ru   rv   <module>   s         	             
            



# 
      
     	
	



(




