U
    ;g+t                  R   @   s  d Z ddlmZ ddlm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
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 zdd	l	mZ W n ek
r   dZY nX zdd
l	mZ W n ek
r   dZY nX ejd dkZerddlZndZeedZe Z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&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^gRZejd_kZejd`kZ ej!"daZ#ej!"dbZ$e$Z%ej!"dcZ&ej!"ddZ'ej!"deZ(e&p0e'p0e(Z)ej!"dfZ*ej!"dgZ+dhZ,diZ-djZ.dkZ/dlZ0dmZ1dnZ2doZ3dpZ4dqZ5drZ6dsZ7dtZ8duZ9dvZ:dwZ;dxZ<dyZ=dzZ>d{Z?d|Z@d}ZAd~ZBdZCdZDdZEedkrdZFdZGdZHn G dd dejIZJeK LeJjM edkrdZNdZOn G dd dejIZPeK LePjM eQ ZRes2dZSn2zeT ZSW n$ eUk
rb   erZdndZSY nX edHddddddgZVedDddddgZWedBddddddgZXedCddddgZYedEddddddddgZZedIdddddgZ[ed@dddddddgZ\edFdddddgZ]edGdddddgZ^edAddddgZ_eddddgZ`edddddgZaeddddgZbedddgZced8ddddgZded=ddgZeed>dddgZfed?dddgZged:dddgZhed;ddddgZied<ddgZjed9ddgZked7ddddddgZledddgZmeeegeegfeegegfegegfeegegfegegfeegeegfegeegfegeegfdڜZnedk	renLegegfegegfdۜ edk	renLdegeegfi G ddބ deoZpG dd depZqG dd deqZrG dd depZsG dd depZterJeuevewr6evd Zxn
eyevdZxexd ndd ZzdddPZ{ddM Z|dd Z}ddL Z~ddO Ze|ddQ ZddN ZddR ZddS ZdddZZddK ZG dd dZddT Ze Zeje_eje_dZddV ZddU ZeefddWZefddXZdd dYZdd Zer>dd Zndd Ze|ejfdd]Zddd\Zddejfd	d^Zd
d[ ZdS (  z9Common objects shared by __init__.py and _ps*.py modules.    )division)print_functionN)
namedtuple)AF_INET)
SOCK_DGRAM)SOCK_STREAM)AF_INET6)AF_UNIX   PSUTIL_DEBUGFREEBSDBSDLINUXNETBSDOPENBSDMACOSOSXPOSIXSUNOSWINDOWS
CONN_CLOSECONN_CLOSE_WAITCONN_CLOSINGCONN_ESTABLISHEDCONN_FIN_WAIT1CONN_FIN_WAIT2CONN_LAST_ACKCONN_LISTEN	CONN_NONECONN_SYN_RECVCONN_SYN_SENTCONN_TIME_WAITNIC_DUPLEX_FULLNIC_DUPLEX_HALFNIC_DUPLEX_UNKNOWNSTATUS_DEADSTATUS_DISK_SLEEPSTATUS_IDLESTATUS_LOCKEDSTATUS_RUNNINGSTATUS_SLEEPINGSTATUS_STOPPEDSTATUS_SUSPENDEDSTATUS_TRACING_STOPSTATUS_WAITINGSTATUS_WAKE_KILLSTATUS_WAKINGSTATUS_ZOMBIESTATUS_PARKEDENCODINGENCODING_ERRSr   pconn	pcputimespctxswpgidspiopionice	popenfilepthreadpuidssconn	scpustatssdiskio	sdiskpart
sdiskusagesnetiosnicaddr	snicstatssswapsuser	conn_tmapdeprecated_methodisfile_strictmemoizeparse_environ_blockpath_exists_strictusage_percentsupports_ipv6sockfam_to_enumsocktype_to_enumwrap_numbers	open_textopen_binarycatbcatbytes2humanconn_to_ntupledebughiliteterm_supports_colorsprint_colorposixntlinuxdarwin)ZfreebsdZmidnightbsdZopenbsdZnetbsd)sunossolarisaixZrunningZsleepingz
disk-sleepstoppedztracing-stopZzombieZdeadz	wake-killZwakingZidlelockedZwaitingZ	suspendedZparkedZESTABLISHEDZSYN_SENTZSYN_RECVZ	FIN_WAIT1Z	FIN_WAIT2Z	TIME_WAITZCLOSEZ
CLOSE_WAITZLAST_ACKZLISTENZCLOSINGNONE      c                   @   s   e Zd ZdZdZdZdS )	NicDuplexrg   rh   r   N)__name__
__module____qualname__r"   r#   r$    rm   rm   2/tmp/pip-unpacked-wheel-u0epr1ex/psutil/_common.pyri      s   ri   c                   @   s   e Zd ZdZdZdS )BatteryTimero   rp   N)rj   rk   rl   POWER_TIME_UNKNOWNPOWER_TIME_UNLIMITEDrm   rm   rm   rn   rq      s   rq   replacesurrogateescapetotalusedfreepercentsinZsoutZ
read_countZwrite_count
read_byteswrite_bytesZ	read_timeZ
write_timeZdeviceZ
mountpointZfstypeoptsZ
bytes_sentZ
bytes_recvZpackets_sentZpackets_recvZerrinZerroutZdropinZdropoutnameZterminalhoststartedpidfdfamilytypeladdrraddrstatusaddressnetmask	broadcastZptpZisupZduplexspeedmtuflagsZctx_switchesZ
interruptsZsoft_interruptsZsyscallsscpufreqcurrentminmaxshwtemplabelhighcriticalsbatteryZsecsleftZpower_pluggedsfanusersystemchildren_userchildren_systempathidZ	user_timeZsystem_timerealZ	effectiveZsavedZioclassvalueZ	voluntaryZinvoluntaryaddripport)allZtcpZtcp4ZudpZudp4ZinetZinet4Zinet6)Ztcp6Zudp6unixc                   @   s,   e Zd ZdZdZdd Zdd Zdd Zd	S )
ErrorzQBase exception class. All other psutil exceptions inherit
    from this one.
    psutilc                 C   sH   t  }|D ]6}t| |d }|r*|||< q|dkr|dkr|||< q|S )Nr   r   )collectionsOrderedDictgetattr)selfattrsinfor~   r   rm   rm   rn   	_infodict$  s    

