U +-eqR@sdZddlmZddlmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZddl mZddl mZddl mZzdd l mZWnek rdZYnXzdd l mZWnek rdZYnXe jdd kZerddlZndZeed ZeZd ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^gRZejd_kZejd`kZ e j!"daZ#e j!"dbZ$e$Z%e j!"dcZ&e j!"ddZ'e j!"deZ(e&p0e'p0e(Z)e j!"dfZ*e j!"dgZ+dhZ,diZ-djZ.dkZ/dlZ0dmZ1dnZ2doZ3dpZ4dqZ5drZ6dsZ7dtZ8duZ9dvZ:dwZ;dxZd{Z?d|Z@d}ZAd~ZBdZCdZDdZEedkrdZFdZGdZHn GdddejIZJeKLeJjMedkrdZNdZOn GdddejIZPeKLePjMe QZRes2dZSn2z e TZSWn$eUk rberZdndZSYnXedHddddddgZVedDddddgZWedBddddddgZXedCddddddgZYedEddddddddgZZedIdddddgZ[ed@dddddddgZ\edFdddddgZ]edGdddddgZ^edAddddgZ_eddddgZ`edddddgZaeddddgZbedddgZced8ddddgZded=ddgZeed>dddgZfed?dddgZged:dddgZhed;ddddgZied-sz!Error.__str__.. cSsg|] }|r|qSrtrt)rxrtrtrur0smsg)rjoinitemsrrrdetailsrtrtru__str__(s z Error.__str__cCs2|d}ddd|D}d|jj|fS)N)rrrsecondsrrcSsg|]\}}d||fqSrrtrrtrtrur5sz"Error.__repr__..z psutil.%s(%s))rrr __class__rqrrtrtru__repr__2s zError.__repr__N)rqrrrs__doc__rrrrtrtrtrurs   rc@seZdZdZdZdddZdS) NoSuchProcesszXException raised when a process with a certain PID doesn't or no longer exists. rNcCs$t|||_||_|pd|_dS)Nzprocess no longer existsr__init__rrrrrrrrtrtrur@s zNoSuchProcess.__init__)NNrqrrrsrrrtrtrtrur9src@seZdZdZdZdddZdS) 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. rNcCs$t||||||_|pd|_dS)Nz"PID still exists but it's a zombie)rrrr)rrrrrrtrtrurQszZombieProcess.__init__)NNNrrtrtrtrurGsrc@seZdZdZdZdddZdS) AccessDeniedz@Exception raised when permission to perform an action is denied.rNcCs$t|||_||_|pd|_dS)Nrrrrtrtrur\s zAccessDenied.__init__)NNNrrtrtrtrurWsrc@seZdZdZdZdddZdS)TimeoutExpiredzWRaised on Process.wait(timeout) if timeout expires and process is still alive. rNcCs*t|||_||_||_d||_dS)Nztimeout after %s seconds)rrrrrr)rrrrrtrtrurjs  zTimeoutExpired.__init__)NNrrtrtrtrurcsrexeczvdef raise_from(value, from_value): try: raise value from from_value finally: value = None cCs|dSNrt)r from_valuertrtru raise_fromsrcCsFzt||d}Wntk r*YdSX|dk r>t||}|SdS)z5Calculate percentage usage of 'used' against 'total'.dgN)floatZeroDivisionErrorround)r~r}Zround_retrtrtrurNs 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 rzz||}|<Wn,tk rp}zt|dW5d}~XYnX|YSXdSr) frozensetsortedrKeyError Exceptionr)argskwargskeyrerrcachefunrtruwrappers zmemoize..wrappercs dS)z Clear cache.N)clearrt)rrtru cache_clearszmemoize..cache_clear) functoolswrapsr)rrrrtrrurKs  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 r^z|WYStk rX}zt|dW5d}~XYnXYnrtk rz |}Wn,tk r}zt|dW5d}~XYnXz||j<Wntk rYnXYnX|Sr)_cacheAttributeErrorrrr)rrrrrtrurs"   z'memoize_when_activated..wrappercSs i|_dS)zsActivate cache. Expects a Process instance. Cache will be stored as a "_cache" instance attribute. N)rprocrtrtrucache_activatesz.memoize_when_activated..cache_activatecSs"z|`Wntk rYnXdS)zDeactivate and clear cache.N)rrrrtrtrucache_deactivatesz0memoize_when_activated..cache_deactivate)rrrr)rrrrrtrrumemoize_when_activatedsrc Cs\zt|}Wn<tk rJ}z|jtjtjfkr4WYdSd}~XYnXt|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)osstatOSErrorerrnoEPERMEACCESS_ISREGst_mode)rstrrtrtrurJsc CsTzt|Wn<tk rJ}z|jtjtjfkr4WYdSd}~XYnXdSdS)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)rrrrrr)rrrtrtrurMsc CsdtjrtdkrdSz4tttj}t||dW5QRXWdStjk r^YdSXdS)z2Return True if IPv6 is supported on this platform.NF)z::1rT)sockethas_ipv6rr contextlibclosingbinderror)sockrtrtrurO&s cCsvi}d}t}|d|}||kr"qr|d||}||krh|||}||d|}|r`|}|||<|d}q |S)zCParse a C environ block of environment variables into a dictionary.r=ro)rfindupper)datarposZWINDOWS_Znext_posZ equal_posrrrtrtrurL4s   cCs6tdkr |Sz t|WStk r0|YSXdS)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 ValueErrornumrtrtrurPPs  cCs6tdkr |Sz t|WStk r0|YSXdS)zConvert a numeric socket type value to an IntEnum member. If it's not a known member, return the numeric value itself. N)rr SocketKindr r rtrtrurQ]s  cCs|tjtfkr&|rt|}|r&t|}|tjkrJ|ttfkrJ||t}nt}t|}t|}|dkrxt ||||||St |||||||SdS)z2Convert a raw connection tuple to a proper ntuple.N) rrrrrgetrrPrQr5r>)rfamtype_rrrZ status_maprrtrtrurXjscsfdd}|S)zA decorator which can be used to mark a method as deprecated 'replcement' is the method name which will be called instead. cs:d|jf|jdkr|_t|fdd}|S)Nz8%s() is deprecated and will be removed; use %s() insteadcs tjtddt|||S)Nrn)category stacklevel)warningswarnDeprecationWarningr)rrr)r replacementrtruinnersz/deprecated_method..outer..inner)rqrrr)rrr)rruouters z deprecated_method..outerrt)rrrtrrurI}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_keysrrtrtrurs z_WrapNumbers.__init__cCsX||jkst||jkst||jks*t||j|<tt|j|<tt|j|<dSr)rAssertionErrorrrr defaultdictintset)r input_dictrrtrtru _add_dicts  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)rr$keysrr)rr%rold_dictZ gone_keysZgone_keyremkeyrtrtru_remove_dead_reminderss  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`. ) rr&r*rrangelenrraddappendtuple) rr%rr(Znew_dictrZ input_tupleZ old_tuplebitsiZ input_value old_valuer)rtrtruruns2       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)rrrtrtrurs   z_WrapNumbers.cache_clearc Cs.|j|j|j|jfW5QRSQRXdS)z5Return internal cache dicts as a tuple of 3 elements.N)rrrrr rtrtru cache_infosz_WrapNumbers.cache_info)N) rqrrrsrrr&r*r3rr5rtrtrtrurs ' 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)_wnrr3)r%rrtrtrurRsicCst|dtdS)Nrb buffering)openFILE_READ_BUFFER_SIZE)fnamertrtrurTscCs`tst|tdSt|tttd}z t|_Wn0tk r>Yntk rZ|YnX|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'). r8)r9encodingerrors) PY3r:r;r3r4 _CHUNK_SIZErrclose)r<ZfobjrtrtrurSs  c Csz|tkr.||}|W5QRSQRXnHz*||}|W5QRWSQRXWnttfk rt|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)_DEFAULTreadIOErrorr)r<fallback_openfrtrtrurU!s   cCst||tdS)z,Same as above but opens file in binary mode.)rErF)rUrT)r<rErtrtrurV2s%(value).1f%(symbol)scCsd}i}t|ddD]\}}d|dd>||<qt|ddD]2}t|||krFt|||}|tSqF|t|d|dS)zUsed by various scripts. See: http://goo.gl/zeJZl. >>> bytes2human(10000) '9.8K' >>> bytes2human(100001221) '95.4M' ) BKMGTPEZYroN r)symbolr) enumeratereversedabsrlocalsdict)nformatsymbolsprefixr1srSrrtrtrurW7scCs tjdjS)z+Return updated psutil.PROCFS_PATH constant.r)sysmodulesZ PROCFS_PATHrtrtrtruget_procfs_pathJsr`cCs|jttdS)N)r=r>)decoder3r4r]rtrtruraPsracCs|SrrtrbrtrtruraSscCs`tjdkrdSz2ddl}|s$t||ddks>tWntk rVYdSXdSdS)Nr^TrcolorsF)rrcursesisattyr!Z setuptermZtigetnumr)filerdrtrtrur[\s  Fc Csts |Sg}tddddddddd d }d |d <z ||}Wn(tk rdtd t|YnX|||r~|ddd||fS)z*Return an highlighted version of 'string'.Z32Z91Z33Z93Z34Z35Z36Z37Z30) greenredbrownyellowbluevioletZ lightblueZgreyZdarkgreyZ29Nz#invalid color %r; choose between %s1z [%sm%s;)r[rXrr listr'r.r)r]colorboldattrrcrtrtrurZks(     c Cstst||dntr.tt||||dnddl}d}|jjj}|jjj}t ddddd}||d<z ||}Wn,t k rt d |t | fYnX|r|dkr|d 7}|tjkrd nd } |j|_|| } || |zt||dW5|| |XdS) z$Print a colorized version of string.rfrNrn)rgrhrirjz#invalid color %r; choose between %rii)r[printrrZctypeswindllZKernel32 GetStdHandleSetConsoleTextAttributerXrr ror'r^stderrZc_ulongrestype) r]rprqrfryZ DEFAULT_COLORr{r|rcZ handle_idhandlertrtrur\~s6     cCsntrjddl}||j\}}}}}t|trRt|ttt frJd|}nd|}t d|||ft j ddS)z@If PSUTIL_DEBUG env var is set, print a debug message to stderr.rNz ignoring %sz ignoring %rzpsutil-debug [%s:%s]> %srs) r inspect getframeinfo currentframef_back isinstancerrrDEnvironmentErrorrxr^r})rrr<lineno_linesindexrtrtrurYs  )N)N)rH)NF)r __future__rrrrrrrrrr^rrrrrrr ImportErrorr version_infor?rboolgetenvr objectrB__all__rrrplatform startswithrrrr rrr rZAIXr)r*r&r+r-r1r%r/r0r'r(r.r,r2rr rrrr!rrrrrrr"r#r$IntEnumrpglobalsupdate __members__ryrzrxgetfilesystemencodingr3r4getfilesystemencodeerrorsrrFrBr@rArCrGr>rDrEr?rrrrr6r;r<r=r8r9r:r7r5rrHrrrrrrr __builtins__rXexec_rrrNrKrrJrMrOrLrPrQrXrIrrRr6rr5r;rTrSrUrVrWr`rastdoutr[rZr\rYrtrtrtrusL         %                                "      ,F   W    #