o
    F+ ia                     @   s  d Z ddlZddl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 g dZe ZejZejdkZ dZ!dZ"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.ej/iZ0e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jGejHejIe!ejJe"iZKeLdddddddddd d!d"d#ZMed$g d%ZNed&g d'ZOed(g d)ZPed*d+d,gZQeQZRed-g d.ZSed/d0d1TeSjU ZVd2d3 ZWd4d5 ZXd6d7 ZYd8d9 ZZd:d; Z[d<d= Z\d>d? Z]ej^Z^ej_Z_dTdAdBZ`ejaZaejbZbdUdDdEZcdFdG ZddHdI ZedJdK ZfdLdM ZgdNdO ZhdPdQ ZiG dRdS dSZjdS )Vz'Sun OS Solaris platform implementation.    N)
namedtuple)AF_INET   )_common)_psposix)_psutil_sunos)AF_INET6)ENCODING)AccessDenied)NoSuchProcess)ZombieProcess)debug)get_procfs_path)isfile_strict)memoize_when_activated)sockfam_to_enum)socktype_to_enum)usage_percent)	CONN_IDLE
CONN_BOUNDZPROCFS_PATHl        ZIDLEZBOUND                     	   
      )ppidrssvmscreate_timenicenum_threadsstatusttynruideuidgidegid	scputimes)usersystemidleZiowait	pcputimes)r-   r.   children_userchildren_systemsvmem)total	availablepercentusedfreepmemr!   r"   pmmap_grouped)pathr!   Z	anonymouslocked	pmmap_extzaddr perms  c                  C   sF   t dt } t dt  }}| | }t|| dd}t| ||||S )zReport virtual memory metrics.SC_PHYS_PAGESSC_AVPHYS_PAGESr   Zround_)ossysconf	PAGE_SIZEr   r3   )r4   r8   Zavailr7   r6    rE   [/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/psutil/_pssunos.pyvirtual_memoryy   s
   rG   c               	   C   s  t  \} }tjddtjd  ddgtjd}| \}}|t	j
j}|jdkr5d|j d	}t|| d
dd }|sHd}t|d }}|D ]"}	|	 }	|	dd \}
}|tt|
d 7 }|tt|d 7 }qN|| }t||dd}t||||| t |t S )zReport swap memory metrics.z/usr/bin/envzPATH=/usr/sbin:/sbin:PATHZswapz-l)stdoutr   z'swap -l' failed (retcode=)
r   Nzno swap device(s) configuredr   r   i   rA   )cextZswap_mem
subprocessPopenrB   environPIPEcommunicatedecodesysrI   encoding
returncodeRuntimeErrorstripsplitintr   r   ZsswaprD   )sinZsoutprI   _msglinesr4   r8   linetfr7   r6   rE   rE   rF   swap_memory   s:   	
rb   c                  C   s   t  } tdd t|  D  S )z.Return system-wide CPU times as a named tuple.c                 S   s   g | ]}t |qS rE   )sum.0xrE   rE   rF   
<listcomp>       zcpu_times.<locals>.<listcomp>)rL   per_cpu_timesr,   zipretrE   rE   rF   	cpu_times   s   rm   c                  C   s   t  } dd | D S )z6Return system per-CPU times as a list of named tuples.c                 S   s   g | ]}t | qS rE   )r,   rd   rE   rE   rF   rg      rh   z!per_cpu_times.<locals>.<listcomp>)rL   ri   rk   rE   rE   rF   ri      s   ri   c                   C   s"   zt dW S  ty   Y dS w )z0Return the number of logical CPUs in the system.SC_NPROCESSORS_ONLNN)rB   rC   
ValueErrorrE   rE   rE   rF   cpu_count_logical   s
   rp   c                   C      t  S )z-Return the number of CPU cores in the system.)rL   cpu_count_coresrE   rE   rE   rF   rr         rr   c                  C   s$   t  \} }}}d}t| |||S )z*Return various CPU stats as a named tuple.r   )rL   	cpu_statsr   Z	scpustats)Zctx_switchesZ
interruptsZsyscallsZ_trapsZsoft_interruptsrE   rE   rF   rt      s
   rt   Fc           
      C   s   g }t  }|D ]C}|\}}}}|dkrd}| s>z	t|js W qW n ty= } ztd|d|  W Y d}~qd}~ww t||||}	||	 q|S )zReturn system disk partitions.none z	skipping z: N)	rL   disk_partitions