zError._infodictc                 C   sP   |  d}|r,dddd | D  }nd }ddd t| dd	|fD S )
N)r   ppidr~   z(%s), c                 S   s   g | ]\}}d ||f qS z%s=%rrm   .0kvrm   rm   rn   
<listcomp>3  s     z!Error.__str__.<locals>.<listcomp> c                 S   s   g | ]}|r|qS rm   rm   )r   xrm   rm   rn   r   7  s      msg )r   joinitemsr   r   r   detailsrm   rm   rn   __str__.  s    
zError.__str__c                 C   s2   |  d}ddd | D }d| jj|f S )N)r   r   r~   secondsr   r   c                 S   s   g | ]\}}d ||f qS r   rm   r   rm   rm   rn   r   <  s     z"Error.__repr__.<locals>.<listcomp>zpsutil.%s(%s))r   r   r   	__class__rj   r   rm   rm   rn   __repr__9  s    
zError.__repr__N)rj   rk   rl   __doc__r   r   r   rm   rm   rm   rn   r     s
   
r   c                   @   s&   e Zd ZdZdZdddZdd ZdS )	NoSuchProcesszXException raised when a process with a certain PID doesn't
    or no longer exists.
    r   Nc                 C   s$   t |  || _|| _|pd| _d S )Nzprocess no longer existsr   __init__r   r~   r   r   r   r~   r   rm   rm   rn   r   G  s    
