���]= � @ s` d d l m Z m 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
Td d l
m Z m Z d d � Z
Gd d � d � Z Gd
d � d � Z d d
� Z e Gd d � d � � Z e Gd d � d e � � Z d d � Z Gd d � d � Z Gd d � d e � Z d d � Z Gd d � d � Z Gd d � d � Z d d � Z d S) � )�Sequence�Iterable)�total_orderingN)�*)�_get_object_traceback� _get_tracesc C s� x� d
D]� } t | � d k rM | d k rM | r? d | | f Sd | | f St | � d k sk | d k r� | r d | | f Sd | | f S| d
} q Wd S)N�B�KiB�MiB�GiB�TiB�d z%+.1f %sz%.1f %s�
i z%+.0f %sz%.0f %s)r r r
r r i ( )�abs)�size�signZunit� r �0/opt/alt/python35/lib64/python3.5/tracemalloc.py� _format_size
s
r c @ s^ e Z d Z d Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d S)� Statistic� tracebackr �countc C s | | _ | | _ | | _ d S)N)r r r )�selfr r r r r r �__init__% s zStatistic.__init__c C s t | j | j | j f � S)N)�hashr r r )r r r r �__hash__* s zStatistic.__hash__c C s4 | j | j k o3 | j | j k o3 | j | j k S)N)r r r )r �otherr r r �__eq__- s zStatistic.__eq__c C sY d | j t | j d � | j f } | j rU | j | j } | d t | d � 7} | S)Nz%s: size=%s, count=%iFz , average=%s)r r r r )r �text�averager r r �__str__2 s
zStatistic.__str__c C s d | j | j | j f S)Nz)<Statistic traceback=%r size=%i count=%i>)r r r )r r r r �__repr__< s zStatistic.__repr__c C s | j | j | j f S)N)r r r )r r r r � _sort_key@ s zStatistic._sort_keyN)r r r )
�__name__�
__module__� __qualname__� __slots__r r r r r! r" r r r r r s
r c @ s^ e Z d Z d Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z d S)�
StatisticDiffr r � size_diffr �
count_diffc C s1 | | _ | | _ | | _ | | _ | | _ d S)N)r r r( r r) )r r r r( r r) r r r r K s
zStatisticDiff.__init__c C s( t | j | j | j | j | j f � S)N)r r r r( r r) )r r r r r R s zStatisticDiff.__hash__c C sX | j | j k oW | j | j k oW | j | j k oW | j | j k oW | j | j k S)N)r r r( r r) )r r r r r r V s
zStatisticDiff.__eq__c C sn d | j t | j d � t | j d � | j | j f } | j rj | j | j } | d t | d � 7} | S)Nz %s: size=%s (%s), count=%i (%+i)FTz , average=%s)r r r r( r r) )r r r r r r r ] s
zStatisticDiff.__str__c C s& d | j | j | j | j | j f S)Nz9<StatisticDiff traceback=%r size=%i (%+i) count=%i (%+i)>)r r r( r r) )r r r r r! i s zStatisticDiff.__repr__c C s. t | j � | j t | j � | j | j f S)N)r r( r r) r r )r r r r r" n s zStatisticDiff._sort_keyN)r r r( r r) )
r# r$ r% r& r r r r r! r" r r r r r' D s r' c C s� g } x� | j � D]� \ } } | j | d � } | d k rr t | | j | j | j | j | j | j � } n$ t | | j | j | j | j � } | j | � q WxG | j � D]9 \ } } t | d | j d | j � } | j | � q� W| S)Nr )�items�popr' r r �append)� old_group� new_group�
statisticsr �statZpreviousr r r �_compare_grouped_statst s r1 c @ s� e Z d Z d Z d d � Z e d d � � Z e d d � � Z d d � Z d
d � Z d d
� Z
d d � Z d d � Z d S)�Frame�_framec C s
| | _ d S)N)r3 )r �framer r r r � s zFrame.__init__c C s | j d S)Nr )r3 )r r r r �filename� s zFrame.filenamec C s | j d S)N� )r3 )r r r r �lineno� s z Frame.linenoc C s | j | j k S)N)r3 )r r r r r r � s z Frame.__eq__c C s | j | j k S)N)r3 )r r r r r �__lt__� s z Frame.__lt__c C s
t | j � S)N)r r3 )r r r r r � s zFrame.__hash__c C s d | j | j f S)Nz%s:%s)r5 r7 )r r r r r � s z
Frame.__str__c C s d | j | j f S)Nz <Frame filename=%r lineno=%r>)r5 r7 )r r r r r! � s zFrame.__repr__N)r3 )
r# r$ r% r& r �propertyr5 r7 r r8 r r r! r r r r r2 � s r2 c @ s� e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z d d d � Z
d S)� Traceback�_framesc C s t j | � | | _ d S)N)r r r; )r �framesr r r r � s
zTraceback.__init__c C s
t | j � S)N)�lenr; )r r r r �__len__� s zTraceback.__len__c C sB t | t � r- t d d � | j | D� � St | j | � Sd S)Nc s s | ] } t | � Vq d S)N)r2 )�.0�tracer r r � <genexpr>� s z(Traceback.__getitem__.<locals>.<genexpr>)�
isinstance�slice�tupler; r2 )r �indexr r r � __getitem__� s zTraceback.__getitem__c C s | j | j k S)N)r3 r; )r r4 r r r � __contains__� s zTraceback.__contains__c C s
t | j � S)N)r r; )r r r r r � s zTraceback.__hash__c C s | j | j k S)N)r; )r r r r r r � s zTraceback.__eq__c C s | j | j k S)N)r; )r r r r r r8 � s zTraceback.__lt__c C s t | d � S)Nr )�str)r r r r r � s zTraceback.__str__c C s d t | � f S)Nz<Traceback %r>)rD )r r r r r! � s zTraceback.__repr__Nc C s� g } | d k r" | d k r" | Sxj | d | � D]X } | j d | j | j f � t j | j | j � j � } | r3 | j d | � q3 W| S)Nr z File "%s", line %sz %s)r, r5 r7 � linecache�getline�strip)r �limit�linesr4 �liner r r �format� s zTraceback.format)r; )r# r$ r% r& r r> rF rG r r r8 r r! rO r r r r r: � s r: c C s* t | � } | d k r" t | � Sd Sd S)N)r r: )�objr< r r r �get_object_traceback� s
rQ c @ sv e Z d Z d Z d d � Z e d d � � Z e d d � � Z d d � Z d
d � Z d d
� Z
d d � Z d S)�Trace�_tracec C s
| | _ d S)N)rS )r r@ r r r r � s zTrace.__init__c C s | j d S)Nr )rS )r r r r r � s z
Trace.sizec C s t | j d � S)Nr6 )r: rS )r r r r r � s zTrace.tracebackc C s | j | j k S)N)rS )r r r r r r s z Trace.__eq__c C s
t | j � S)N)r rS )r r r r r s zTrace.__hash__c C s d | j t | j d � f S)Nz%s: %sF)r r r )r r r r r s z
Trace.__str__c C s d t | j d � | j f S)Nz <Trace size=%s, traceback=%r>F)r r r )r r r r r! s zTrace.__repr__N)rS ) r# r$ r% r& r r9 r r r r r r! r r r r rR � s rR c @ sX e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�_Tracesc C s t j | � | | _ d S)N)r r �_traces)r �tracesr r r r s
z_Traces.__init__c C s
t | j � S)N)r= rU )r r r r r> s z_Traces.__len__c C sB t | t � r- t d d � | j | D� � St | j | � Sd S)Nc s s | ] } t | � Vq d S)N)rR )r? r@ r r r rA s z&_Traces.__getitem__.<locals>.<genexpr>)rB rC rD rU rR )r rE r r r rF s z_Traces.__getitem__c C s | j | j k S)N)rS rU )r r@ r r r rG s z_Traces.__contains__c C s | j | j k S)N)rU )r r r r r r # s z_Traces.__eq__c C s d t | � S)Nz<Traces len=%s>)r= )r r r r r! & s z_Traces.__repr__N) r# r$ r% r r> rF rG r r! r r r r rT s rT c C s5 t j j | � } | j d � r1 | d d � } | S)Nz.pycr6 ���)�os�path�normcase�endswith)r5 r r r �_normalize_filename* s r\ c @ sX e Z d Z d d d d � Z e d d � � Z d d � Z d d
� Z d d � Z d S)
�FilterNFc C s. | | _ t | � | _ | | _ | | _ d S)N)� inclusiver\ �_filename_patternr7 �
all_frames)r r^ �filename_patternr7 r` r r r r 2 s zFilter.__init__c C s | j S)N)r_ )r r r r ra 9 s zFilter.filename_patternc C sI t | � } t j | | j � s% d S| j d k r8 d S| | j k Sd S)NFT)r\ �fnmatchr_ r7 )r r5 r7 r r r Z
__match_frame= s zFilter.__match_framec C s | j | | � | j AS)N)�_Filter__match_framer^ )r r5 r7 r r r � _match_frameF s zFilter._match_framec s^ � j r: t � f d d � | D� � r/ � j S� j Sn | d \ } } � j | | � Sd S)Nc 3 s' | ] \ } } � j | | � Vq d S)N)rc )r? r5 r7 )r r r rA K s z*Filter._match_traceback.<locals>.<genexpr>r )r` �anyr^ rd )r r r5 r7 r )r r �_match_tracebackI s
zFilter._match_traceback) r# r$ r% r r9 ra rc rd rf r r r r r] 1 s
r] c @ s| e Z d Z d d � Z d d � Z e d d � � Z d d � Z d d
� Z d d � Z d
d d � Z
d
d d � Z d S)�Snapshotc C s t | � | _ | | _ d S)N)rT rV �traceback_limit)r rV rh r r r r Z s zSnapshot.__init__c C s3 t | d � � } t j | | t j � Wd QRXd S)N�wb)�open�pickle�dump�HIGHEST_PROTOCOL)r r5 �fpr r r rl ` s z
Snapshot.dumpc C s* t | d � � } t j | � SWd QRXd S)N�rb)rj rk �load)r5 rn r r r rp g s z
Snapshot.loadc s` | d � | r3 t � f d d � | D� � s3 d S| r\ t � f d d � | D� � r\ d Sd S)Nr6 c 3 s | ] } | j � � Vq d S)N)rf )r? � trace_filter)r r r rA r s z)Snapshot._filter_trace.<locals>.<genexpr>Fc 3 s | ] } | j � � Vq d S)N)rf )r? rq )r r r rA v s T)re )r �include_filters�exclude_filtersr@ r )r r �
_filter_traceo s
zSnapshot._filter_tracec s� t | t � s( t d t | � j � � | r� g � g � x4 | D], } | j r` � j | � qA � j | � qA W� � � f d d � � j j D� } n � j j j � } t
| � j � S)Nz)filters must be a list of filters, not %sc s+ g | ]! } � j � � | � r | � q Sr )rt )r? r@ )rs rr r r r �
<listcomp>� s z*Snapshot.filter_traces.<locals>.<listcomp>) rB r � TypeError�typer# r^ r, rV rU �copyrg rh )r �filtersrq Z
new_tracesr )rs rr r r �
filter_traces{ s
(zSnapshot.filter_tracesc C sG | d k r t d | f � � | rA | d k rA t d | � � i } i } | sTx�| j j D]� } | \ } } y | | } Wns t k
r� | d k r� | } n6 | d k r� | d d � } n | d d d f f } t | � } | | | <Yn Xy, | | }
|
j | 7_ |
j d 7_ Wq` t k
rLt | | d � | | <Yq` Xq` Wn� x� | j j D]� } | \ } } x� | D]� } y | | } WnS t k
r�| d k r�| f } n | d d f f } t | � } | | | <Yn Xy, | | }
|
j | 7_ |
j d 7_ Wqzt k
r:t | | d � | | <YqzXqzWqaW| S)
Nr r5 r7 zunknown key_type: %rz/cumulative mode cannot by used with key type %rr6 r )r r5 r7 )r7 r5 )�
ValueErrorrV rU �KeyErrorr: r r r ) r �key_type�
cumulativeZstatsZ
tracebacksr@ r Ztrace_tracebackr r< r0 r4 r r r � _group_by� sX
"
#zSnapshot._group_byFc C sA | j | | � } t | j � � } | j d d d t j � | S)N�reverseT�key)r �list�values�sortr r" )r r} r~ Zgroupedr/ r r r r/ � s zSnapshot.statisticsc C sP | j | | � } | j | | � } t | | � } | j d d d t j � | S)Nr� Tr� )r r1 r� r' r" )r Z old_snapshotr} r~ r. r- r/ r r r �
compare_to� s
zSnapshot.compare_toN) r# r$ r% r rl � staticmethodrp rt rz r r/ r� r r r r rg U s 3
rg c C s4 t � s t d � � t � } t � } t | | � S)NzLthe tracemalloc module must be tracing memory allocations to take a snapshot)Z
is_tracing� RuntimeErrorr Zget_traceback_limitrg )rV rh r r r �
take_snapshot� s
r� )� collectionsr r � functoolsr rb rI Zos.pathrX rk Z _tracemallocr r r r r' r1 r2 r: rQ rR rT r\ r] rg r� r r r r �<module> s*
&0 # 5 ! $