U jb?@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddlm Z ddlm Z ddlm Z zddlm Z Wnek rdZ YnXzddlmZWnek rdZYnXeedZeZd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGg?ZejdHkZejdIkZejdJZejdKZeZejdLZejdMZejdNZ epepe Z!ejdOZ"ejdPZ#dQZ$dRZ%dSZ&dTZ'dUZ(dVZ)dWZ*dXZ+dYZ,dZZ-d[Z.d\Z/d]Z0d^Z1d_Z2d`Z3daZ4dbZ5dcZ6ddZ7deZ8dfZ9dgZ:dhZ;diZZ?e@Ae?jBGdmdndnej>ZCe@AeCjBeDZEeFZGe e ege e gfe e ge gfe ge gfe e ge gfe ge gfe e ge e gfe ge e gfe ge e gfdoZHe dk r eHAe ge gfe ge gfdpedk rHe"sHeHAdqege e gfiGdrdsdseIZJGdtdudueJZKGdvdwdweKZLGdxdydyeJZMGdzd{d{eJZNdd|d9ZOd}d6ZPd~dZQdd5ZRdd8ZSdd:ZTdd7ZUdd;ZVdd<ZWdddCZXddZYdd4ZZGdddZ[dd=Z\e[Z]e]j^e\_^e]j_e\__dZ`dd?Zadd>Zbeebfdd@ZcefddAZddddBZeddZfddZgePejhfddFZidddEZjddejhfddGZkddDZldS)zCommon objects shared by __init__.py and _ps*.py modules. Note: this module is imported by setup.py, so it should not import psutil or third-party modules. N)AF_INET) SOCK_DGRAM) SOCK_STREAM)AF_INET6)AF_UNIX PSUTIL_DEBUGFREEBSDBSDLINUXNETBSDOPENBSDMACOSOSXPOSIXSUNOSWINDOWS CONN_CLOSECONN_CLOSE_WAIT CONN_CLOSINGCONN_ESTABLISHEDCONN_FIN_WAIT1CONN_FIN_WAIT2 CONN_LAST_ACK CONN_LISTEN CONN_NONE CONN_SYN_RECV CONN_SYN_SENTCONN_TIME_WAITNIC_DUPLEX_FULLNIC_DUPLEX_HALFNIC_DUPLEX_UNKNOWN STATUS_DEADSTATUS_DISK_SLEEP STATUS_IDLE STATUS_LOCKEDSTATUS_RUNNINGSTATUS_SLEEPINGSTATUS_STOPPEDSTATUS_SUSPENDEDSTATUS_TRACING_STOPSTATUS_WAITINGSTATUS_WAKE_KILL STATUS_WAKING STATUS_ZOMBIE STATUS_PARKEDENCODING ENCODING_ERRSr conn_tmapdeprecated_method isfile_strictmemoizeparse_environ_blockpath_exists_strict usage_percent supports_ipv6sockfam_to_enumsocktype_to_enum wrap_numbers open_text open_binarycatbcat bytes2humanconn_to_ntupledebughiliteterm_supports_colors print_colorposixntlinuxdarwin)ZfreebsdZ midnightbsdZopenbsdZnetbsd)sunossolarisaixZrunningZsleepingz disk-sleepstoppedz tracing-stopZzombieZdeadz wake-killZwakingZidlelockedZwaitingZ suspendedZparkedZ ESTABLISHEDZSYN_SENTZSYN_RECVZ FIN_WAIT1Z FIN_WAIT2Z TIME_WAITZCLOSEZ CLOSE_WAITZLAST_ACKZLISTENZCLOSINGNONEc@seZdZdZdZdZdS) NicDuplexrN)__name__ __module__ __qualname__rrr rVrV2/tmp/pip-unpacked-wheel-3no1_rt3/psutil/_common.pyrPzsrPc@seZdZdZdZdS) BatteryTimeN)rSrTrUZPOWER_TIME_UNKNOWNZPOWER_TIME_UNLIMITEDrVrVrVrWrXsrX)allZtcpZtcp4ZudpZudp4ZinetZinet4Zinet6)Ztcp6Zudp6unixc@s,eZdZdZdZddZddZddZd S) ErrorzQBase exception class. All other psutil exceptions inherit from this one. psutilcCs>t}|D],}t||d}|s0|dkr |dkr |||<q |S)Npidr) collections OrderedDictgetattr)selfattrsinfonamevaluerVrVrW _infodicts   zError._infodictcCsR|d}|r.dddd|D}nd}dddt|dd |fDS) N)r_ppidrfz({}), cSsg|]\}}|d|qS=rV.0kvrVrVrW sz!Error.__str__.. cSsg|] }|r|qSrVrV)rnxrVrVrWrqsmsg)rhformatjoinitemsrbrcredetailsrVrVrW__str__s z Error.__str__cCs8|d}ddd|D}d|jjd|dS)N)r_rirfsecondsrtrjcSsg|]\}}|d|qSrkrVrmrVrVrWrqsz"Error.__repr__..zpsutil.())rhrwrx __class__rSryrVrVrW__repr__s zError.__repr__N)rSrTrU__doc__rhr{rrVrVrVrWr]s  r]c@s&eZdZdZdZdddZddZdS) NoSuchProcesszXException raised when a process with a certain PID doesn't or no longer exists. r^NcCs$t|||_||_|pd|_dS)Nzprocess no longer existsr]__init__r_rfrtrcr_rfrtrVrVrWrs zNoSuchProcess.__init__cCs|j|j|j|jffSNrr_rfrtrcrVrVrW __reduce__szNoSuchProcess.__reduce__)NNrSrTrUrrrrVrVrVrWrs rc@s&eZdZdZdZdddZddZdS) ZombieProcessa1Exception 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^NcCs$t||||||_|pd|_dS)Nz"PID still exists but it's a zombie)rrrirt)rcr_rfrirtrVrVrWrszZombieProcess.__init__cCs|j|j|j|j|jffSr)rr_rfrirtrrVrVrWrszZombieProcess.__reduce__)NNNrrVrVrVrWrs rc@s&eZdZdZdZdddZddZdS) AccessDeniedz@Exception raised when permission to perform an action is denied.r^NcCs$t|||_||_|pd|_dS)NrurrrVrVrWrs zAccessDenied.__init__cCs|j|j|j|jffSrrrrVrVrWrszAccessDenied.__reduce__)NNNrrVrVrVrWrs rc@s&eZdZdZdZdddZddZdS) TimeoutExpiredzWRaised on Process.wait(timeout) if timeout expires and process is still alive. r^NcCs.t|||_||_||_d|d|_dS)Nztimeout after z seconds)r]rr|r_rfrt)rcr|r_rfrVrVrWr s  zTimeoutExpired.__init__cCs|j|j|j|jffSr)rr|r_rfrrVrVrWrszTimeoutExpired.__reduce__)NNrrVrVrVrWrs rcCsFzt||d}Wntk r*YdSX|dk r>t||}|SdS)z5Calculate percentage usage of 'used' against 'total'.dgN)floatZeroDivisionErrorround)usedtotalZround_retrVrVrWr7s cs2tfdd}fdd}i||_|S)aA 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||tt|f}z |WStk rvz||}|<Wn(tk rl}z |dW5d}~XYnX|YSXdSr) frozensetsortedrxKeyError Exception)argskwargskeyrerrcachefunrVrWwrapper?s zmemoize..wrappercs dS)z Clear cache.N)clearrV)rrVrW cache_clearKszmemoize..cache_clear) functoolswrapsr)rrrrVrrWr4's  cs6tfdd}dd}dd}||_||_|S)aA 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 sz|j}Wntk rZz|WYStk rT}z |dW5d}~XYnXYnntk rz |}Wn(tk r}z |dW5d}~XYnXz||j<Wntk rYnXYnX|Sr)_cacheAttributeErrorrr)rcrrrrVrWros"  z'memoize_when_activated..wrappercSs i|_dS)zsActivate cache. Expects a Process instance. Cache will be stored as a "_cache" instance attribute. N)rprocrVrVrWcache_activatesz.memoize_when_activated..cache_activatecSs"z|`Wntk rYnXdS)zDeactivate and clear cache.N)rrrrVrVrWcache_deactivatesz0memoize_when_activated..cache_deactivate)rrrr)rrrrrVrrWmemoize_when_activatedTsrcCsJzt|}Wn*tk r$Yn"tk r8YdSXt|jSdS)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)osstatPermissionErrorOSErrorS_ISREGst_mode)pathstrVrVrWr3scCsBzt|Wn*tk r$Yntk r8YdSXdSdS)zSame as os.path.exists() but does not swallow EACCES / EPERM exceptions. See: http://mail.python.org/pipermail/python-dev/2012-June/120787.html. FTN)rrrr)rrVrVrWr6sc CsXtjrtdkrdSz*tttj}|dW5QRXWdStk rRYdSXdS)z2Return True if IPv6 is supported on this platform.NF)z::1rT)sockethas_ipv6rrbindr)sockrVrVrWr8scCsvi}d}t}|d|}||kr"qr|d||}||krh|||}||d|}|r`|}|||<|d}q |S)zCParse a C environ block of environment variables into a dictionary.rrlrR)rfindupper)datarposZWINDOWS_Znext_posZ equal_posrrgrVrVrWr5s   cCs*z t|WStk r$|YSXdS)zConvert a numeric socket family value to an IntEnum member. If it's not a known member, return the numeric value itself. N)r AddressFamily ValueErrornumrVrVrWr9s cCs*z t|WStk r$|YSXdS)zConvert a numeric socket type value to an IntEnum member. If it's not a known member, return the numeric value itself. N)r SocketKindrrrVrVrWr:s c Csddlm}|tjthkr6|r(|j|}|r6|j|}|tjkrZ|tthkrZ||t}nt}t |}t |}|dkr| ||||||S| |||||||SdS)z2Convert a raw connection tuple to a proper ntuple.rR)_ntuplesN) rurrrraddrrgetrr9r:pconnZsconn) fdZfamtype_laddrraddrstatusZ status_mapr_ZntprVrVrWrAs   cCsxddl}|jr|jsdS|jtjkrFt|j|jd|jddjS|jtj krtt|j |jd|jddjSdS)zbGiven the address ntuple returned by ``net_if_addrs()`` calculates the broadcast address. rN/F)strict) ipaddressaddressnetmaskfamilyrrstr IPv4Networkbroadcast_addressr IPv6Network)rrrVrVrWbroadcast_addr s"   rcsfdd}|S)zA decorator which can be used to mark a method as deprecated 'replcement' is the method name which will be called instead. cs>|jdd|jdkr"|_t|fdd}|S)Nz*() is deprecated and will be removed; use z () insteadcs tjtddt|||S)NrQ)category stacklevel)warningswarnDeprecationWarningrb)rcrr)rt replacementrVrWinner-sz/deprecated_method..outer..inner)rSrrr)rrr)rtrWouter%s z deprecated_method..outerrV)rrrVrrWr2 s c@sBeZdZdZddZddZddZdd Zdd d Zd dZ d S) _WrapNumberszNWatches numbers so that they don't overflow and wrap (reset to zero). cCs t|_i|_i|_i|_dSr) threadingLocklockr reminders reminder_keysrrVrVrWr<s z_WrapNumbers.__init__cCsX||jkst||jkst||jks*t||j|<tt|j|<tt|j|<dSr)rAssertionErrorrrr` defaultdictintset)rc input_dictrfrVrVrW _add_dictBs  z_WrapNumbers._add_dictcCs\|j|}t|t|}|D]0}|j||D]}|j||=q8|j||=q&dS)zIn case the number of keys changed between calls (e.g. a disk disappears) this removes the entry from self.reminders. N)rrkeysrr)rcrrfold_dictZ gone_keysZgone_keyremkeyrVrVrW_remove_dead_remindersJs  z#_WrapNumbers._remove_dead_remindersc Cs||jkr||||S||||j|}i}|D]}||}z ||}Wn tk rp|||<Yq8YnXg}tt|D]f} || } || } || f} | | kr|j|| | 7<|j||| | | |j|| qt |||<q8||j|<|S)zlCache dict and sum numbers which overflow and wrap. Return an updated copy of `input_dict`. ) rrrrrangelenrraddappendtuple) rcrrfrZnew_dictrZ input_tupleZ old_tuplebitsiZ input_value old_valuerrVrVrWrunUs2       z_WrapNumbers.runNc Csh|jX|dkr0|j|j|jn*|j|d|j|d|j|dW5QRXdS)z>Clear the internal cache, optionally only for function 'name'.N)rrrrrpop)rcrfrVrVrWr|s   z_WrapNumbers.cache_clearc Cs.|j|j|j|jfW5QRSQRXdS)z5Return internal cache dicts as a tuple of 3 elements.N)rrrrrrVrVrW cache_infosz_WrapNumbers.cache_info)N) rSrTrUrrrrrrrrVrVrVrWr7s ' rc Cs*tjt||W5QRSQRXdS)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)_wnrr)rrfrVrVrWr;sicCst|dtdS)Nrb) buffering)openFILE_READ_BUFFER_SIZE)fnamerVrVrWr=scCsPt|tttd}z t|_Wn0tk r.Yntk rJ|YnX|S)zaOpen a file in text mode by using the proper FS encoding and en/decoding error handlers. )rencodingerrors)rrr/r0 _CHUNK_SIZErrclose)rZfobjrVrVrWr<s c Csv|tkr.||}|W5QRSQRXnDz*||}|W5QRWSQRXWntk rp|YSXdS)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)_DEFAULTreadr)rfallback_openfrVrVrWr>s   cCst||tdS)z,Same as above but opens file in binary mode.)rr)r>r=)rrrVrVrWr?s%(value).1f%(symbol)scCsd}i}t|ddD]\}}d|dd>||<qt|ddD]2}t|||krFt|||}|tSqF|t|d|dS)zUsed by various scripts. See: https://code.activestate.com/recipes/578019-bytes-to-human-human-to-bytes-converter/?in=user-4178764. >>> bytes2human(10000) '9.8K' >>> bytes2human(100001221) '95.4M' ) BKMGTPEZYrRN r)symbolrg) enumeratereversedabsrlocalsdict)nrvsymbolsprefixrsrrgrVrVrWr@scCs tjdjS)z+Return updated psutil.PROCFS_PATH constant.r^)sysmodulesZ PROCFS_PATHrVrVrVrWget_procfs_pathsr!cCs|jttdS)N)rr)decoder/r0)rrVrVrWr"sr"cCs>t|dr|sdSz |Wntk r8YdSXdS)NisattyFT)hasattrr#filenorfilerVrVrWrDs Fc Csts |Sg}tddddddddd d }d |d <z ||}Wn6tk rrd |dt|}t|d YnX|||r|ddd|d|dS)z*Return an highlighted version of 'string'.Z34Z33Z30Z32Z37Z36Z91Z35Z93) ZbluebrownZdarkgreygreenZgreyZ lightblueredZvioletyellowZ29Ninvalid color z; choose amongst 1z[;mz)rDrrlistrrrrw)rcolorboldattrcolorsrtrVrVrWrC s0    c Cs tst||dntr.tt||||dnddl}d}|jjj}|jjj}t ddddd}||d<z ||}Wn6t k rd |d t | } t | dYnX|r|dkr|d 7}|tjkrd nd } |j|_|| } || |zt||dW5|| |XdS)z$Print a colorized version of string.r&rNrQ)r)r*r(r+r,z; choose between ii)rDprintrrCctypeswindllKernel32 GetStdHandleSetConsoleTextAttributerrr0rrrstderrc_ulongrestype) rr1r2r'r:Z DEFAULT_COLORr=r>r4rtZ handle_idhandlerVrVrWrE%s4    cCsttrpddl}||j\}}}}}t|trPt|trFd|}n d|}td|d|d|t j ddS)z@If PSUTIL_DEBUG env var is set, print a debug message to stderr.rNz ignoring zpsutil-debug [:z]> r&) rinspect getframeinfo currentframef_back isinstancerrr9rr?)rtrDrlineno__lines_indexrVrVrWrBMs    )N)N)r )NF)mrr`enumrrrrrrrrrrr ImportErrorrboolgetenvrobjectr__all__rfrrplatform startswithr r rrr r r rZAIXr%r&r"r'r)r-r!r+r,r#r$r*r(r.rrrrrrrrrrrrIntEnumrPglobalsupdate __members__rXgetfilesystemencodingr/getfilesystemencodeerrorsr0r1rr]rrrrr7r4rr3r6r8r5r9r:rArr2rr;rrrrr=r<r>r?r@r!r"stdoutrDrCrErBrVrVrVrWsn     "                   ! -G   W       (