zNoSuchProcess.__init__c                 C   s   | j | j| j| jffS Nr   r   r~   r   r   rm   rm   rn   
__reduce__M  s    zNoSuchProcess.__reduce__)NNrj   rk   rl   r   r   r   rm   rm   rm   rn   r   @  s   
r   c                   @   s&   e Zd ZdZdZdddZdd ZdS )	ZombieProcessa1  Exception raised when querying a zombie process. This is
    raised on macOS, BSD and Solaris only, and not always: depending
    on the query the OS may be able to succeed anyway.
    On Linux all zombie processes are querable (hence this is never
    raised). Windows doesn't have zombie processes.
    r   Nc                 C   s$   t | ||| || _|pd| _d S )Nz"PID still exists but it's a zombie)r   r   r   r   )r   r   r~   r   r   rm   rm   rn   r   [  s    zZombieProcess.__init__c                 C   s   | j | j| j| j| jffS r   )r   r   r~   r   r   r   rm   rm   rn   r   `  s    zZombieProcess.__reduce__)NNNr   rm   rm   rm   rn   r   Q  s   
r   c                   @   s&   e Zd ZdZdZdddZdd ZdS )	AccessDeniedz@Exception raised when permission to perform an action is denied.r   Nc                 C   s$   t |  || _|| _|pd| _d S )Nr   r   r   rm   rm   rn   r   i  s    
zAccessDenied.__init__c                 C   s   | j | j| j| jffS r   r   r   rm   rm   rn   r   o  s    zAccessDenied.__reduce__)NNNr   rm   rm   rm   rn   r   d  s   
r   c                   @   s&   e Zd ZdZdZdddZdd ZdS )	TimeoutExpiredzWRaised on Process.wait(timeout) if timeout expires and process
    is still alive.
    r   Nc                 C   s*   t |  || _|| _|| _d| | _d S )Nztimeout after %s seconds)r   r   r   r   r~   r   )r   r   r   r~   rm   rm   rn   r   z  s
    
zTimeoutExpired.__init__c                 C   s   | j | j| j| jffS r   )r   r   r   r~   r   rm   rm   rn   r     s    zTimeoutExpired.__reduce__)NNr   rm   rm   rm   rn   r   s  s   
r   execzvdef raise_from(value, from_value):
    try:
        raise value from from_value
    finally:
        value = None
    c                 C   s   | d S r   rm   )r   
from_valuerm   rm   rn   
raise_from  s    r   c                 C   sF   zt | | d }W n tk
r*   Y dS X |dk	r>t||}|S dS )z5Calculate percentage usage of 'used' against 'total'.d   g        N)floatZeroDivisionErrorround)rw   rv   Zround_retrm   rm   rn   rN     s    
c                    s2   t  fdd} fdd}i  ||_|S )a  A simple memoize decorator for functions supporting (hashable)
    positional arguments.
    It also provides a cache_clear() function for clearing the cache:

    >>> @memoize
    ... def foo()
    ...     return 1
        ...
    >>> foo()
    1
    >>> foo.cache_clear()
    >>>

    It supports:
     - functions
     - classes (acts as a @singleton)
     - staticmethods
     - classmethods

    It does NOT support:
     - methods
    c                     s   | t t| f}z
 | W S  tk
rz   z| | } |< W n, tk
rp } zt|d W 5 d }~X Y nX | Y S X d S r   )	frozensetsortedr   KeyError	Exceptionr   )argskwargskeyr   errcachefunrm   rn   wrapper  s    
zmemoize.<locals>.wrapperc                      s       dS )zClear cache.N)clearrm   )r   rm   rn   cache_clear  s    zmemoize.<locals>.cache_clear)	functoolswrapsr   )r   r   r   rm   r   rn   rK     s    c                    s6   t   fdd}dd }dd }||_||_|S )a  A memoize decorator which is disabled by default. It can be
    activated and deactivated on request.
    For efficiency reasons it can be used only against class methods
    accepting no arguments.

    >>> class Foo:
    ...     @memoize
    ...     def foo()
    ...         print(1)
    ...
    >>> f = Foo()
    >>> # deactivated (default)
    >>> foo()
    1
    >>> foo()
    1
    >>>
    >>> # activated
    >>> foo.cache_activate(self)
    >>> foo()
    1
    >>> foo()
    >>> foo()
    >>>
    c                    s   z| j   }W n tk
