o
    jĎiOr                     @   st  d Z ddlZddlZddlZddl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rejejej ej!ej"ej#ej$ej%ej&ej'ej(ej)ej*ej+iZ,n=erejejej"ej#ej$ej%ej-ej'ej&ej'ej ej.ej/ej!iZ,nerejejej"ej#ej$ej%ej&ej'ej ej.ej/ej!iZ,ej0ej1ej2ej3ej4ej5ej6ej7ej8ej9ej:ej;ej<ej=ej>ej?ej@ejAejBejCejDejEejFejGiZHeI ZJejKZKeLedZMeNdzi ddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGZOdHdI ZPdJdK ZQeLedLrejRZRejSZSdMdN ZTdOdP ZUdQdR ZVeserdSdT ZWndUdT ZWdVdW ZXerdXdY ZYnerdZdY ZYd{d\d]ZZej[Z[ej\Z\ej]Z]ej^Z^d^d_ Z_d`da Z`erdbdc Zaddde Zbdfdg Zcerzec ZdW n eey  Zf zedhef dZdW Y dZf[fndZf[fww didj Zgdkdl Zhedmdn Zidodp Zjerdqdr Zkner#dsdr ZknejkZkdtdu Zlejmdvdw ZnG dxdy dyZodS )|z5FreeBSD, OpenBSD and NetBSD platforms implementation.    N)defaultdict)
namedtuple)ElementTree   )_common)_ntuples)_psposix)_psutil_bsd)FREEBSD)NETBSD)OPENBSD)AccessDenied)NoSuchProcess)ZombieProcess)	conn_tmap)conn_to_ntuple)debug)memoize)memoize_when_activated)usage_percentproc_num_threads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   c                  C   s   t  } trP| \}}}}}}tdd,}|D ]!}|dr(t| d d }	q|dr7t| d d }
qW d    n1 sBw   Y  || }|| }n| \}}}}}}}	}
|| | }|| | }t|| |dd}t	||||||||	||
|S )Nz/proc/meminforbs   Buffers:r   i   s
   MemShared:round_)
