U ;goR@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)Nrr) collections OrderedDictgetattr)selfattrsinfor}rrprprq _infodict s   zError._infodictcCsR|d}|r.dddd|D}nd}dddt|dd |fDS) N)rppidr}z({}), cSsg|]\}}|d|qS=rp.0kvrprprq sz!Error.__str__.. cSsg|] }|r|qSrprp)rxrprprqrsmsg)rformatjoinitemsrrrdetailsrprprq__str__s z Error.__str__cCs8|d}ddd|D}d|jjd|dS)N)rrr}secondsrrcSsg|]\}}|d|qSrrprrprprqr"sz"Error.__repr__..zpsutil.())rrr __class__rmrrprprq__repr__s zError.__repr__N)rmrnro__doc__rrrrprprprqrs  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__rr}rrrr}rrprprqr-s zNoSuchProcess.__init__cCs|j|j|j|jffSNrrr}rrrprprq __reduce__3szNoSuchProcess.__reduce__)NNrmrnrorrrrprprprqr&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)rrr}rrrprprqrAszZombieProcess.__init__cCs|j|j|j|j|jffSr)rrr}rrrrprprqrFszZombieProcess.__reduce__)NNNrrprprprqr7s rc@s&eZdZdZdZdddZddZdS) AccessDeniedz@Exception raised when permission to perform an action is denied.rNcCs$t|||_||_|pd|_dS)NrrrrprprqrOs zAccessDenied.__init__cCs|j|j|j|jffSrrrrprprqrUszAccessDenied.__reduce__)NNNrrprprprqrJs 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)rrrrr}r)rrrr}rprprqr`s  zTimeoutExpired.__init__cCs|j|j|j|jffSr)rrrr}rrprprqrgszTimeoutExpired.__reduce__)NNrrprprprqrYs rcCsFzt||d}Wntk r*YdSX|dk r>t||}|SdS)z5Calculate percentage usage of 'used' against 'total'.dgN)floatZeroDivisionErrorround)rvruZround_retrprprqrKps 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)argskwargskeyrerrcachefunrprqwrappers zmemoize..wrappercs dS)z Clear cache.N)clearrp)rrprq cache_clearszmemoize..cache_clear) functoolswrapsr)rrrrprrqrH|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)rrrrrprqrs"  z'memoize_when_activated..wrappercSs i|_dS)zsActivate cache. Expects a Process instance. Cache will be stored as a "_cache" instance attribute. N)rprocrprprqcache_activatesz.memoize_when_activated..cache_activatecSs"z|`Wntk rYnXdS)zDeactivate and clear cache.N)rrrrprprqcache_deactivatesz0memoize_when_activated..cache_deactivate)rrrr)rrrrrprrqmemoize_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)rstrprprqrGscCsBzt|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)rrprprqrJsc CsXtjrtdkrdSz*tttj}|dW5QRXWdStk rRYdSXdS)z2Return True if IPv6 is supported on this platform.NF)z::1rT)sockethas_ipv6rrbindr)sockrprprqrLscCsvi}d}t}|d|}||kr"qr|d||}||krh|||}||d|}|r`|}|||<|d}q |S)zCParse a C environ block of environment variables into a dictionary.rrrl)rfindupper)datarposZWINDOWS_Znext_posZ equal_posrrrprprqrIs   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 ValueErrornumrprprqrM7s 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 SocketKindrrrprprqrNAs 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;)rZfamtype_rrrZ status_maprrprprqrUKscCsxddl}|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) ipaddressrrrrrstr IPv4Networkbroadcast_addressr IPv6Network)rrrprprqbroadcast_addr^s"   r csfdd}|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)Nrk)category stacklevel)warningswarnDeprecationWarningr)rrr)r replacementrprqinnersz/deprecated_method..outer..inner)rmrrr)rrr)rrqouterys z deprecated_method..outerrp)rrrprrqrFts 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_keysrrprprqrs z_WrapNumbers.__init__cCsX||jkst||jkst||jks*t||j|<tt|j|<tt|j|<dSr)rAssertionErrorrrr defaultdictintset)r input_dictr}rprprq _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)rr r}old_dictZ gone_keysZgone_keyremkeyrprprq_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 r}r#Znew_dictrZ input_tupleZ old_tuplebitsiZ input_value old_valuer$rprprqruns2       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)rr}rprprqrs   z_WrapNumbers.cache_clearc Cs.|j|j|j|jfW5QRSQRXdS)z5Return internal cache dicts as a tuple of 3 elements.N)rrrrrrprprq cache_infosz_WrapNumbers.cache_info)N) rmrnrorrr!r%r.rr0rprprprqrs ' 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.)r r}rprprqrOsicCst|dtdS)Nrb) buffering)openFILE_READ_BUFFER_SIZE)fnamerprprqrQscCsPt|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. )r3encodingerrors)r4r5r0r1 _CHUNK_SIZErrclose)r6ZfobjrprprqrPs 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)r6fallback_openfrprprqrRs   cCst||tdS)z,Same as above but opens file in binary mode.)r=r>)rRrQ)r6r=rprprqrS,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' ) BKMGTPEZYrlN r)symbolr) enumeratereversedabsrlocalsdict)nrsymbolsprefixr,srKrrprprqrT1scCs tjdjS)z+Return updated psutil.PROCFS_PATH constant.r)sysmodulesZ PROCFS_PATHrprprprqget_procfs_pathDsrWcCs|jttdS)N)r7r8)decoder0r1)rTrprprqrXIsrXcCs`tjdkrdSz2ddl}|s$t||ddks>tWntk rVYdSXdSdS)Nr[TrcolorsF)rr}cursesisattyrZ setuptermZtigetnumr)filerZrprprqrXRs  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) bluebrownZdarkgreygreenZgreyZ lightblueredvioletyellowZ29Ninvalid color z; choose amongst 1z[;mz)rXrPrlistr"rr)r)rTcolorboldattrrYrrprprqrWbs0    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\rNrk)r_r`r^rbrcz; choose between ii)rXprintrrWctypeswindllZKernel32 GetStdHandleSetConsoleTextAttributerPrrgr"rrUstderrZc_ulongrestype) rTrhrir\rqZ DEFAULT_COLORrsrtrYrZ handle_idhandlerprprqrY~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]> rk) rinspect getframeinfo currentframef_back isinstancerrrprUru)rryr6lineno__lines_indexrprprqrVs    )N)N)r@)NF)rrenumrrrrrUrrrrrrr ImportErrorrboolgetenvrobjectr;__all__r}rrplatform startswithr rrr r r r rZAIXr&r'r#r(r*r.r"r,r-r$r%r+r)r/rrrrrrrrrrrrIntEnumrjglobalsupdate __members__rrgetfilesystemencodingr0getfilesystemencodeerrorsr1rCr?r=r>r@rDr;rArBr<rrrrr3r8r9r:r5r6r7r4r2rrErrrrrrrKrHrrGrJrLrIrMrNrUr rFrrOr1rr0r5rQrPrRrSrTrWrXstdoutrXrWrYrVrprprprqs      &                          ! -G   W       (