r^   z | W  Y S  tk
rX } zt|d W 5 d }~X Y nX Y nr tk
r   z | }W n, tk
r } zt|d W 5 d }~X Y nX z|| j  < W n tk
r   Y nX Y nX |S r   )_cacheAttributeErrorr   r   r   )r   r   r   r   rm   rn   r     s"     z'memoize_when_activated.<locals>.wrapperc                 S   s
   i | _ dS )zsActivate cache. Expects a Process instance. Cache will be
        stored as a "_cache" instance attribute.
        N)r   procrm   rm   rn   cache_activate  s    z.memoize_when_activated.<locals>.cache_activatec                 S   s"   z| ` W n tk
r   Y nX dS )zDeactivate and clear cache.N)r   r   r   rm   rm   rn   cache_deactivate  s    z0memoize_when_activated.<locals>.cache_deactivate)r   r   r   r   )r   r   r   r   rm   r   rn   memoize_when_activated  s    r   c              
   C   s\   zt | }W n< tk
rJ } z|jtjtjfkr4 W Y dS d}~X Y nX t|jS dS )zSame as os.path.isfile() but does not swallow EACCES / EPERM
    exceptions, see:
    http://mail.python.org/pipermail/python-dev/2012-June/120787.html.
    FN)osstatOSErrorerrnoEPERMEACCESS_ISREGst_mode)r   str   rm   rm   rn   rJ     s    c              
   C   sT   zt |  W n< tk
rJ } z|jtjtjfkr4 W Y dS d}~X Y nX dS dS )zSame as os.path.exists() but does not swallow EACCES / EPERM
    exceptions. See:
    http://mail.python.org/pipermail/python-dev/2012-June/120787.html.
    FNT)r   r   r   r   r   r   )r   r   rm   rm   rn   rM   -  s    c               	   C   sd   t jrtdkrdS z4t  tt j} t|  | d W 5 Q R X W dS  t jk
r^   Y dS X dS )z2Return True if IPv6 is supported on this platform.NF)z::1r   T)sockethas_ipv6r   r   
contextlibclosingbinderror)sockrm   rm   rn   rO   <  s    c                 C   sv   i }d}t }| d|}||kr"qr| d||}||krh| || }| |d | }|r`| }|||< |d }q|S )zCParse a C environ block of environment variables into a dictionary.r    =rh   )r   findupper)datar   posZWINDOWS_Znext_posZ	equal_posr   r   rm   rm   rn   rL   J  s    
c                 C   s6   t dkr| S zt| W S  tk
r0   |  Y S X dS )zConvert a numeric socket family value to an IntEnum member.
    If it's not a known member, return the numeric value itself.
    N)enumr   AddressFamily
ValueErrornumrm   rm   rn   rP   f  s    c                 C   s6   t dkr| S zt| W S  tk
r0   |  Y S X dS )zConvert a numeric socket type value to an IntEnum member.
    If it's not a known member, return the numeric value itself.
    N)r  r   
SocketKindr  r  rm   rm   rn   rQ   s  s    c                 C   s   |t jtfkr&|rt| }|r&t| }|t jkrJ|ttfkrJ||t}nt}t|}t|}|dkrxt	| |||||S t
| ||||||S dS )z2Convert a raw connection tuple to a proper ntuple.N)r   r   r   r   r   getr   rP   rQ   r5   r>   )r   Zfamtype_r   r   r   Z
status_mapr   rm   rm   rn   rX     s    c                    s    fdd}|S )zA decorator which can be used to mark a method as deprecated
    'replcement' is the method name which will be called instead.
    c                    s:   d| j f  | jd kr | _t|  fdd}|S )Nz8%s() is deprecated and will be removed; use %s() insteadc                    s    t j tdd t| ||S )Nrg   )category
stacklevel)warningswarnDeprecationWarningr   )r   r   r   )r   replacementrm   rn   inner  s    z/deprecated_method.<locals>.outer.<locals>.inner)rj   r   r   r   )r   r  r  )r   rn   outer  s    
z deprecated_method.<locals>.outerrm   )r  r  rm   r  rn   rI     s    c                   @   sB   e Zd ZdZdd Zdd Zdd Zdd	 ZdddZdd Z	d
S )_WrapNumberszNWatches numbers so that they don't overflow and wrap
    (reset to zero).
    c                 C   s    t  | _i | _i | _i | _d S r   )	threadingLocklockr   	remindersreminder_keysr   rm   rm   rn   r     s    