cextvirtual_memr   open
startswithintsplitr   ntpsvmem)memtotalfreeactiveinactivewiredcachedflinebufferssharedusedavailpercent r`   I/home/jeff/fluffinator/venv/lib/python3.10/site-packages/psutil/_psbsd.pyvirtual_memory   s>   


rb   c                  C   s4   t  \} }}}}t|| dd}t| |||||S )z@System swap memory as (total, used, free, sin, sout) namedtuple.r   rH   )rJ   swap_memr   rP   sswap)rS   r]   rT   sinsoutr_   r`   r`   ra   swap_memory   s   rg   	heap_infoc                  C   s$   t  \} }}}}t| ||||S )z,Return system per-CPU times as a namedtuple.)rJ   	cpu_timesrP   	scputimes)usernicesystemidleirqr`   r`   ra   ri      s   ri   c                  C   s@   g } t  D ]}|\}}}}}t|||||}| | q| S )z(Return system CPU times as a namedtuple.)rJ   per_cpu_timesrP   rj   append)retcpu_trk   rl   rm   rn   ro   itemr`   r`   ra   rp      s   rp   c                   C      t  S )z0Return the number of logical CPUs in the system.)rJ   cpu_count_logicalr`   r`   r`   ra   rv         rv   c                   C   s   t  dkrdS d S )Nr   )rv   r`   r`   r`   ra   cpu_count_cores   s   rx   c                  C   s   d} t  }|dur5|d}|dkr5|d|d  }t|}zt|dp)d} W |  n|  w | s>t dkr>dS | S )z-Return the number of CPU cores in the system.Nz	</groups>r(   zgroup/children/group/cpur   )	rJ   cpu_topologyrfindr   
fromstringlenfindallclearrv   )rr   sindexrootr`   r`   ra   rx      s   


c            	      C   s   t rt \} }}}}nBtrAt \} }}}}}}tdd}|D ]}|dr0t| d }q!W d   n1 s;w   Y  ntrNt \} }}}}}}t	
| |||S )z*Return various CPU stats as a named tuple.z
/proc/statrG   s   intrr   N)r
   rJ   	cpu_statsr   rL   rM   rN   rO   r   rP   	scpustats)	ctxswintrs
soft_intrssyscalls_traps_faults_forksrY   rZ   r`   r`   ra   r     s"   
r   c               
   C   s   g } t  }t|D ]]}z	t|\}}W n	 ty   Y q	w |r\zt|dd dd }W n ttfy=   d}Y nw zt|dd dd }W n ttfy[   d}Y nw | 	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.
         ry   /r   N)rv   rangerJ   cpu_freqNotImplementedErrorrN   rO   
IndexError
ValueErrorrq   rP   scpufreq)rr   num_cpuscpucurrentavailable_freqmin_freqmax_freqr`   r`   ra   r   #  s*     r   c                  C   s   t t } t| ddgS )Ng        )floatrJ   r   rP   r   )currr`   r`   ra   r   =  s   Fc           	      C   s@   g }t  }|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.
    )rJ   disk_partitionsrP   	sdiskpartrq   )	allretlist
partitions	partitiondevice
mountpointfstypeoptsntupler`   r`   ra   r   G  s   r   c            
      C   s   t   } i }| D ]M}zt|}t|}t|\}}W n ty7 } z|jtjkr- W Y d}~q	d}~ww t	t
drBt
|}d|}d|v }	t|	||||||< q	|S )z)Get NIC stats (isup, duplex, speed, mtu).N	NicDuplex,running)net_io_counterskeysrJ   
net_if_mtunet_if_flagsnet_if_duplex_speedOSErrorerrnoENODEVhasattrr   r   joinrP   	snicstats)
namesrr   rE   mtuflagsduplexspeederroutput_flagsisupr`   r`   ra   net_if_statsb  s&   





r   c              
   C   s   t |  \}}t }trtd||}ntrtd| }nt||}|D ]}|\}}}}	}
}}t||||	|
|t|}|| q$t	|S )z System-wide network connections.ry   )
r   setr   rJ   net_connectionsr   r   TCP_STATUSESaddlist)kindfamiliestypesrr   rawlistrt   fdfamtypeladdrraddrr   pidntr`   r`   ra   r   x  s   r   c                  C   sb   z	t  \} }}W n
 ty   Y dS w |dk}|rtj}n|dkr&tj}n|d }t| ||S )zReturn battery info.Nr   ry   <   )rJ   sensors_batteryr   r   POWER_TIME_UNLIMITEDPOWER_TIME_UNKNOWNrP   sbattery)r_   minsleftpower_pluggedsecsleftr`   r`   ra   r     s   r   c               	   C   st   t t} t }t|D ],}z!t|\}}|dkrd}d| }| d t|||| W q t	y7   Y qw | S )z?Return CPU cores temperatures if available, else an empty dict.r   NzCore coretemp)
r   r   rv   r   rJ   sensors_cpu_temperaturerq   rP   shwtempr   )rr   r   r   r   highrE   r`   r`   ra   sensors_temperatures  s   
r   c                   C   ru   )z:The system boot time expressed in seconds since the epoch.)rJ   	boot_timer`   r`   r`   ra   r     rw   r   zignoring exception on import: c                 C   sN   t dkr| S t t  }|dkst|dk r| S td |dk r#| | S | | S )z!Account for system clock updates.r   r   z9system clock was updated; adjusting process create_time())INIT_BOOT_TIMEr   absr   )ctimediffr`   r`   ra   adjust_proc_create_time  s   
r   c            	      C   sR   g } t  }|D ]}|\}}}}}|dkrqt||pd|||}| | q| S )z:Return currently connected users as a list of namedtuples.~N)rJ   usersrP   suserrq   )	r   r   rt   rk   ttyhostnametstampr   r   r`   r`   ra   r     s   r   c                   C   s:   z	t d  W dS  ty   Y dS  ty   Y dS w )Nr   FT)ProcessrE   r   r   r`   r`   r`   ra   _pid_0_exists  s   r   c                  C   s*   t  } trd| vrt r| dd | S )z7Returns a list of PIDs currently running on the system.r   )rJ   pidsr   r   insert)rr   r`   r`   ra   r     s   r   c                 C   s   t | }|s| t v S dS )NTr   
pid_existsr   r   existsr`   r`   ra   r      s   

r   c                 C   s   t | }|s	dS | t v S )NFr   r   r`   r`   ra   r     s   

c                    s   t   fdd}|S )z`Decorator which translates bare OSError exceptions into
    NoSuchProcess and AccessDenied.
    c              
      s   | j | j| j}}}z | g|R i |W S  ty5 } zt|r+t||||t|||d }~w tyF } zt	|||d }~w tj
