U (cDnR@sbdZddlmZddlmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZddl mZddl mZddl mZzdd l mZWnek rdZYnXzdd l mZWnek rdZYnXe jd krddlZndZe jdd kZeed dZeZddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_gRZejd`kZejdakZ e j!"dbZ#e j!"dcZ$e$Z%e j!"ddZ&e j!"deZ'e j!"dfZ(e&p8e'p8e(Z)e j!"dgZ*e j!"dhZ+diZ,djZ-dkZ.dlZ/dmZ0dnZ1doZ2dpZ3dqZ4drZ5dsZ6dtZ7duZ8dvZ9dwZ:dxZ;dyZd|Z?d}Z@d~ZAdZBdZCdZDdZEedkrdZFdZGdZHn GdddejIZJeKLeJjMedkrdZNdZOn GdddejIZPeKLePjMe QZRes:dZSn2z e TZSWn$eUk rjerbdndZSYnXedIddddddgZVedEddddgZWedCddddddgZXedDddddddgZYedFddddddddgZZedJdddddgZ[edAdddddddgZ\edGdddddgZ]edHdddddgZ^edBddddgZ_eddddgZ`edddddgZaeddddgZbedddgZced9ddddgZded>ddgZeed?dddgZfed@dddgZged;dddgZhed+sz!Error.__str__.. cSsg|] }|r|qSrnrn)rxrnrnror.smsg)rjoinitemsrrrdetailsrnrnro__str__&s z Error.__str__cCs2|d}ddd|D}d|jj|fS)N)rrrsecondsrrcSsg|]\}}d||fqSrrnrrnrnror3sz"Error.__repr__..z psutil.%s(%s))rrr __class__rkrrnrnro__repr__0s zError.__repr__N)rkrlrm__doc__rrrrnrnrnrors   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__rrrrrrrrnrnror=s zNoSuchProcess.__init__)NNrkrlrmrrrnrnrnror7src@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)rrrrrrnrnrorMszZombieProcess.__init__)NNNrrnrnrnrorDsrc@seZdZdZdZdddZdS) AccessDeniedz@Exception raised when permission to perform an action is denied.rNcCs$t|||_||_|pd|_dS)NrrrrnrnrorWs zAccessDenied.__init__)NNNrrnrnrnrorSsrc@seZdZdZdZdddZdS)TimeoutExpiredzWRaised on Process.wait(timeout) if timeout expires and process is still alive. rNcCs*t|||_||_||_d||_dS)Nztimeout after %s seconds)rrrrrr)rrrrrnrnrords  zTimeoutExpired.__init__)NNrrnrnrnror^srcCsFzt||d}Wntk r*YdSX|dk r>t||}|SdS)z5Calculate percentage usage of 'used' against 'total'.dgN)floatZeroDivisionErrorround)rxrwZround_retrnrnrorOqs 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() >>> csN|tt|f}z |WStk rH||}|<|YSXdSN) frozensetsortedrKeyError)argskwargskeyrcachefunrnrowrappers  zmemoize..wrappercs dS)z Clear cache.N)clearrn)rrnro cache_clearszmemoize..cache_clear) functoolswrapsr)rrrrnrrorL}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 snz|j}WnZtk r*|YStk rh|}z||j<Wntk rbYnXYnX|Sr)_cacheAttributeErrorr)rrrrnrors  z'memoize_when_activated..wrappercSs i|_dS)zjActivate cache. Expects a Process instance. Cache will be stored as a "_cache" instance attribute.N)rprocrnrnrocache_activatesz.memoize_when_activated..cache_activatecSs"z|`Wntk rYnXdS)zDeactivate and clear cache.N)rrrrnrnrocache_deactivatesz0memoize_when_activated..cache_deactivate)rrrr)rrrrrnrromemoize_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)rsterrrnrnrorKsc 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)rrrnrnrorNsc CsdtjrtdkrdSz4tttj}t||dW5QRXWdStjk r^YdSXdS)z2Return True if IPv6 is supported on this platform.NF)z::1rT)sockethas_ipv6rr contextlibclosingbinderror)sockrnrnrorPs 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=ri)rfindupper)datarposZWINDOWS_Znext_posZ equal_posrrrnrnrorMs   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 ValueErrornumrnrnrorQ$s  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 SocketKindrrrnrnrorR1s  cCs|tjtfkr&|rt|}|r&t|}|tjkrJ|ttfkrJ||t}nt}t|}t|}|dkrxt ||||||St |||||||SdS)z2Convert a raw connection tuple to a proper ntuple.N) rrrrrgetrrQrRr6r?)rZfamtype_rrrZ status_maprrnrnrorY>scsfdd}|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)Nrh)category stacklevel)warningswarnDeprecationWarningr)rrr)r replacementrnroinner[sz/deprecated_method..outer..inner)rkrrr)rr r )rroouterUs z deprecated_method..outerrn)r rrnr rorJQs 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_keysrrnrnrorhs z_WrapNumbers.__init__cCsX||jkst||jkst||jks*t||j|<tt|j|<tt|j|<dSr)rAssertionErrorrrr defaultdictintset)r input_dictrrnrnro _add_dictns  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)rrrold_dictZ gone_keysZgone_keyremkeyrnrnro_remove_dead_remindersvs  z#_WrapNumbers._remove_dead_remindersc Cs ||jkr||||S||||j|}i}|D]}||}z ||}Wn tk rt|||<YqClear the internal cache, optionally only for function 'name'.N)rrrrrpop)rrrnrnrors   z_WrapNumbers.cache_clearc Cs.|j|j|j|jfW5QRSQRXdS)z5Return internal cache dicts as a tuple of 3 elements.N)rrrrrrnrnro cache_infosz_WrapNumbers.cache_info)N) rkrlrmrrrrr(rr*rnrnrnrorcs ' 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()rrrnrnrorSsicCst|dtdS)Nrb buffering)openFILE_READ_BUFFER_SIZE)fnamernrnrorUscCsdtst|dtdSt|dtttd}z t|_Wn0tk rBYntk r^|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'). rtr-)r.encodingerrors) PY3r/r0r4r5 _CHUNK_SIZEr Exceptionclose)r1ZfobjrnrnrorTs 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)r1fallback_openfrnrnrorVs   cCst||tdS)z,Same as above but opens file in binary mode.)r<r=)rVrU)r1r<rnrnrorWs%(value).1f%(symbol)scCsd}i}t|ddD]\}}d|dd>||<qt|ddD].}|||krFt|||}|tSqF|t|d|dS)zUsed by various scripts. See: http://goo.gl/zeJZl >>> bytes2human(10000) '9.8K' >>> bytes2human(100001221) '95.4M' ) BKMGTPEZYriN r)symbolr) enumeratereversedrlocalsdict)nformatsymbolsprefixr&srJrrnrnrorX s  cCs tjdjS)z+Return updated psutil.PROCFS_PATH constant.r)sysmodulesZ PROCFS_PATHrnrnrnroget_procfs_pathsrVcCs|jttdS)N)r3r4)decoder4r5rSrnrnrorW%srWcCs|SrrnrXrnrnrorW(scCs`tjdkrdSz2ddl}|s$t||ddks>tWntk rVYdSXdSdS)Nr_TrcolorsF)rrcursesisattyrZ setuptermZtigetnumr7)filerZrnrnror\1s  Fc Csts |Sg}tddddddddd d }d |d <z ||}Wn(tk rdtd t|YnX|||r~|ddd||fS)z*Return an highlighted version of 'string'.Z32Z91Z33Z93Z34Z35Z36Z37Z30) greenredbrownyellowZblueZvioletZ lightblueZgreyZdarkgreyZ29Nz#invalid color %r; choose between %s1z [%sm%s;)r\rNrrlistrr#r)rScolorboldattrrYrnrnror[@s(     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.r\rNrhr )r]r^r_r`z#invalid color %r; choose between %rii)r\printrr[ctypeswindllKernel32 GetStdHandleSetConsoleTextAttributerNrrrcrrTstderrc_ulongrestype) rSrdrer\rlZ DEFAULT_COLORrorprYZ handle_idhandlernrnror]Ss6     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]> %srg) r inspect getframeinfo currentframef_back isinstancer7rr;EnvironmentErrorrkrTrq)rrur1lineno func_namelinesindexrnrnrorZvs  )N)N)r?)NF)r __future__rrrrrrrrrrTrrrrrrr ImportErrorr version_inforr5boolgetenvr objectr9__all__rrrplatform startswithrrrr rrrrZAIXr*r+r'r,r.r2r&r0r1r(r)r/r-r3rr!r rrr"rrrrrrr#r$r%IntEnumrjglobalsupdate __members__rsrtrrgetfilesystemencodingr4r5getfilesystemencodeerrorsrrGrCrArBrDrHr?rErFr@rrrrr7r<r=r>r9r:r;r8r6rrIr7rrrrrrOrLrrKrNrPrMrQrRrYrJrrSr+rr*r0rUrTrVrWrXrVrWstdoutr\r[r]rZrnrnrnros@          %                                !    ?   W      #