z_WrapNumbers.__init__c                 C   sX   || j kst|| jkst|| jks*t|| j |< tt| j|< tt| j|< d S r   )r   AssertionErrorr  r  r   defaultdictintset)r   
input_dictr~   rm   rm   rn   	_add_dict  s    
z_WrapNumbers._add_dictc                 C   s\   | j | }t| t|  }|D ]0}| j| | D ]}| j| |= q8| j| |= q&dS )zIn case the number of keys changed between calls (e.g. a
        disk disappears) this removes the entry from self.reminders.
        N)r   r  keysr  r  )r   r   r~   old_dictZ	gone_keysZgone_keyremkeyrm   rm   rn   _remove_dead_reminders  s    
z#_WrapNumbers._remove_dead_remindersc              	   C   s  || j kr| || |S | || | j | }i }|D ]}|| }z|| }W n  tk
rp   |||< Y q8Y nX g }tt|D ]f}	||	 }
||	 }||	f}|
|k r| j| |  |7  < | j| | | |	|
| j| |   qt
|||< q8|| j |< |S )zlCache dict and sum numbers which overflow and wrap.
        Return an updated copy of `input_dict`.
        )r   r!  r%  r   rangelenr  r  addappendtuple)r   r   r~   r#  Znew_dictr   Zinput_tupleZ	old_tuplebitsiZinput_value	old_valuer$  rm   rm   rn   run  s2    



z_WrapNumbers.runNc              	   C   sh   | j X |dkr0| j  | j  | j  n*| j|d | j|d | j|d W 5 Q R X dS )z>Clear the internal cache, optionally only for function 'name'.N)r  r   r   r  r  pop)r   r~   rm   rm   rn   r     s    

z_WrapNumbers.cache_clearc              
   C   s.   | j  | j| j| jfW  5 Q R  S Q R X dS )z5Return internal cache dicts as a tuple of 3 elements.N)r  r   r  r  r   rm   rm   rn   
cache_info  s    z_WrapNumbers.cache_info)N)
rj   rk   rl   r   r   r!  r%  r.  r   r0  rm   rm   rm   rn   r    s   '
r  c              
   C   s*   t j t | |W  5 Q R  S Q R X dS )zGiven an `input_dict` and a function `name`, adjust the numbers
    which "wrap" (restart from zero) across different calls by adding
    "old value" to "new value" and return an updated dict.
    N)_wnr  r.  )r   r~   rm   rm   rn   rR     s    i   c                 C   s   t | dtdS )Nrb	buffering)openFILE_READ_BUFFER_SIZE)fnamerm   rm   rn   rT     s    c                 C   s`   t st| tdS t| tttd}z
t|_W n0 tk
r>   Y n tk
rZ   |   Y nX |S )zOn Python 3 opens a file in text mode by using fs encoding and
    a proper en/decoding errors handler.
    On Python 2 this is just an alias for open(name, 'rt').
    r3  )r4  encodingerrors)	PY3r5  r6  r3   r4   _CHUNK_SIZEr   r   close)r7  Zfobjrm   rm   rn   rS   "  s     
c              
   C   sz   |t kr.|| }| W  5 Q R  S Q R X nHz*|| }| W  5 Q R  W S Q R X W n ttfk
rt   | Y S X dS )zRead entire file content and return it as a string. File is
    opened in text mode. If specified, `fallback` is the value
    returned in case of error, either if the file does not exist or
    it can't be read().
    N)_DEFAULTreadIOErrorr   )r7  fallback_openfrm   rm   rn   rU   @  s    

 c                 C   s   t | |tdS )z,Same as above but opens file in binary mode.)r@  rA  )rU   rT   )r7  r@  rm   rm   rn   rV   Q  s    %(value).1f%(symbol)sc                 C   s   d}i }t |dd D ]\}}d|d d > ||< qt|dd D ]2}t| || krFt| ||  }|t    S qF|t|d | d S )zUsed by various scripts. See: http://goo.gl/zeJZl.

    >>> bytes2human(10000)
    '9.8K'
    >>> bytes2human(100001221)
    '95.4M'
    )	BKMGTPEZYrh   N
   r   )symbolr   )	enumeratereversedabsr   localsdict)nformatsymbolsprefixr,  srN  r   rm   rm   rn   rW   V  s    c                   C   s   t jd jS )z+Return updated psutil.PROCFS_PATH constant.r   )sysmodulesZPROCFS_PATHrm   rm   rm   rn   get_procfs_pathi  s    r[  c                 C   s   | j ttdS )N)r8  r9  )decoder3   r4   rX  rm   rm   rn   r\  p  s    r\  c                 C   s   | S r   rm   r]  rm   rm   rn   r\  u  s    c                 C   s`   t jdkrdS z2dd l}|  s$t|  |ddks>tW n tk
