U !hoR@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddlm Z ddlm Z ddlm Z ddlm Z zddlmZWnek rdZYnXzddlmZWnek rdZYnXeed ZeZd d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[gRZejd\kZejd]kZejd^Zejd_ZeZejd`ZejdaZ ejdbZ!epe pe!Z"ejdcZ#ejddZ$deZ%dfZ&dgZ'dhZ(diZ)djZ*dkZ+dlZ,dmZ-dnZ.doZ/dpZ0dqZ1drZ2dsZ3dtZ4duZ5dvZ6dwZ7dxZ8dyZ9dzZ:d{Z;d|ZGdddej?Z@eABe@jCGdddej?ZDeABeDjCeEZFeGZHe dEddddddgZIe dAddddgZJe d?ddddddgZKe d@ddddgZLe dBddddddddgZMe dFdddddgZNe d=dddddddgZOe dCdddddgZPe dDdddddgZQe d>ddddgZRe ddddgZSe dddddgZTe ddddgZUe dddgZVe d5ddddgZWe d:ddgZXe d;dddgZYe dt}|D],}t||d}|s0|dkr |dkr |||<q |S)Nrqr) collections OrderedDictgetattr)selfZattrsinfororrfrfrg _infodict s   zError._infodictcCsR|d}|r.dddd|D}nd}dddt|dd |fDS) N)rqppidroz({}), cSsg|]\}}|d|qS=rf.0kvrfrfrg sz!Error.__str__.. cSsg|] }|r|qSrfrf)rxrfrfrgrsmsg)rformatjoinitemsrrrZdetailsrfrfrg__str__s z Error.__str__cCs8|d}ddd|D}d|jjd|dS)N)rqrrosecondsrrcSsg|]\}}|d|qSrrfrrfrfrgr!sz"Error.__repr__..zpsutil.())rrr __class__rcrrfrfrg__repr__s zError.__repr__N)rcrdre__doc__rrrrfrfrfrgrs  rc@s&eZdZdZdZdddZddZdS) NoSuchProcesszXException raised when a process with a certain PID doesn't or no longer exists. rNcCs$t|||_||_|pd|_dS)Nzprocess no longer existsr__init__rqrorrrqrorrfrfrgr,s zNoSuchProcess.__init__cCs|j|j|j|jffSNrrqrorrrfrfrg __reduce__2szNoSuchProcess.__reduce__)NNrcrdrerrrrfrfrfrgr%s 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. rNcCs$t||||||_|pd|_dS)Nz"PID still exists but it's a zombie)rrrr)rrqrorrrfrfrgr@szZombieProcess.__init__cCs|j|j|j|j|jffSr)rrqrorrrrfrfrgrEszZombieProcess.__reduce__)NNNrrfrfrfrgr6s rc@s&eZdZdZdZdddZddZdS) AccessDeniedz@Exception raised when permission to perform an action is denied.rNcCs$t|||_||_|pd|_dS)NrrrrfrfrgrNs zAccessDenied.__init__cCs|j|j|j|jffSrrrrfrfrgrTszAccessDenied.__reduce__)NNNrrfrfrfrgrIs rc@s&eZdZdZdZdddZddZdS) TimeoutExpiredzWRaised on Process.wait(timeout) if timeout expires and process is still alive. rNcCs.t|||_||_||_d|d|_dS)Nztimeout after z seconds)rrrrqror)rrrqrorfrfrgr_s  zTimeoutExpired.__init__cCs|j|j|j|jffSr)rrrqrorrfrfrgrfszTimeoutExpired.__reduce__)NNrrfrfrfrgrXs rcCsFzt||d}Wntk r*YdSX|dk r>t||}|SdS)z5Calculate percentage usage of 'used' against 'total'.dgN)floatZeroDivisionErrorround)rlrkZround_retrfrfrgrKos 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) frozensetsortedrKeyError Exception)argskwargskeyrerrcachefunrfrgwrappers zmemoize..wrappercs dS)z Clear cache.N)clearrf)rrfrg cache_clearszmemoize..cache_clear) functoolswrapsr)rrrrfrrgrH{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)rrrrrfrgrs"  z'memoize_when_activated..wrappercSs i|_dS)zsActivate cache. Expects a Process instance. Cache will be stored as a "_cache" instance attribute. N)rprocrfrfrgcache_activatesz.memoize_when_activated..cache_activatecSs"z|`Wntk rYnXdS)zDeactivate and clear cache.N)rrrrfrfrgcache_deactivatesz0memoize_when_activated..cache_deactivate)rrrr)rrrrrfrrgmemoize_when_activatedsrcCsJzt|}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)rstrfrfrgrGscCsBzt|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)rrfrfrgrJsc CsXtjrtdkrdSz*tttj}|dW5QRXWdStk rRYdSXdS)z2Return True if IPv6 is supported on this platform.NF)z::1rT)socketZhas_ipv6rrZbindr)ZsockrfrfrgrL scCsvi}d}t}|d|}||kr"qr|d||}||krh|||}||d|}|r`|}|||<|d}q |S)zCParse a C environ block of environment variables into a dictionary.rrrb)rfindupper)datarposZWINDOWS_Znext_posZ equal_posrrrfrfrgrIs   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)rZ AddressFamily ValueErrorZnumrfrfrgrM5s 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)rZ SocketKindrrrfrfrgrN?s cCs|tjthkr&|rt|}|r&t|}|tjkrJ|tthkrJ||t}nt}t|}t|}|dkrxt ||||||St |||||||SdS)z2Convert a raw connection tuple to a proper ntuple.N) rrrrrgetrrMrNr2r;)rrZfamZtype_rurvrwZ status_maprqrfrfrgrUIscCsxddl}|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) ipaddressrxryrsrrstrZ IPv4NetworkZbroadcast_addressrZ IPv6Network)rrrfrfrgbroadcast_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)Nra)category stacklevel)warningswarnDeprecationWarningr)rrr)r replacementrfrginnersz/deprecated_method..outer..inner)rcrrr)rrr)rrgouterws z deprecated_method..outerrf)rrrfrrgrFrs 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_keysrrfrfrgrs z_WrapNumbers.__init__cCsX||jkst||jkst||jks*t||j|<tt|j|<tt|j|<dSr)rAssertionErrorrrr defaultdictintset)r input_dictrorfrfrg _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)rrkeysrr)rrroold_dictZ gone_keysZgone_keyremkeyrfrfrg_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`. ) rrrrrangelenrraddappendtuple) rrrorZnew_dictrZ input_tupleZ old_tuplebitsiZ input_value old_valuerrfrfrgruns2       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)rrorfrfrgrs   z_WrapNumbers.cache_clearc Cs.|j|j|j|jfW5QRSQRXdS)z5Return internal cache dicts as a tuple of 3 elements.N)rrrrrrfrfrg cache_infosz_WrapNumbers.cache_info)N) rcrdrerrrrr rr rfrfrfrgrs ' 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 )rrorfrfrgrOsicCst|dtdS)Nrb) buffering)openFILE_READ_BUFFER_SIZE)fnamerfrfrgrQscCsPt|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)rrr0r1 _CHUNK_SIZErrclose)rZfobjrfrfrgrPs 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_openfrfrfrgrRs   cCst||tdS)z,Same as above but opens file in binary mode.)rr)rRrQ)rrrfrfrgrS*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' ) BKMGTPEZYrbN r)symbolr) enumeratereversedabsrlocalsdict)nrZsymbolsprefixr sr(rrfrfrgrT/scCs tjdjS)z+Return updated psutil.PROCFS_PATH constant.r)sysmodulesZ PROCFS_PATHrfrfrfrgget_procfs_pathBsr3cCs|jttdS)N)rr)decoder0r1)r0rfrfrgr4Gsr4cCs`tjdkrdSz2ddl}|s$t||ddks>tWntk rVYdSXdSdS)Nr[TrcolorsF)rrocursesisattyrZ setuptermZtigetnumr)filer6rfrfrgrXPs  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)rXr-rlistrrrr)r0colorboldattrr5rrfrfrgrW`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.r8rNra)r:r;r9r<r=z; choose between ii)rXprintrrWctypesZwindllZKernel32 GetStdHandleSetConsoleTextAttributer-rrArrr1stderrZc_ulongZrestype) r0rBrCr8rKZ DEFAULT_COLORrLrMr5rZ handle_idZhandlerfrfrgrY|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]> rE) rinspectZ getframeinfoZ currentframef_back isinstancerrrJr1rN)rrPrlineno_Z_linesZ_indexrfrfrgrVs    )N)N)r)NF)rrenumrrrrr1rrrrrrr ImportErrorrboolgetenvrobjectr__all__rorrplatform startswithr rrr r r r rZAIXr&r'r#r(r*r.r"r,r-r$r%r+r)r/rrrrrrrrrrrrIntEnumr`globalsupdate __members__rhgetfilesystemencodingr0getfilesystemencodeerrorsr1rCr?r=r>r@rDr;rArBr<r{rrrr3r8r9r:r5r6r7r4r2rrErrrrrrrKrHrrGrJrLrIrMrNrUrrFrrOrrr rrQrPrRrSrTr3r4stdoutrXrWrYrVrfrfrfrgs      &                          ! -G   W       (