yY } zt||||d }~w tyv } z|dkrodt v rot	||||d d }~ww )Nr   )r   _ppid_nameProcessLookupErrorrJ   proc_is_zombier   r   PermissionErrorr   ZombieProcessErrorr   r   )selfargskwargsr   r   rE   r   funr`   ra   wrapper  s(   
z wrap_exceptions.<locals>.wrapper)	functoolswraps)r   r   r`   r   ra   wrap_exceptions  s   r   c              
   c   s    | j | j| j}}}zdV  W dS  ttfy3 } zt| j r)t||||t|||d}~w t	yD } zt
|||d}~ww )z8Same as above, for routines relying on reading /proc fs.N)r   r   r   r   FileNotFoundErrorrJ   r   r   r   r   r   )instr   rE   r   r   r`   r`   ra   wrap_exceptions_procfs3  s   r   c                   @   s  e Zd ZdZg d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r`edd  Zed!d" ZeZedKd$d%Zed&d' Zed(d) Zed*d+ ZedLd-d.ZedMd0d1Zed2d3 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'edAdB Z(eredCdD Z)edEdF Z*edGdH Z+edMdIdJZ,d/S d/S )Nr   z1Wrapper class around underlying C implementation.)_cacher   r   r   c                 C   s   || _ d | _d | _d S N)r   r   r   )r   r   r`   r`   ra   __init__J  s   
zProcess.__init__c                 C   s   t | j dS )z+Raise NSP if the process disappeared on us.N)rJ   	proc_namer   r   r`   r`   ra   _assert_aliveO  s   zProcess._assert_alivec                 C   s$   t | j}t|ttksJ |S )z;Retrieves multiple process info in one shot as a raw tuple.)rJ   proc_oneshot_infor   r}   kinfo_proc_mapr   rr   r`   r`   ra   oneshotU  s   zProcess.oneshotc                 C      | j |  d S r  )r
  cache_activater  r`   r`   ra   oneshot_enter]     zProcess.oneshot_enterc                 C   r  r  )r
  cache_deactivater  r`   r`   ra   oneshot_exit`  r  zProcess.oneshot_exitc                 C   s(   |   td  }|d ur|S t| jS )NrE   )r
  r  rJ   r  r   )r   rE   r`   r`   ra   rE   c  s   zProcess.namec                 C   s   t r| jdkr	dS t| jS tr8| jdkrdS t|  td| j dW  d    S 1 s1w   Y  d S dd l}| 	 }|rK|
|d pJdS dS )Nr    z/proc/z/exe)r
   r   rJ   proc_exer   r   osreadlinkshutilcmdlinewhich)r   r  r  r`   r`   ra   exeh  s   


$zProcess.exec              
   C   s   t r	| jdkr	g S trZzt| jW S  tyY } z;|jtjkrT| j| j| j	}}}t
