B *dqR@sdZddlmZddlmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZddl mZddl mZddl mZydd l mZWnek rdZYnXydd l mZWnek rdZYnXe jd krddlZndZe jdd kZeed ZeZddddddddddddddddddd 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&p6e'p6e(Z)e j!"dgZ*e j!"dhZ+diZ,djZ-dkZ.dlZ/dmZ0dnZ1doZ2dpZ3dqZ4drZ5dsZ6dtZ7duZ8dvZ9dwZ:dxZ;dyZd|Z?d}Z@d~ZAdZBdZCdZDdZEedkrdZFdZGdZHn GdddejIZJeKLeJjMedkrdZNdZOn GdddejIZPeKLePjMe QZRes8dZSn2y e TZSWn$eUk rher`dndZSYnXedIddddddgZVedEddddgZWedCddddddgZXedDddddddgZYedFddddddddgZZedJdddddgZ[edAdddddddgZ\edGdddddgZ]edHdddddgZ^edBddddgZ_eddddgZ`edddddgZaeddddgZbedddgZced9ddddgZded>ddgZeed?dddgZfed@dddgZged;dddgZhed|D]6}t||d}|r,|||<q|dkr|dkr|||<qW|S)Nrr) collections OrderedDictgetattr)selfattrsinforrrrrrrs _infodicts    zError._infodictcCsP|d}|r,dddd|D}nd}dddt|dd |fDS) N)rppidrz(%s)z, cSsg|]\}}d||fqS)z%s=%rrr).0kvrrrrrs +sz!Error.__str__.. cSsg|] }|r|qSrrrr)rxrrrrrsr.smsg)rjoinitemsr)rrdetailsrrrrrs__str__&s  z Error.__str__cCs2|d}ddd|D}d|jj|fS)N)rrrsecondsrz, cSsg|]\}}d||fqS)z%s=%rrr)rrrrrrrrsr3sz"Error.__repr__..z psutil.%s(%s))rrr __class__ro)rrrrrrrrs__repr__0s zError.__repr__N)rorprq__doc__rrrrrrrrrrsrs   rc@seZdZdZdZdddZdS) NoSuchProcesszXException raised when a process with a certain PID doesn't or no longer exists. rNcCs$t|||_||_|pd|_dS)Nzprocess no longer exists)r__init__rrr)rrrrrrrrrsr=s zNoSuchProcess.__init__)NN)rorprqrrrrrrrrrsr7src@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)rrrrrrrrrrsrMszZombieProcess.__init__)NNN)rorprqrrrrrrrrrsrDsrc@seZdZdZdZdddZdS) AccessDeniedz@Exception raised when permission to perform an action is denied.rNcCs$t|||_||_|pd|_dS)Nr)rrrrr)rrrrrrrrrsrWs zAccessDenied.__init__)NNN)rorprqrrrrrrrrrsrSsrc@seZdZdZdZdddZdS)TimeoutExpiredzWRaised on Process.wait(timeout) if timeout expires and process is still alive. rNcCs*t|||_||_||_d||_dS)Nztimeout after %s seconds)rrrrrr)rrrrrrrrrsrds  zTimeoutExpired.__init__)NN)rorprqrrrrrrrrrsr^srexeczvdef raise_from(value, from_value): try: raise value from from_value finally: value = None cCs|dS)Nrr)r from_valuerrrrrs raise_fromsrcCsDyt||d}Wntk r(dSX|dk r>> @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 sz|tt|f}y|Stk rty||}|<Wn,tk rn}zt|dWdd}~XYnX|SXdS)N) frozensetsortedrKeyError Exceptionr)argskwargskeyrerr)cachefunrrrswrapperszmemoize..wrappercs dS)z Clear cache.N)clearrr)rrrrs cache_clearszmemoize..cache_clear) functoolswrapsr)rrrrr)rrrsrLs  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 sy|j}Wntk rXy|Stk rR}zt|dWdd}~XYnXYnrtk ry |}Wn,tk r}zt|dWdd}~XYnXy||j<Wntk rYnXYnX|S)N)_cacheAttributeErrorrrr)rrr)rrrrsrs"   z'memoize_when_activated..wrappercSs i|_dS)zjActivate cache. Expects a Process instance. Cache will be stored as a "_cache" instance attribute.N)r)procrrrrrscache_activatesz.memoize_when_activated..cache_activatecSs"y|`Wntk rYnXdS)zDeactivate and clear cache.N)rr)rrrrrrscache_deactivatesz0memoize_when_activated..cache_deactivate)rrrr)rrrrrr)rrsmemoize_when_activateds rc CsVyt|}Wn6tk rD}z|jtjtjfkr4dSd}~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)rstrrrrrrsrKsc CsNyt|Wn6tk rD}z|jtjtjfkr4dSd}~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)rrrrrrrsrNsc Cs`tjrtdkrdSy2tttj}t||dWdQRXdStjk rZdSXdS)z2Return True if IPv6 is supported on this platform.NF)z::1rT)sockethas_ipv6rr contextlibclosingbinderror)sockrrrrrsrPs cCszi}d}t}xh|d|}||kr$P|d||}||krj|||}||d|}|rb|}|||<|d}qW|S)zCParse a C environ block of environment variables into a dictionary.r=rm)rfindupper)datarposZWINDOWS_Znext_posZ equal_posrrrrrrrsrM-s    cCs0tdkr |Sy t|Stk r*|SXdS)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)numrrrrrsrQIs  cCs0tdkr |Sy t|Stk r*|SXdS)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)rrrrrrsrRVs  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?)rfamtype_rrrZ status_maprrrrrrsrYcscsfdd}|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)Nrl)category stacklevel)warningswarnDeprecationWarningr)rrr)r replacementrrrsinnersz/deprecated_method..outer..inner)rorrr)rr )r)rrsouterzs   z deprecated_method..outerrr)rr rr)rrsrJvs 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|_dS)N) threadingLocklockr reminders reminder_keys)rrrrrrsrs z_WrapNumbers.__init__cCsX||jkst||jkst||jks*t||j|<tt|j|<tt|j|<dS)N)rAssertionErrorrrr defaultdictintset)r input_dictrrrrrrs _add_dicts  z_WrapNumbers._add_dictcCsd|j|}t|t|}x<|D]4}x"|j||D]}|j||=qYnXg}xvtt|D]f} || } || } || f} | | kr|j|| | 7<|j|| | | | |j|| qWt |||<q>W||j|<|S)zkCache dict and sum numbers which overflow and wrap. Return an updated copy of `input_dict` ) rrrrrrangelenrraddappendtuple) rrrrZnew_dictrZ input_tupleZ old_tuplebitsiZ input_value old_valuerrrrrrsruns2      z_WrapNumbers.runNc Csh|jX|dkr0|j|j|jn*|j|d|j|d|j|dWdQRXdS)z>Clear the internal cache, optionally only for function 'name'.N)rrrrrpop)rrrrrrrsrs   z_WrapNumbers.cache_clearc Cs"|j|j|j|jfSQRXdS)z5Return internal cache dicts as a tuple of 3 elements.N)rrrr)rrrrrrs cache_infosz_WrapNumbers.cache_info)N) rorprqrrrrr#rr%rrrrrrrsr s ' r c Cstjt||SQRXdS)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#)rrrrrrrsrSsicCst|dtdS)Nrb) buffering)openFILE_READ_BUFFER_SIZE)fnamerrrrrsrUscCsdtst|dtdSt|dtttd}y 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'). rt)r()r(encodingerrors) PY3r)r*r4r5 _CHUNK_SIZErrclose)r+ZfobjrrrrrsrTs  c Cs\|tkr"|| }|SQRXn6y|| }|SQRXWnttfk rV|SXdS)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_openfrrrrrsrVs  cCst||tdS)z,Same as above but opens file in binary mode.)r5r6)rVrU)r+r5rrrrrsrW+s%(value).1f%(symbol)scCsd}i}x0t|ddD]\}}d|dd>||<qWx>t|ddD]*}|||krLt|||}|tSqLW|t|d|dS)zUsed by various scripts. See: http://goo.gl/zeJZl >>> bytes2human(10000) '9.8K' >>> bytes2human(100001221) '95.4M' ) BKMGTPEZYrmN r)symbolr) enumeratereversedrlocalsdict)nformatsymbolsprefixr!srCrrrrrrsrX0s  cCs tjdjS)z+Return updated psutil.PROCFS_PATH constant.r)sysmodulesZ PROCFS_PATHrrrrrrrsget_procfs_pathDsrOcCs|jttdS)N)r-r.)decoder4r5)rLrrrrrsrPJsrPcCs|S)Nrr)rLrrrrrsrPMscCs^tjdkrdSy2ddl}|s$t||ddks>tWntk rTdSXdSdS)Nr_TrcolorsF)rrcursesisattyrZ setuptermZtigetnumr)filerRrrrrrsr\Vs  Fc Csts |Sg}tddddddddd d }d |d <y ||}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\rGrrlistrrr)rLcolorboldattrrQrrrrrsr[es    c Cstst||dntr.tt||||dnddl}d}|jjj}|jjj}t ddddd}||d<y ||}Wn,t k rt d |t | fYnX|r|dkr|d 7}|tjkrd nd } |j|_|| } || |zt||dWd|| |XdS) z$Print a colorized version of string.)rTrNrlr )rUrVrWrXz#invalid color %r; choose between %rii)r\printrr[ctypeswindllZKernel32 GetStdHandleSetConsoleTextAttributerGrrr]rrMstderrZc_ulongrestype) rLr^r_rTreZ DEFAULT_COLORrgrhrQZ handle_idhandlerrrrrsr]xs0     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]> %s)rT) r inspect getframeinfo currentframef_back isinstancerrr4EnvironmentErrorrdrMri)rrlr+lineno_linesindexrrrrrsrZs  )N)N)r8)NF)r __future__rrrrrrrrrrMr rrrrrr ImportErrorr version_inforr/boolgetenvr objectr2__all__rrrplatform startswithrrrr rrrrZAIXr*r+r'r,r.r2r&r0r1r(r)r/r-r3rr!r rrr"rrrrrrr#r$r%IntEnumrnglobalsupdate __members__rwrxrvgetfilesystemencodingr4r5getfilesystemencodeerrorsrrGrCrArBrDrHr?rErFr@rrrrr7r<r=r>r9r:r;r8r6rrIrrrrrrrp __builtins__rGexec_rrrOrLrrKrNrPrMrQrRrYrJr rSr&rr%r*rUrTrVrWrXrOrPstdoutr\r[r]rZrrrrrrrss                                         !      ,E   W   "