B
Œ=Ý_jf ã N @ s8 d Z ddlmZmZ ddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ ddl
mZ ddlmZ ddlmZ ddlmZ ydd lmZ W n ek
rÀ dZY nX ydd
lmZ W n ek
rê dZY nX e
jd krddlZndZe
jd d kZd
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZgNZejd[kZejd\kZ e
j d]¡Z e
j d^¡Z e Z!e
j d_¡Z"e
j d`¡Z#e
j da¡Z$e"pe#pe$Z%e
j db¡Z&e
j dc¡Z'ddZ(deZ)dfZ*dgZ+dhZ,diZ-djZ.dkZ/dlZ0dmZ1dnZ2doZ3dpZ4dqZ5drZ6dsZ7dtZ8duZ9dvZ:dwZ;dxZd{Z?d|Z@d}ZAedkr°d~ZBdZCdZDn G d€d„ dejEƒZFeGƒ HeFjI¡ edkräd‚ZJdƒZKn G d„d…„ d…ejEƒZLeGƒ HeLjI¡ e
M¡ ZNesd†ZOn2y e
P¡ ZOW n$ eQk
rH er@d‡nd†ZOY nX edHdˆd‰dŠd‹dŒdgƒZRedDdˆd‰dŠd‹gƒZSedBdŽddd‘d’d“gƒZTedCd”d•d–d—d˜d™gƒZUedEdšd›dœddždŸd d¡gƒZVedId¢d£d¤d¥d¦gƒZWed@d§d¨d©dªd«d¬d¦gƒZXedFd¨dd®d¯d°gƒZYedGd±d²d³d´gƒZZedAdµd¶d·d¸gƒZ[ed¹dºd»d¼gƒZ\ed½d¾dºd¿dÀgƒZ]edÁd‹dÂdÃgƒZ^edÄd¾dºgƒZ_ed8dÅdÆdÇdÈgƒZ`ed=dÉd§gƒZaed>dÊdËdÌgƒZbed?dÍdÎdÏgƒZced:dÍdÎdÏgƒZded;dŽddd‘gƒZeed>> @memoize
... def foo()
... return 1
...
>>> foo()
1
>>> foo.cache_clear()
>>>
c sH | t t| ¡ ƒƒf}yˆ | S tk
rB ˆ| |Ž }ˆ |< |S X d S )N)Ú frozensetÚsortedÚitemsÚKeyError)rö ÚkwargsÚkeyrë )ÚcacheÚfunr r€ Úwrapper s zmemoize..wrapperc s ˆ ¡ dS )z Clear cache.N)Úclearr )r r r€ Ú cache_clear˜ s z memoize..cache_clear)Ú functoolsÚwrapsr )r r r r )r r r€ rL s
c s6 t ˆ ¡‡ fdd„ƒ}dd„ }dd„ }||_||_|S )aú A 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 sP y| j ˆ }W n< tk
r& ˆ | ƒS tk
rJ ˆ | ƒ }| j ˆ < Y nX |S )N)Ú_cacheÚAttributeErrorr )rè rë )r r r€ r » s z'memoize_when_activated..wrapperc S s
i | _ dS )zjActivate cache. Expects a Process instance. Cache will be
stored as a "_cache" instance attribute.N)r )Úprocr r r€ Úcache_activateÉ s z.memoize_when_activated..cache_activatec S s" y| ` W n tk
r Y nX dS )zDeactivate and clear cache.N)r r
)r r r r€ Úcache_deactivateÎ s z0memoize_when_activated..cache_deactivate)r
r r r )r r r r r )r r€ Úmemoize_when_activated¡ s r c
C sV yt | ¡}W n6 tk
rD } z|jtjtjfkr4‚ dS d}~X Y nX t |j¡S dS )zœSame as os.path.isfile() but does not swallow EACCES / EPERM
exceptions, see:
http://mail.python.org/pipermail/python-dev/2012-June/120787.html
FN)ÚosÚstatÚOSErrorÚerrnoÚEPERMÚEACCESÚS_ISREGÚst_mode)rÉ ÚstÚerrr r r€ rK Ú s c
C sN yt | ¡ W n6 tk
rD } z|jtjtjfkr4‚ dS d}~X Y nX dS dS )zœSame as os.path.exists() but does not swallow EACCES / EPERM
exceptions, see:
http://mail.python.org/pipermail/python-dev/2012-June/120787.html
FNT)r r r r r r )rÉ r r r r€ rN é s c C s` t jrtdkrdS y2t tt j¡} t | ¡ | d¡ W dQ R X dS t jk
rZ dS X dS )z2Return True if IPv6 is supported on this platform.NF)z::1r T)ÚsocketÚhas_ipv6r r Ú
contextlibÚclosingÚbindÚerror)Úsockr r r€ rP ø s c C sz i }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 ú ú=rz )r ÚfindÚupper)Údatarë ÚposÚWINDOWS_Únext_posÚ equal_posr rÑ r r r€ rM s c C s0 t dkr | S y
t | ¡S tk
r* | S X dS )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)Únumr r r€ rQ " s
c C s0 t dkr | S y
t | ¡S tk
r* | S X dS )zConvert a numeric socket type value to an IntEnum member.
If it's not a known member, return the numeric value itself.
N)r, r Ú
SocketKindr. )r/ r r r€ rR / s
c C s |t jtfkr&|rt|Ž }|r&t|Ž }|t jkrJ|ttfkrJ| |t¡}nt}t|ƒ}t|ƒ}|dkrxt | |||||ƒS t
| ||||||ƒS dS )z2Convert a raw connection tuple to a proper ntuple.N) r r r rÔ r Úgetr rQ rR r6 r? )r§ ÚfamÚtype_rª r« r¬ Ú
status_mapr¦ r r r€ rU < s c s ‡ fdd„}|S )zƒA decorator which can be used to mark a method as deprecated
'replcement' is the method name which will be called instead.
c s: d| j ˆf ‰ | jd kr ˆ | _t | ¡‡ ‡fdd„ƒ}|S )Nz8%s() is deprecated and will be removed; use %s() insteadc s t jˆ tdd t| ˆƒ||ŽS )Nry )ÚcategoryÚ
stacklevel)ÚwarningsÚwarnÚDeprecationWarningÚgetattr)rè rö r )rç Ú replacementr r€ ÚinnerY s z/deprecated_method..outer..inner)r| rí r
r )r r< )r; )rç r€ ÚouterS s
z deprecated_method..outerr )r; r= r )r; r€ rJ O s c @ sB e Zd ZdZdd„ Zdd„ Zdd„ Zdd „ Zdd d „Zd
d„ Z d
S )Ú _WrapNumberszNWatches numbers so that they don't overflow and wrap
(reset to zero).
c C s t ¡ | _i | _i | _i | _d S )N)Ú threadingÚLockÚlockr Ú remindersÚ
reminder_keys)rè r r r€ ræ f s
z_WrapNumbers.__init__c C sT || j kst‚|| jks t‚|| jks*t‚|| j |< ttƒ| j|< ttƒ| j|< d S )N)r ÚAssertionErrorrB rC r ÚintÚset)rè Ú
input_dictr¢ r r r€ Ú _add_dictl s
z_WrapNumbers._add_dictc C sd | j | }t| ¡ ƒt| ¡ ƒ }x<|D ]4}x"| j| | D ]}| j| |= qY nX g }xvtt|ƒƒD ]f} || }
|| } || f} |
| k rÖ| j| | | 7 < | j| | | ¡ |
|
| j| | ¡ qˆW t |ƒ||< q>W || j |< |S )zkCache dict and sum numbers which overflow and wrap.
Return an updated copy of `input_dict`
) r rH rN rI r ÚrangeÚlenrB rC Úaddrô Útuple)
rè rG r¢ rJ Únew_dictr Ú input_tupleÚ old_tupleÚbitsÚiÚ input_valueÚ old_valuerM r r r€ Úrun s2
z_WrapNumbers.runNc C sh | j X |dkr0| j ¡ | j ¡ | j ¡ n*| j |d¡ | j |d¡ | j |d¡ W dQ R X dS )z>Clear the internal cache, optionally only for function 'name'.N)rA r r rB rC Úpop)rè r¢ r r r€ r ¦ s
z_WrapNumbers.cache_clearc C s" | j | j| j| jfS Q R X dS )z5Return internal cache dicts as a tuple of 3 elements.N)rA r rB rC )rè r r r€ Ú
cache_info² s z_WrapNumbers.cache_info)N)
r| r} r~ rí ræ rH rN rZ r r\ r r r r€ r> a s '
r> c C s t j t | |¡S Q R X dS )zÅGiven 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)Ú_wnrA rZ )rG r¢ r r r€ rS ¸ s c K s t | df|ŽS )NÚrb)Úopen)Úfnamer r r r€ Ú open_binaryÆ s ra c K s* t r | dt¡ | dt¡ t| df|ŽS )z¨On 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').
ÚencodingÚerrorsÚrt)ÚPY3Ú
setdefaultr4 r5 r_ )r` r r r r€ Ú open_textÊ s rg ú%(value).1f%(symbol)sc C sŽ d}i }x0t |dd… ƒD ] \}}d|d d > ||< qW x>t|dd… ƒD ]*}| || krLt| ƒ|| }|tƒ S qLW |t|d | d S )zŒUsed by various scripts. See:
http://goo.gl/zeJZl
>>> bytes2human(10000)
'9.8K'
>>> bytes2human(100001221)
'95.4M'
) ÚBÚKÚMÚGÚTÚPÚEÚZÚYrz Né
r )ÚsymbolrÑ )Ú enumerateÚreversedrû ÚlocalsÚdict)ÚnÚformatÚsymbolsÚprefixrW Úsrs rÑ r r r€ rT Ø s c C s t jd jS )z+Return updated psutil.PROCFS_PATH constant.rã )ÚsysÚmodulesÚ PROCFS_PATHr r r r€ Úget_procfs_pathì s r€ c C s | j ttdS )N)rb rc )Údecoder4 r5 )r| r r r€ r ò s r c C s | S )Nr )r| r r r€ r õ s c C s^ t jdkrdS y2dd l}| ¡ s$t‚| ¡ | d¡dks>t‚W n tk
rT dS X dS d S )Nr[ Tr ÚcolorsF)r r¢ ÚcursesÚisattyrD Ú setuptermÚtigetnumrå )Úfilerƒ r r r€ rX þ s
Fc C s t ƒ s
| S g }tddddddddd d
}d |d < y || }W n( tk
rd td
t| ¡ ƒ ƒ‚Y nX | |¡ |r~| d¡ dd |¡| f S )z*Return an highlighted version of 'string'.Ú32Ú91Ú33Ú93Ú34Ú35Ú36Ú37Ú30) ÚgreenÚredÚbrownÚyellowÚblueÚvioletÚ lightblueÚgreyÚdarkgreyÚ29Nz#invalid color %r; choose between %sÚ1z [%sm%s[0mú;)rX rw r r. ÚlistrI rô rõ )r| ÚcolorÚboldÚattrr‚ r r r€ rW
s
c C s t ƒ st| |d nêtr.tt| ||ƒ|d nÐddl}d}|jjj}|jjj}t ddddd}||d< y || }W n, t
k
rž t d |t |
¡ ƒf ƒ‚Y nX |r´|dkr´|d
7 }|tjkrÂd nd } |j|_|| ƒ}
||
|ƒ zt| |d W d||
|ƒ X dS )
z$Print a colorized version of string.)r‡ r Né ry r é )r‘ r’ r“ r” z#invalid color %r; choose between %ré iôÿÿÿiõÿÿÿ)rX Úprintr rW ÚctypesÚwindllÚKernel32Ú GetStdHandleÚSetConsoleTextAttributerw r r. r rI r} ÚstderrÚc_ulongÚrestype) r| rž rŸ r‡ r¥ Ú
DEFAULT_COLORr¨ r© r‚ Ú handle_idÚhandler r r€ rY s0
Z PSUTIL_DEBUGc C s6 t t ¡ j¡\}}}}}td||| f tjd dS )z@If PSUTIL_DEBUG env var is set, print a debug message to stderr.zpsutil-debug [%s:%s]> %s)r‡ N)ÚinspectÚ getframeinfoÚ currentframeÚf_backr¤ r} rª )rç r` ÚlinenoÚ func_nameÚlinesÚindexr r r€ rV F s c C s d S )Nr )rç r r r€ rV M s )N)N)rh )NF)Žrí Ú
__future__r r r r r
r r r r} r? r7 Ú collectionsr r r r r r Ú ImportErrorr
Ú version_infor, re Ú__all__r¢ r r ÚplatformÚ
startswithr r r r
r r r r ÚAIXr* r+ r' r, r. r2 r& r0 r1 r( r) r/ r- r3 r r! r r r r" r r r r r r r# r$ r% ÚIntEnumr{ ÚglobalsÚupdateÚ __members__r„ r… rƒ Úgetfilesystemencodingr4 r5 Úgetfilesystemencodeerrorsr
rG rC rA rB rD rH r? rE rF r@ r¹ r½ rÁ rÄ r7 r< r= r> r9 r: r; r8 r6 rÔ rI rå râ rï rò r÷ rø rO rL r rK rN rP rM rQ rR rU rJ r> rS r] r r\ ra rg rT r€ r ÚstdoutrX rW rY ÚboolÚgetenvr° rV r r r r€ Ú s|
9
W
"