| jr8t||||t| jsDt||||td|d g W  Y d }~S  d }~ww t| jS )Nr   z	ignoring z and returning an empty list)r   r   r   rJ   proc_cmdliner   r   EINVALr   r   r   r   r   r   r   )r   r   r   rE   r   r`   r`   ra   r    s$   
zProcess.cmdlinec                 C      t | jS r  )rJ   proc_environr   r  r`   r`   ra   environ     zProcess.environc                 C   s8   |   td  }t }z|| W S  ty   Y d S w )Nr%   )r
  r  r   get_terminal_mapKeyError)r   tty_nrtmapr`   r`   ra   terminal  s   
zProcess.terminalc                 C   s   |   td  | _| jS )Nr   )r
  r  r   r  r`   r`   ra   r     s   zProcess.ppidc                 C   .   |   }t|td  |td  |td  S )Nr   r   r   )r
  rP   puidsr  r   rawtupler`   r`   ra   uids     


zProcess.uidsc                 C   r$  )Nr   r!   r#   )r
  rP   pgidsr  r&  r`   r`   ra   gids  r)  zProcess.gidsc                 C   s8   |   }t|td  |td  |td  |td  S )Nr1   r3   r5   r7   )r
  rP   	pcputimesr  r&  r`   r`   ra   ri     s   



zProcess.cpu_timesc                 C   s   |   td  S )NrC   )r
  r  r  r`   r`   ra   cpu_num  s   zProcess.cpu_numc              	   C   sB   |   }t|td  |td  |td  |td  |td  S )Nr9   r;   r=   r?   rA   )r
  rP   pmemr  r&  r`   r`   ra   memory_info  s   




zProcess.memory_infoFc                 C   s$   |   td  }tr|st|}|S )Nr'   )r
  r  r   r   )r   	monotonicr   r`   r`   ra   r'     s   zProcess.create_timec                 C   s   t rt| jS t|  S r  )HAS_PROC_NUM_THREADSrJ   r   r   r}   threadsr  r`   r`   ra   num_threads  s   zProcess.num_threadsc                 C   s$   |   }t|td  |td  S )Nr)   r+   )r
  rP   pctxswr  r&  r`   r`   ra   num_ctx_switches  s
   

zProcess.num_ctx_switchesc                 C   sH   t | j}g }|D ]\}}}t|||}|| q
tr"|   |S r  )rJ   proc_threadsr   rP   pthreadrq   r   r  )r   r   r   	thread_idutimestimer   r`   r`   ra   r2    s   zProcess.threadsinetc              	   C   s   t | \}}g }trt| j|}ntrt| j||}nt| j||}|D ](}|d d \}}}	}
}}tr@||vs?|	|vr@q't|||	|
||t	}|
| q'|   |S )Nr"   )r   r   rJ   r   r   r   proc_net_connectionsr
   r   r   rq   r  )r   r   r   r   rr   r   rt   r   r   r   r   r   r   r   r`   r`   ra   r     s$   zProcess.net_connectionsNc                 C      t | j|S r  )r   wait_pidr   )r   timeoutr`   r`   ra   wait     zProcess.waitc                 C   r  r  )rJ   proc_priority_getr   r  r`   r`   ra   nice_get  r  zProcess.nice_getc                 C   r=  r  )rJ   proc_priority_setr   )r   valuer`   r`   ra   nice_set#  rA  zProcess.nice_setc                 C   s   |   td  }t|dS )Nr   ?)r
  r  PROC_STATUSESget)r   coder`   r`   ra   r   '  s   zProcess.statusc                 C   s(   |   }t|td  |td  ddS )Nr-   r/   ry   )r
  rP   pior  r&  r`   r`   ra   io_counters-  s   