disk_usager4   OSErrorr   r   Z	sdiskpartappend)
allretlistZ
partitions	partitionZdeviceZ
mountpointZfstypeoptserrZntuplerE   rE   rF   rw      s&   
rw   c              
   C   s   t j|  \}}t|}t }|D ]T}|\}}}	}
}}}||vr!q|	|vr&q|tthv r:|
r3t j|
 }
|r:t j| }t| }t	|}t
|	}	|dkrVt |||	|
|||}n
t |||	|
||}|| qt|S )zReturn socket connections.  If pid == -1 return system-wide
    connections (as opposed to connections opened by one process only).
    Only INET sockets are returned (UNIX are not).
    r   )r   Z	conn_tmaprL   net_connectionssetr   r   addrTCP_STATUSESr   r   Zsconnpconnaddlist)kind_pidZfamiliestypesrawlistrl   itemfdfamtype_laddrraddrr&   pidntrE   rE   rF   r     s,   


r   c                  C   sT   t  } |  D ]\}}|\}}}}ttdrt|}t||||d| |< q| S )z)Get NIC stats (isup, duplex, speed, mtu).	NicDuplexrv   )rL   net_if_statsitemshasattrr   r   Z	snicstats)rl   namer   ZisupZduplexspeedZmturE   rE   rF   r      s   

r   c                   C   rq   )z:The system boot time expressed in seconds since the epoch.)rL   	boot_timerE   rE   rE   rF   r   0  rs   r   c                  C   s\   g } t  }d}|D ]!}|\}}}}}}	|sq
||v rd}t|||||	}
| |
 q
| S )z:Return currently connected users as a list of namedtuples.)z:0.0z:0	localhost)rL   usersr   Zsuserrz   )r|   r   r   r   r-   ttyhostnameZtstampZuser_processr   r   rE   rE   rF   r   5  s   r   c                  C   s    t  t} dd t| D S )z7Returns a list of PIDs currently running on the system.c                 S   s   g | ]
}|  rt|qS rE   )isdigitrY   rd   rE   rE   rF   rg   P  s    zpids.<locals>.<listcomp>)r   encoder	   rB   listdir)r;   rE   rE   rF   pidsM  s   r   c                 C   s
   t | S )z&Check for the existence of a unix pid.)r   
pid_exists)r   rE   rE   rF   r   S  s   
r   c                    s   t   fdd}|S )zCall callable into a try/except clause and translate ENOENT,
    EACCES and EPERM in NoSuchProcess or AccessDenied exceptions.
    c              
      s   | j | j| j}}}z | g|R i |W S  ttfy6 } zt|s+t|||t||||d }~w tyG } zt	|||d }~w t
yc } z|dkr^dt v r]t	|||  d }~ww Nr   )r   _ppid_nameFileNotFoundErrorProcessLookupErrorr   r   r   PermissionErrorr
   ry   r   )selfargskwargsr   r    r   r   funrE   rF   wrapper]  s&   
z wrap_exceptions.<locals>.wrapper)	functoolswraps)r   r   rE   r   rF   wrap_exceptionsX  s   r   c                   @   s  e Zd ZdZg dZdd Zdd Zdd Zd	d
 Ze	e
dd Ze	e
dd Ze	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	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/d0 ZeZe	d1d2 Ze	d3d4 Z e	d5d6 Z!d7d8 Z"e	dHd:d;Z#e$d<d=Z%e$d<d>Z&e	d?d@ Z'e	dAdB Z(e	dCdD Z)e	dIdFdGZ*dES )JProcessz1Wrapper class around underlying C implementation.)_cacher   r   _procfs_pathr   c                 C   s   || _ d | _d | _t | _d S N)r   r   r   r   r   )r   r   rE   rE   rF   __init__z  s   zProcess.__init__c                 C   s   t | j d| j  dS )z+Raise NSP if the process disappeared on us./N)rB   statr   r   r   rE   rE   rF   _assert_alive  s   zProcess._assert_alivec                 C   (   | j |  | j|  | j|  d S r   )_proc_name_and_argsZcache_activate_proc_basic_info
_proc_credr   rE   rE   rF   oneshot_enter     zProcess.oneshot_enterc                 C   r   r   )r   Zcache_deactivater   r   r   rE   rE   rF   oneshot_exit  r   zProcess.oneshot_exitc                 C      t | j| jS r   )rL   Zproc_name_and_argsr   r   r   rE   rE   rF   r        zProcess._proc_name_and_argsc                 C   sX   | j dkrtj| j d| j  dst| j t| j | j}t|tt	ks*J |S )Nr   r   z/psinfo)
r   rB   r;   existsr   r
   rL   Zproc_basic_infolenproc_info_map)r   rl   rE   rE   rF   r     s   
