­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ U =_jfN@s8dZddlmZmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZddl mZddlmZddlmZddlmZzdd lmZWnek rdZYnXzdd lmZWnek rdZYnXe jd krddlZndZe jdd kZd ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZgNZejd[kZejd\kZe jd]Ze jd^Z e Z!e jd_Z"e jd`Z#e jdaZ$e"pe#pe$Z%e jdbZ&e jdcZ'ddZ(deZ)dfZ*dgZ+dhZ,diZ-djZ.dkZ/dlZ0dmZ1dnZ2doZ3dpZ4dqZ5drZ6dsZ7dtZ8duZ9dvZ:dwZ;dxZd{Z?d|Z@d}ZAedkrd~ZBdZCdZDn GdddejEZFeGHeFjIedkrdZJdZKn GdddejEZLeGHeLjIe MZNesdZOn2z e PZOWn$eQk rHer@dndZOYnXedHddddddgZRedDddddgZSedBddddddgZTedCddddddgZUedEddddddddgZVedIdddddgZWed@dddddddgZXedFdddddgZYedGddddgZZedAddddgZ[eddddgZ\edddddgZ]eddddgZ^edddgZ_ed8ddddgZ`ed=ddgZaed>dddgZbed?dddgZced:dddgZded;ddddgZeedt||}|SdS)z5Calculate percentage usage of 'used' against 'total'.dgN)floatZeroDivisionErrorround)rrround_rrrrrOus 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||}|<|YSXdSr) frozensetsorteditemsKeyError)rkwargskeyrcachefunrrwrappers  zmemoize..wrappercs dS)z Clear cache.N)clearr)r rr cache_clearszmemoize..cache_clear) functoolswrapsr)r rrrr rrLs  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() >>> csTz|j}Wn@tk r*|YStk rN|}|j<YnX|Sr)_cacheAttributeErrorrrr rrrs z'memoize_when_activated..wrappercSs i|_dS)zjActivate cache. Expects a Process instance. Cache will be stored as a "_cache" instance attribute.N)rprocrrrcache_activatesz.memoize_when_activated..cache_activatecSs"z|`Wntk rYnXdS)zDeactivate and clear cache.N)rrrrrrcache_deactivatesz0memoize_when_activated..cache_deactivate)rrrr)r rrrrrrmemoize_when_activateds rc 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)rsterrrrrrKsc 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 )rr$rrrrNsc CsdtjrtdkrdSz4tttj}t||dW5QRXWdStjk r^YdSXdS)z2Return True if IPv6 is supported on this platform.NF)z::1rT)sockethas_ipv6r r contextlibclosingbinderror)sockrrrrPs 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=rz)rfindupper)datarposWINDOWS_next_pos equal_posr rrrrrMs   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 ValueErrornumrrrrQ"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)r5r% SocketKindr7r8rrrrR/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) r%rr rrgetrrQrRr6r?)rfamtype_rrr status_maprrrrrU<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)Nry)category stacklevel)warningswarnDeprecationWarninggetattr)rrr )r replacementrrinnerYsz/deprecated_method..outer..inner)r|rrr)r rFrErrouterSs z deprecated_method..outerr)rErIrrGrrJOs 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_keysrrrrrfs z_WrapNumbers.__init__cCsT||jkst||jkst||jks*t||j|<tt|j|<tt|j|<dSr)r AssertionErrorrNrOrintset)r input_dictrrrr _add_dictls  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)r rSkeysrOrN)rrTrold_dict gone_keysgone_keyremkeyrrr_remove_dead_remindersts  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)rMr rrNrOpop)rrrrrrs   z_WrapNumbers.cache_clearc Cs.|j|j|j|jfW5QRSQRXdS)z5Return internal cache dicts as a tuple of 3 elements.N)rMr rNrOrPrrr cache_infosz_WrapNumbers.cache_info)N) r|r}r~rrrUr[rgrrirrrrrJas ' rJc 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)_wnrMrg)rTrrrrrSscKst|df|S)Nrb)openfnamer rrr open_binarysrocKs*tr|dt|dtt|df|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'). encodingerrorsrt)PY3 setdefaultr4r5rlrmrrr open_texts  ru%(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' ) BKMGTPEZYrzN r)symbolr) enumeratereversedrlocalsdict)nformatsymbolsprefixrdsrrrrrrTs  cCs tjdjS)z+Return updated psutil.PROCFS_PATH constant.r)sysmodules PROCFS_PATHrrrrget_procfs_pathsrcCs|jttdS)N)rprq)decoder4r5rrrrrsrcCs|SrrrrrrrscCs`tjdkrdSz2ddl}|s$t||ddks>tWntk rVYdSXdSdS)Nr[TrcolorsF)rrcursesisattyrQ setuptermtigetnumr)filerrrrrXs  Fc Csts |Sg}tddddddddd d }d |d <z ||}Wn(tk rdtd t|YnX|||r~|ddd||fS)z*Return an highlighted version of 'string'.329133933435363730) greenredbrownyellowblueviolet lightbluegreydarkgrey29Nz#invalid color %r; choose between %s1z [%sm%s;)rXrrr7listrVrr)rcolorboldattrrrrrrW 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.rrNryr )rrrrz#invalid color %r; choose between %rii)rXprintrrWctypeswindllKernel32 GetStdHandleSetConsoleTextAttributerrr7rrVrstderrc_ulongrestype) rrrrr DEFAULT_COLORrrr handle_idhandlerrrrY s6     Z PSUTIL_DEBUGcCs6ttj\}}}}}td|||ftjddS)z@If PSUTIL_DEBUG env var is set, print a debug message to stderr.zpsutil-debug [%s:%s]> %srN)inspect getframeinfo currentframef_backrrr)rrnlineno func_namelinesindexrrrrVFs cCsdSrrrHrrrrVMs)N)N)rv)NF)r __future__rrr'rrrr%rrrKrA collectionsrrrrrr ImportErrorr version_infor5rs__all__rrrplatform startswithrrrr rrrrAIXr*r+r'r,r.r2r&r0r1r(r)r/r-r3rr!r rrr"rrrrrrr#r$r%IntEnumr{globalsupdate __members__rrrgetfilesystemencodingr4r5getfilesystemencodeerrorsrrGrCrArBrDrHr?rErFr@rrrrr7r<r=r>r9r:r;r8r6rrIrrrrrrrOrLrrKrNrPrMrQrRrUrJrJrSrjrrirorurTrrstdoutrXrWrYboolgetenvrrVrrrrs.         $                                 9   W     #