B �=�_jf�N@s8dZddlmZmZddlZddlZddlZddlZddlZddl Z ddl Z ddlZddlZddl mZddl mZddlmZddlmZddlmZydd lmZWnek r�dZYnXydd lmZWnek r�dZYnXe jdk�rddlZndZe jddkZd ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZgNZejd[kZejd\kZe j�d]�Ze 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}ZAedk�r�d~ZBdZCdZDn Gd�d��d�ejE�ZFeG��HeFjI�edk�r�d�ZJd�ZKn Gd�d��d�ejE�ZLeG��HeLjI�e �M�ZNe�sd�ZOn2ye �P�ZOWn$eQk �rHe�r@d�nd�ZOYnXedHd�d�d�d�d�d�g�ZRedDd�d�d�d�g�ZSedBd�d�d�d�d�d�g�ZTedCd�d�d�d�d�d�g�ZUedEd�d�d�d�d�d�d�d�g�ZVedId�d�d�d�d�g�ZWed@d�d�d�d�d�d�d�g�ZXedFd�d�d�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�d�d�d�g�Zeed>> @memoize ... def foo() ... return 1 ... >>> foo() 1 >>> foo.cache_clear() >>> csH|tt|��f}y�|Stk rB�||�}�|<|SXdS)N)� frozenset�sorted�items�KeyError)r��kwargs�keyr�)�cache�funrr��wrapper�szmemoize..wrappercs��dS)zClear cache.N)�clearr)rrr��cache_clear�szmemoize..cache_clear)� functools�wrapsr )rrr r)rrr�rL�s cs6t��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() >>> csPy|j�}Wn<tk r&�|�Stk rJ�|�}|j�<YnX|S)N)�_cache�AttributeErrorr)r�r�)rrr�r�sz'memoize_when_activated..wrappercSs i|_dS)zjActivate cache. Expects a Process instance. Cache will be stored as a "_cache" instance attribute.N)r)�procrrr��cache_activate�sz.memoize_when_activated..cache_activatecSs"y|`Wntk rYnXdS)zDeactivate and clear cache.N)rr )rrrr��cache_deactivate�sz0memoize_when_activated..cache_deactivate)r rrr)rrrrr)rr��memoize_when_activated�src CsVyt�|�}Wn6tk rD}z|jtjtjfkr4�dSd}~XYnXt�|j�SdS)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�errrrr�rK�sc CsNyt�|�Wn6tk rD}z|jtjtjfkr4�dSd}~XYnXdSdS)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)rrrrrr)r�rrrr�rN�sc Cs`tjrtdkrdSy2t�ttj�}t�|��|�d�WdQRXdStjk rZdSXdS)z2Return True if IPv6 is supported on this platform.NF)z::1rT)�socket�has_ipv6r r� contextlib�closing�bind�error)�sockrrr�rP�scCszi}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_posrr�rrr�rMs cCs0tdkr|Sy t�|�Stk r*|SXdS)z�Convert 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)�numrrr�rQ"s 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)r,r� SocketKindr.)r/rrr�rR/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)rrr r�r�getrrQrRr6r?)r��fam�type_r�r�r�� status_mapr�rrr�rU<scs�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. cs:d|j�f�|jdkr�|_t�|��fdd��}|S)Nz8%s() is deprecated and will be removed; use %s() insteadcs tj�tdd�t|��||�S)Nry)�category� stacklevel)�warnings�warn�DeprecationWarning�getattr)r�r�r)r��replacementrr��innerYsz/deprecated_method..outer..inner)r|r�r r)rr<)r;)r�r��outerSs z deprecated_method..outerr)r;r=r)r;r�rJOsc@sBeZdZdZdd�Zdd�Zdd�Zdd �Zddd�Zd d�Z d S)�_WrapNumberszNWatches numbers so that they don't overflow and wrap (reset to zero). cCs t��|_i|_i|_i|_dS)N)� threading�Lock�lockr� reminders� reminder_keys)r�rrr�r�fs z_WrapNumbers.__init__cCsT||jkst�||jkst�||jks*t�||j|<tt�|j|<tt�|j|<dS)N)r�AssertionErrorrBrCr�int�set)r�� input_dictr�rrr�� _add_dictls z_WrapNumbers._add_dictcCsd|j|}t|��t|��}x<|D]4}x"|j||D]}|j||=qYnXg}xvtt|��D]f} || } || }|| f}| |kr�|j|||7<|j||� |�|� | |j||�q�Wt|�||<q>W||j|<|S)zkCache dict and sum numbers which overflow and wrap. Return an updated copy of `input_dict` )rrHrNrIr�range�lenrBrC�addr��tuple) r�rGr�rJ�new_dictr�input_tuple� old_tuple�bits�i�input_value� old_valuerMrrr��runs2 z_WrapNumbers.runNc Csh|j�X|dkr0|j��|j��|j��n*|j�|d�|j�|d�|j�|d�WdQRXdS)z>Clear the internal cache, optionally only for function 'name'.N)rArrrBrC�pop)r�r�rrr�r �s z_WrapNumbers.cache_clearc Cs"|j�|j|j|jfSQRXdS)z5Return internal cache dicts as a tuple of 3 elements.N)rArrBrC)r�rrr�� cache_info�sz_WrapNumbers.cache_info)N) r|r}r~r�r�rHrNrZr r\rrrr�r>as' r>c Cstj�t�||�SQRXdS)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)�_wnrArZ)rGr�rrr�rS�scKst|df|�S)N�rb)�open)�fnamerrrr��open_binary�sracKs*tr|�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� setdefaultr4r5r_)r`rrrr�� open_text�srg�%(value).1f%(symbol)scCs�d}i}x0t|dd��D]\}}d|dd>||<qWx>t|dd��D]*}|||krLt|�||}|t�SqLW|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�YrzN� r)�symbolr�)� enumerate�reversedr��locals�dict)�n�format�symbols�prefixrW�srsr�rrr�rT�s cCstjdjS)z+Return updated psutil.PROCFS_PATH constant.r�)�sys�modules�PROCFS_PATHrrrr��get_procfs_path�sr�cCs|jttd�S)N)rbrc)�decoder4r5)r|rrr�r��sr�cCs|S)Nr)r|rrr�r��scCs^tjdkrdSy2ddl}|��s$t�|��|�d�dks>t�Wntk rTdSXdSdS)Nr[Tr�colorsF)rr��curses�isattyrD� setupterm�tigetnumr�)�filer�rrr�rX�s FcCs�t�s |Sg}tddddddddd d � }d|d<y||}Wn(tk rdtd t|��YnX|�|�|r~|�d�dd�|�|fS)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�;)rXrwrr.�listrIr�r�)r|�color�bold�attrr�rrr�rW s c Cst�st||d�n�tr.tt|||�|d�n�ddl}d}|jjj}|jjj}t ddddd�}||d<y||}Wn,t k r�td |t|� ��f��YnX|r�|dkr�|d 7}|tjkr�dnd} |j|_|| �} || |�zt||d�Wd|| |�XdS) z$Print a colorized version of string.)r�rN�ryr�)r�r�r�r�z#invalid color %r; choose between %r�i��i��)rX�printrrW�ctypes�windll�Kernel32�GetStdHandle�SetConsoleTextAttributerwrr.r�rIr}�stderr�c_ulong�restype)r|r�r�r�r�� DEFAULT_COLORr�r�r�� handle_id�handlerrr�rY s0 ZPSUTIL_DEBUGcCs6t�t��j�\}}}}}td|||ftjd�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�indexrrr�rVFscCsdS)Nr)r�rrr�rVMs)N)N)rh)NF)�r�� __future__rrrrr rrrr}r?r7�collectionsrrrrrr �ImportErrorr �version_infor,re�__all__r�rr�platform� startswithrrrr rrrr�AIXr*r+r'r,r.r2r&r0r1r(r)r/r-r3rr!r rrr"rrrrrrr#r$r%�IntEnumr{�globals�update�__members__r�r�r��getfilesystemencodingr4r5�getfilesystemencodeerrorsr rGrCrArBrDrHr?rErFr@r�r�r�r�r7r<r=r>r9r:r;r8r6r�rIr�r�r�r�r�r�rOrLrrKrNrPrMrQrRrUrJr>rSr]r r\rargrTr�r��stdoutrXrWrY�bool�getenvr�rVrrrr��s| 9 W "