zProcess.io_countersc                 C   s   t r	| jdkr	dS t| jS )z)Return process current working directory.r   r  )r   r   rJ   proc_cwdr  r`   r`   ra   cwd7  s   zProcess.cwdmmapz*path rss, private, ref_count, shadow_countz6addr, perms path rss, private, ref_count, shadow_countc                 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 r`   )rP   	popenfile).0pathr   r`   r`   ra   
<listcomp>K  s    z&Process.open_files.<locals>.<listcomp>)rJ   proc_open_filesr   )r   r   r`   r`   ra   
open_filesG  s   zProcess.open_filesc                 C   s   t | j}tr|   |S )z=Return the number of file descriptors opened by this process.)rJ   proc_num_fdsr   r   r  r	  r`   r`   ra   num_fdsM  s   zProcess.num_fdsc                 C   r  r  )rJ   proc_cpu_affinity_getr   r  r`   r`   ra   cpu_affinity_getY  r  zProcess.cpu_affinity_getc              
   C   s   t ttt }|D ]}||vrd|d| d}t|qz
t| j| W d S  tyV } z!|j	t	j
t	jhv rQ|D ]}||vrPd|d| d}t||q< d }~ww )Nzinvalid CPU z (choose between ))r   r   r}   rp   r   rJ   proc_cpu_affinity_setr   r   r   r  EDEADLK)r   cpusallcpusr   msgr   r`   r`   ra   cpu_affinity_set]  s*   
zProcess.cpu_affinity_setc                 C   r  r  )rJ   proc_memory_mapsr   r  r`   r`   ra   memory_mapsx  r  zProcess.memory_mapsc                 C   sN   |d u rt | j|S t|dkrd|}t||\}}t | j|||S )Nr   z2second argument must be a (soft, hard) tuple, got )rJ   proc_getrlimitr   r}   r   proc_setrlimit)r   resourcelimitsr_  softhardr`   r`   ra   rlimit|  s   zProcess.rlimitF)r;  r  )-__name__
__module____qualname____doc__	__slots__r  r  r   r   r
  r  r  rE   r  r  r  r#  r   r(  r+  ri   r
   r-  r/  memory_full_infor'   r3  r5  r2  r   r@  rC  rF  r   rL  rN  r   nt_mmap_groupednt_mmap_extrU  rW  rY  r`  rb  ri  r`   r`   r`   ra   r   E  s    








	









	


	


r   r`   rj  )prn  
contextlibr   r   r  collectionsr   r   	xml.etreer   r  r   r   rP   r   r	   rJ   r
   r   r   r   r   r   r   r   r   r   r   r   __extra__all__SIDLSTATUS_IDLESRUNSTATUS_RUNNINGSSLEEPSTATUS_SLEEPINGSSTOPSTATUS_STOPPEDSZOMBSTATUS_ZOMBIESWAITSTATUS_WAITINGSLOCKSTATUS_LOCKEDrH  SDEADSTATUS_WAKINGSONPROCTCPS_ESTABLISHEDCONN_ESTABLISHEDTCPS_SYN_SENTCONN_SYN_SENTTCPS_SYN_RECEIVEDCONN_SYN_RECVTCPS_FIN_WAIT_1CONN_FIN_WAIT1TCPS_FIN_WAIT_2CONN_FIN_WAIT2TCPS_TIME_WAITCONN_TIME_WAITTCPS_CLOSED
CONN_CLOSETCPS_CLOSE_WAITCONN_CLOSE_WAITTCPS_LAST_ACKCONN_LAST_ACKTCPS_LISTENCONN_LISTENTCPS_CLOSINGCONN_CLOSINGPSUTIL_CONN_NONE	CONN_NONEr   getpagesizePAGESIZEAF_LINKr   r1  dictr  rb   rg   rh   	heap_trimri   rp   rv   rx   r   r   r   
disk_usagedisk_io_countersr   net_if_addrsr   r   r   r   r   r   	Exceptionr   r   r   r   r   r   r   contextmanagerr   r   r`   r`   r`   ra   <module>   sN  	

	
"/

 






	