zProcess._proc_basic_infoc                 C   r   r   )rL   Z	proc_credr   r   r   rE   rE   rF   r     r   zProcess._proc_credc                 C      |   d S r   r   r   rE   rE   rF   r     s   zProcess.namec                 C   s<   zt | j d| j dW S  ty   Y nw |   dS )Nr   z/path/a.outrv   )rB   readlinkr   r   ry   cmdliner   rE   rE   rF   exe  s   zProcess.exec                 C   r   )Nr   r   r   rE   rE   rF   r     s   zProcess.cmdlinec                 C   r   r   )rL   Zproc_environr   r   r   rE   rE   rF   rO        zProcess.environc                 C      |   td  S )Nr#   r   r   r   rE   rE   rF   r#     r   zProcess.create_timec                 C   r   )Nr%   r   r   rE   rE   rF   r%     r   zProcess.num_threadsc                 C   r   )Nr$   r   r   rE   rE   rF   nice_get  s   zProcess.nice_getc                 C   s&   | j dv rt| j | jt| j |S )N>   r   r   )r   r
   r   rL   Zproc_priority_set)r   valuerE   rE   rF   nice_set  s   
zProcess.nice_setc                 C   s   |   td  | _| jS )Nr    )r   r   r   r   rE   rE   rF   r      s   zProcess.ppidc                 C   s^   z|   \}}}}}}W n ty'   |  td  }|  td  }d }Y nw t|||S )Nr(   r)   r   r
   r   r   r   Zpuids)r   real	effectivesavedr\   rE   rE   rF   uids     zProcess.uidsc                 C   s^   z|   \}}}}}}W n ty'   |  td  }|  td  }d }Y nw t|||S )Nr*   r+   r   )r   r\   r   r   r   rE   rE   rF   gids  r   zProcess.gidsc              
   C   s\   z
t | j| j}W n ty( } z|jtjkrtsd}n W Y d }~nd }~ww tj	| S )N)        r   r   r   )
rL   Zproc_cpu_timesr   r   ry   errno	EOVERFLOW	IS_64_BITr   r0   )r   timesr   rE   rE   rF   rm     s   
zProcess.cpu_timesc                 C   r   r   )rL   Zproc_cpu_numr   r   r   rE   rE   rF   cpu_num  r   zProcess.cpu_numc              	   C   s~   | j }d}t|  td  }|tjkr5dD ]}zt| d| j d| W   S  t	y4   d}Y qw |r=| 
  d S d S )NFr'   )r   r   r      r   /path/T)r   r   r   r   rL   ZPRNODEVrB   r   r   r   r   )r   procfs_path
hit_enoentr   rf   rE   rE   rF   terminal  s   
"zProcess.terminalc                 C   sL   | j }zt| d| j dW S  ty%   t| d| j  Y dS w )Nr   z	/path/cwdrv   )r   rB   r   r   r   r   )r   r   rE   rE   rF   cwd  s   zProcess.cwdc                 C   s2   |   }|td  d }|td  d }t||S )Nr!   i   r"   )r   r   r9   )r   rl   r!   r"   rE   rE   rF   memory_info"  s   
zProcess.memory_infoc                 C   s   |   td  }t|dS )Nr&   ?)r   r   PROC_STATUSESget)r   coderE   rE   rF   r&   +  s   zProcess.statusc           
      C   s   | j }g }t| d| j d}d}|D ]H}t|}zt| j||\}}W n) tyP } z|jtj	kr=t
s=W Y d }~q|jtjkrKd}W Y d }~q d }~ww t|||}	||	 q|rd|   |S )Nr   z/lwpFT)r   rB   r   r   rY   rL   Zquery_process_threadry   r   r   r   ENOENTr   Zpthreadrz   r   )
r   r   rl   Ztidsr   tidutimeZstimer   r   rE   rE   rF   threads1  s0   zProcess.threadsc              	   C   s   g }d}| j }| d| j d}t| d| j dD ]1}tj||}tj|rMzt|}W n ty=   d}Y qw t	|rM|
t|t| q|rT|   |S )NFr   z/path/fdT)r   r   rB   r   r;   joinislinkr   r   r   rz   r   Z	popenfilerY   r   )r   r|   r   r   Zpathdirr   r;   filerE   rE   rF   
open_filesS  s&   zProcess.open_filesc                 c   s,   dt |g}tj|tjtjd}| \}}dd ||fD \}}|jdkrLd| v r4t| j| j	d| v rAt
| j| j	|d| }t||d	d
d }t|D ]:\}}	|	 }	|	dr|	dd
d
 }