rV   Y dS X dS d S )Nr^   Tr   colorsF)r   r~   cursesisattyr  Z	setuptermZtigetnumr   )filer_  rm   rm   rn   r[   ~  s    
Fc                 C   s   t  s
| S g }tddddddddd	d
	}d|d< z|| }W n( tk
rd   tdt|  Y nX || |r~|d dd|| f S )z*Return an highlighted version of 'string'.Z34Z33Z30Z32Z37Z36Z91Z35Z93)	ZbluebrownZdarkgreygreenZgreyZ	lightblueredZvioletyellowZ29Nz#invalid color %r; choose between %s1z[%sm%s[0m;)r[   rS  r   r  listr"  r)  r   )rX  colorboldattrr^  rm   rm   rn   rZ     s2    


c              	   C   s  t  st| |d ntr.tt| |||d nddl}d}|jjj}|jjj}t	ddddd}||d< z|| }W n, t
k
r   td	|t| f Y nX |r|dkr|d
7 }|tjkrdnd}	|j|_||	}
||
| zt| |d W 5 ||
| X dS )z$Print a colorized version of string.ra  r   N   rg         )rc  rd  rb  re  z#invalid color %r; choose between %r   ii)r[   printr   rZ   ctypeswindllKernel32GetStdHandleSetConsoleTextAttributerS  r   r  rh  r"  rY  stderrc_ulongrestype)rX  ri  rj  ra  rr  ZDEFAULT_COLORru  rv  r^  Z	handle_idhandlerm   rm   rn   r\     s8    


c                 C   sn   t rjddl}|| j\}}}}}t| trRt| ttt	frJd|  } nd|  } t
d||| f tjd dS )z@If PSUTIL_DEBUG env var is set, print a debug message to stderr.r   Nzignoring %szignoring %rzpsutil-debug [%s:%s]> %srl  )r   inspectgetframeinfocurrentframef_back
isinstancer   r   r?  EnvironmentErrorrq  rY  rw  )r   r{  r7  lineno__lines_indexrm   rm   rn   rY     s    

 )N)N)rC  )NF)r   
__future__r   r   r   r   r   r   r   r   r   rY  r  r  r   r   r   r   r   ImportErrorr	   version_infor:  r  boolgetenvr   objectr=  __all__r~   r   r   platform
startswithr   r   r   r   r   r   r   r   ZAIXr)   r*   r&   r+   r-   r1   r%   r/   r0   r'   r(   r.   r,   r2   r   r    r   r   r   r!   r   r   r   r   r   r   r"   r#   r$   IntEnumri   globalsupdate__members__rr   rs   rq   getfilesystemencodingr3   r4   getfilesystemencodeerrorsr   rF   rB   r@   rA   rC   rG   r>   rD   rE   r?   r   r   r   r   r6   r;   r<   r=   r8   r9   r:   r7   r5   r   rH   r   r   r   r   r   r   r  __builtins__rS  exec_r   r   rN   rK   r   rJ   rM   rO   rL   rP   rQ   rX   rI   r  rR   r1  r   r0  r6  rT   rS   rU   rV   rW   r[  r\  stdoutr[   rZ   r\   rY   rm   rm   rm   rn   <module>   sD  


                                                              &

	

       
 
 
 
 	





#


-G
W		  '