Cs[)F � X @ s� d Z d d l m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l Z d d l m
Z
d d l m Z d d l m Z d d l m Z d d l m Z y d d l m Z Wn e k
r d Z Yn Xy d d
l m Z Wn e k
r/d Z Yn Xe j d� k rNd d l Z n d Z e j d d k Z d
d d d d d d d d d d d d d d d d d d d d! d" d# d$ d% d& d' d( d) d* d+ d, d- d. d/ d0 d1 d2 d3 d4 d5 d6 d7 d8 d9 d: d; d< d= d> d? d@ dA dB dC dD dE dF dG dH dI dJ dK dL dM dN dO dP dQ dR dS dT gH Z e j dU k Z e j dV k Z e j j dW � Z e j j dX � Z e Z e j j dY � Z! e j j dZ � Z" e j j d[ � Z# e! p�e" p�e# Z$ e j j d� � Z% e j j d^ � Z& d_ Z' d` Z( da Z) db Z* dc Z+ dd Z, de Z- df Z. dg Z/ dh Z0 di Z1 dj Z2 dk Z3 dl Z4 dm Z5 dn Z6 do Z7 dp Z8 dq Z9 dr Z: ds Z; dt Z< du Z= dv Z> dw Z? dx Z@ e d k r�dy ZA dz ZB d ZC n, Gd{ d| � d| e jD � ZE eF � jG eE jH � e d k r�d� ZI d� ZJ n, Gd} d~ � d~ e jD � ZK eF � jG eK jH � e jL � ZM e sDd ZN n7 y e jO � ZN Wn$ eP k
rze rpd� n d ZN Yn Xe dH d� d� d� d� d� d� g � ZQ e dD d� d� d� d� g � ZR e dB d� d� d� d� d� d� g � ZS e dC d� d� d� d� g � ZT e dE d� d� d� d� d� d� d� d� g � ZU e dI d� d� d� d� d� g � ZV e d@ d� d� d� d� d� d� d� g � ZW e dF d� d� d� d� d� g � ZX e dG d� d� d� d� g � ZY e dA d� d� d� d� g � ZZ e d� d� d� d� g � Z[ e d� d� d� d� d� g � Z\ e d� d� d� d� g � Z] e d� d� d� g � Z^ e d8 d� d� d� d� g � Z_ e d= d� d� g � Z` e d> d� d� d� g � Za e d? d� d� d� g � Zb e d: d� d� d� g � Zc e d; d� d� d� d� g � Zd e d< d� d� g � Ze e d9 d� d� g � Zf e d7 d� d� d� d� d� d� g � Zg e d� d� d� g � Zh d� e e e g e e g f d� e e g e g f d� e g e g f d� e e g e g f d� e g e g f d� e e g e e g f d� e g e e g f d� e g e e g f i Zi e d k r�ei jG d� e g e g f d� e g e g f i � e d k r ei jG d� e g e e g f i � [ [ [ [ d d� dP � Zj d� dM � Zk d� d� � Zl d� dL � Zm d� dO � Zn ek d� dQ � � Zo d� dN � Zp d� dR � Zq d� dS � Zr d� dK � Zs Gd� d� � d� � Zt d� dT � Zu et � Zv ev jw eu _w ev jx eu _x d S)�z9Common objects shared by __init__.py and _ps*.py modules.� )�divisionN)� defaultdict)�
namedtuple)�AF_INET)�
SOCK_DGRAM)� SOCK_STREAM)�AF_INET6)�AF_UNIX� � �FREEBSD�BSD�LINUX�NETBSD�OPENBSD�MACOS�OSX�POSIX�SUNOS�WINDOWS�ENCODING�
ENCODING_ERRSr �
CONN_CLOSE�CONN_CLOSE_WAIT� CONN_CLOSING�CONN_ESTABLISHED�CONN_FIN_WAIT1�CONN_FIN_WAIT2�
CONN_LAST_ACK� CONN_LISTEN� CONN_NONE�
CONN_SYN_RECV�
CONN_SYN_SENT�CONN_TIME_WAIT�NIC_DUPLEX_FULL�NIC_DUPLEX_HALF�NIC_DUPLEX_UNKNOWN� STATUS_DEAD�STATUS_DISK_SLEEP� STATUS_IDLE�
STATUS_LOCKED�STATUS_RUNNING�STATUS_SLEEPING�STATUS_STOPPED�STATUS_SUSPENDED�STATUS_TRACING_STOP�STATUS_WAITING�STATUS_WAKE_KILL�
STATUS_WAKING�
STATUS_ZOMBIE�
STATUS_PARKED�pconn� pcputimes�pctxsw�pgids�pio�pionice� popenfile�pthread�puids�sconn� scpustats�sdiskio� sdiskpart�
sdiskusage�snetio�snicaddr� snicstats�sswap�suser� conn_tmap�deprecated_method�
isfile_strict�memoize�parse_environ_block�path_exists_strict�
usage_percent�
supports_ipv6�sockfam_to_enum�socktype_to_enum� wrap_numbers�posix�nt�linux�darwinZfreebsdZopenbsdZnetbsd�sunos�solaris�aix�runningZsleepingz
disk-sleep�stoppedz tracing-stopZzombieZdeadz wake-killZwakingZidle�lockedZwaitingZ suspendedZparkedZ ESTABLISHEDZSYN_SENTZSYN_RECVZ FIN_WAIT1Z FIN_WAIT2Z TIME_WAITZCLOSEZ
CLOSE_WAITZLAST_ACKZLISTENZCLOSING�NONE� � c @ s" e Z d Z d Z d Z d Z d S)� NicDuplexr^ r_ r N)�__name__�
__module__� __qualname__r$ r% r& � rd rd � /_common.pyr` � s r` c @ s e Z d Z d Z d Z d S)� BatteryTimer_ r^ N������)ra rb rc �POWER_TIME_UNKNOWN�POWER_TIME_UNLIMITEDrd rd rd re rf � s rf �replace�surrogateescape�total�used�freeZpercent�sinZsoutZ
read_countZ write_countZ
read_bytesZ write_bytesZ read_timeZ
write_timeZdeviceZ
mountpointZfstypeZoptsZ
bytes_sentZ
bytes_recvZ packets_sentZ packets_recvZerrinZerroutZdropinZdropout�nameZterminal�host�started�pid�fd�family�type�laddr�raddr�status�addressZnetmaskZ broadcastZptpZisup�duplexZspeedZmtuZ ctx_switchesZ
interruptsZsoft_interruptsZsyscalls�scpufreq�current�min�max�shwtempZlabel�high�critical�sbatteryZsecsleftZ
power_plugged�sfan�user�system�
children_user�children_system�path�idZ user_timeZ system_time�realZ effectiveZsavedZioclass�valueZ voluntaryZ involuntary�addrZip�port�allZtcpZtcp4ZudpZudp4ZinetZinet4Zinet6Ztcp6Zudp6Zunixc C sr y | | d } Wn< t k
rP t | t � s@ t | t � rF d n d } Yn X| d k rj t | | � S| Sd S)z5Calculate percentage usage of 'used' against 'total'.�d g r N)�ZeroDivisionError�
isinstance�float�round)rn rm Zround_�retrd rd re rN s
/
c sI t j � � � � f d d � � } � f d d � } i � | | _ | S)a A 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()
>>>
c s^ | t t | j � � � f } y � | SWn- t k
rY � | | � } � | <| SYn Xd S)N)� frozenset�sorted�items�KeyError)�args�kwargs�keyr� )�cache�funrd re �wrapper% s
zmemoize.<locals>.wrapperc s � j � d S)z Clear cache.N)�clearrd )r� rd re � cache_clear. s z memoize.<locals>.cache_clear)� functools�wrapsr� )r� r� r� rd )r� r� re rK s
$ c ss t j � � � � � f d d � � � � f d d � } � � f d d � } i � d � _ | � _ | � _ � 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()
>>> foo()
1
>>> foo()
>>> foo()
>>>
c sR � j s � | � Sy � � } Wn& t k
rI � | � } � � <Yn X| Sd S)N)�cache_activatedr� )�selfr� )r� r� r� rd re r� Q s
z'memoize_when_activated.<locals>.wrapperc s
d � _ d S)zActivate cache.TN)r� rd )r� rd re �cache_activate\ s z.memoize_when_activated.<locals>.cache_activatec s d � _ � j � d S)zDeactivate and clear cache.FN)r� r� rd )r� r� rd re �cache_deactivate` s z0memoize_when_activated.<locals>.cache_deactivateF)r� r� r� r� r� )r� r� r� rd )r� r� r� re �memoize_when_activated7 s ' r� c C sq y t j | � } WnG t k
r\ } z' | j t j t j f k rF � d SWYd d } ~ Xn Xt j | j � Sd S)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�errrd rd re rJ l s c C sc y t j | � WnG t k
rZ } z' | j t j t j f k rD � d SWYd d } ~ Xn Xd Sd S)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� rd rd re rM { s c C s{ t j s t d k r d SyA t j t t j � } t j | � � | j d � Wd QRXd SWn t j k
rv d SYn Xd S)z2Return True if IPv6 is supported on this platform.NF�::1r T)r� r )�socket�has_ipv6r r �
contextlib�closing�bind�error)�sockrd rd re rO � s c C s� i } d } t } x� | j d | � } | | k r4 P| j d | | � } | | k r� | | | � } | | d | � } | r� | j � } | | | <| d } q W| S)zCParse a C environ block of environment variables into a dictionary.r � �=r_ )r �find�upper)�datar� �posZWINDOWS_Znext_posZ equal_posr� r� rd rd re rL � s
c C sD t d k r | Sy t j | � SWn t t f k
r? | SYn Xd S)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�AttributeError)�numrd rd re rP � s c C sD t d k r | Sy t j | � SWn t t f k
r? | SYn Xd S)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� Z AddressTyper� r� )r� rd rd re rQ � s c s � f d d � } | 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 sS d | j � f � | j d k r+ � | _ t j | � � � f d d � � } | S)Nz8%s() is deprecated and will be removed; use %s() insteadc s/ t j � d t d d �t | � � | | � S)N�category�
stacklevelr^ )�warnings�warn�
FutureWarning�getattr)r� r� r� )�msg� replacementrd re �inner� s z/deprecated_method.<locals>.outer.<locals>.inner)ra �__doc__r� r� )r� r� )r� )r� re �outer� s $z deprecated_method.<locals>.outerrd )r� r� rd )r� re rI � s c @ sa e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d d � Z d
d � Z d
S)� _WrapNumberszNWatches numbers so that they don't overflow and wrap
(reset to zero).
c C s. t j � | _ i | _ i | _ i | _ d S)N)� threading�Lock�lockr� � reminders�
reminder_keys)r� rd rd re �__init__� s z_WrapNumbers.__init__c C s7 | | j | <t t � | j | <t t � | j | <d S)N)r� r �intr� �setr� )r� �
input_dictrq rd rd re � _add_dict� s
z_WrapNumbers._add_dictc C s| | j | } t | j � � t | j � � } xF | D]> } x' | j | | D] } | j | | =qN W| j | | =q6 Wd S)z�In case the number of keys changed between calls (e.g. a
disk disappears) this removes the entry from self.reminders.
N)r� r� �keysr� r� )r� r� rq �old_dictZ gone_keysZgone_key�remkeyrd rd re �_remove_dead_reminders� s
"
z#_WrapNumbers._remove_dead_remindersc
C sP | | j k r# | j | | � | S| j | | � | j | } i } x� | j � D]� } | | } y | | } Wn t k
r� | | | <wS Yn Xg } x� t t | � � D]{ } | | }
| | } | | f } |
| k r | j | | | 7<| j | | j | � | j
|
| j | | � q� Wt | � | | <qS W| | j | <| S)zkCache dict and sum numbers which overflow and wrap.
Return an updated copy of `input_dict`
) r� r� r� r� r� �range�lenr� r� �add�append�tuple)
r� r� rq r� Znew_dictr� Z input_tupleZ old_tuple�bits�iZ input_valueZ old_valuer� rd rd re �run� s2
z_WrapNumbers.runNc
C s� | j �t | d k r@ | j j � | j j � | j j � n9 | j j | d � | j j | d � | j j | d � Wd QRXd S)z>Clear the internal cache, optionally only for function 'name'.N)r� r� r� r� r� �pop)r� rq rd rd re r� % s
z_WrapNumbers.cache_clearc
C s+ | j � | j | j | j f SWd QRXd S)z5Return internal cache dicts as a tuple of 3 elements.N)r� r� r� r� )r� rd rd re �
cache_info1 s
z_WrapNumbers.cache_info)
ra rb rc r� r� r� r� r� r� r� rd rd rd re r� � s ' r� c
C s% t j � t j | | � SWd QRXd S)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)�_wnr� r� )r� rq rd rd re rR 7 s
)r
r )rW rX rg rh )yr� Z
__future__r r� r� r� r� r� r� �sysr� r� � collectionsr r r r r r � ImportErrorr � version_infor� ZPY3�__all__rq r r �platform�
startswithr r r r r r r
r ZAIXr+ r, r( r- r/ r3 r' r1 r2 r) r* r0 r. r4 r r" r! r r r# r r r r r r r$ r% r& �IntEnumr` �globals�update� __members__ri rj rf �getfilesystemencodingr r Zgetfilesystemencodeerrorsr� rF rB r@ rA rC rG r> rD rE r? r} r� r� r� r6 r; r< r= r8 r9 r: r7 r5 r� rH rN rK r� rJ rM rO rL rP rQ rI r� rR r� r� r� rd rd rd re �<module> sN
5
W