||d
   }|dkr~tj}n
|dkrtj}nd}dtj||
dtjfV  qYdS )z<Get UNIX sockets used by process by parsing 'pfiles' output.Zpfiles)rI   stderrc                 s   s    | ]
}| tjjV  qd S r   )rR   rS   rI   rT   rd   rE   rE   rF   	<genexpr>q  s    
z,Process._get_unix_sockets.<locals>.<genexpr>r   zpermission deniedzno such processz command error
rK   r   Nzsockname: AF_UNIXr>   SOCK_STREAM
SOCK_DGRAMr   rv   )strrM   rN   rP   rQ   rU   lowerr
   r   r   r   rV   rX   	enumeratelstrip
startswithrW   socketr   r   AF_UNIXr   	CONN_NONE)r   r   cmdr[   rI   r   r]   r^   ir_   r;   typerE   rE   rF   _get_unix_socketsh  s<   



zProcess._get_unix_socketsinetc                 C   sR   t || jd}|st| j d| j  |dv r'|dd | | jD  |S )N)r   r   >   r{   unixc                 S   s   g | ]}t j| qS rE   )r   r   )re   connrE   rE   rF   rg     s    z+Process.net_connections.<locals>.<listcomp>)r   r   rB   r   r   extendr  )r   r   rl   rE   rE   rF   r     s   

zProcess.net_connectionsmmapzpath rss anon lockedzaddr perms path rss anon lockedc                 C   s*  dd }| j }g }z	t| j|}W n ty/ } z|jtjkr*ts*g W  Y d }~S  d }~ww d}|D ]X}|\}}	}
}}}}|||	}|dszt	
| d| j d| }W n' ty } z|jtjkru| d| j d| }d}n W Y d }~nd }~ww |||
||||f q4|r|   |S )Nc                 S   s0   d t| dd  dt|dd  dS )Nz{}-{}r   L)formathexrW   )startendrE   rE   rF   toaddr  s   (z#Process.memory_maps.<locals>.toaddrF[r   r   T)r   rL   Zproc_memory_mapsr   ry   r   r   r   r   rB   r   r   rz   r   )r   r  r   r|   r   r   r   r   r   Zaddrsizepermr   r!   Zanonr<   rE   rE   rF   memory_maps  s<   

 zProcess.memory_mapsc                 C   s   t t| j d| j dS )Nr   r   )r   rB   r   r   r   r   rE   rE   rF   num_fds  s   zProcess.num_fdsc                 C   s   t jt| j| j S r   )r   ZpctxswrL   Zproc_num_ctx_switchesr   r   r   rE   rE   rF   num_ctx_switches  s   zProcess.num_ctx_switchesNc                 C   s   t | j|| jS r   )r   Zwait_pidr   r   )r   timeoutrE   rE   rF   wait  s   zProcess.wait)r  r   )+__name__
__module____qualname____doc__	__slots__r   r   r   r   r   r   r   r   r   r   r   r   rO   r#   r%   r   r   r    r   r   rm   r   r   r   r   Zmemory_full_infor&   r   r   r  r   r   Znt_mmap_groupedZnt_mmap_extr  r  r  r  rE   rE   rE   rF   r   u  s    	







	

	
	






!
"


.

r   )F)r   )kr  r   r   rB   r   rM   rS   collectionsr   r   rv   r   r   r   rL   r   r	   r
   r   r   r   r   r   r   r   r   r   Z__extra__all__ZgetpagesizerD   ZAF_LINKmaxsizer   r   r   ZSSLEEPZSTATUS_SLEEPINGZSRUNZSTATUS_RUNNINGZSZOMBZSTATUS_ZOMBIEZSSTOPZSTATUS_STOPPEDZSIDLZSTATUS_IDLEZSONPROCZSWAITZSTATUS_WAITINGr   ZTCPS_ESTABLISHEDZCONN_ESTABLISHEDZTCPS_SYN_SENTZCONN_SYN_SENTZTCPS_SYN_RCVDZ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_NONEr   Z	TCPS_IDLEZ
TCPS_BOUNDr   dictr   r,   r0   r3   r9   Zpfullmemr:   r   _fieldsr=   rG   rb   rm   ri   rp   rr   rt   Zdisk_io_countersrx   rw   Znet_io_countersZnet_if_addrsr   r   r   r   r   r   r   r   rE   rE   rE   rF   <module>   s   

.	

