i f#= @ 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) )SequenceIterable)total_orderingN)*)_get_object_traceback _get_tracesc C s x d
D] } t | d k rP | d k rP | r? d | | f Sd | | f Sn t | d k sn | d k r | r d | | f Sd | | f Sn | d
} q Wd S)NBKiBMiBGiBTiBd z%+.1f %sz%.1f %s
i z%+.0f %sz%.0f %s)r r r
r r i ( )abs)sizeZsignZunit r 0/opt/alt/python34/lib64/python3.4/tracemalloc.py _format_size
s
r c @ sd e Z d 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) StatisticzS
Statistic difference on memory allocations between two Snapshot instance.
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 s\ d | j t | j d | j f } | j rX | j | j } | d t | d 7} n | S)Nz%s: size=%s, count=%iFz , average=%s)r r r r )r textaverager r r __str__2 s
zStatistic.__str__c C s d | j | j | j f S)Nz))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)z tracebackzsizezcount) __name__
__module__ __qualname____doc__ __slots__r r r r r r! r r r r r s
r c @ sd e Z d 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)
StatisticDiffzd
Statistic difference on memory allocations between an old and a new
Snapshot instance.
r 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 sq d | j t | j d t | j d | j | j f } | j rm | j | j } | d t | d 7} n | 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)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)z tracebackzsizez size_diffzcountz
count_diff) 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 )itemspopr' 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 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)Framez
Frame of a traceback.
_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 )r5 r7 )r r r r r s zFrame.__repr__N)z_frame)r" 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 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) Tracebackz`
Sequence of Frame instances sorted from the most recent frame
to the oldest frame.
_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 ).0tracer r r s z(Traceback.__getitem__..)
isinstanceslicetupler; 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)rD )r r r r r s zTraceback.__repr__Nc C s g } | d k r" | d k r" | Sxm | d | D][ } | j d | j | j f t j | j | j j } | r3 | j d | q3 q3 W| S)Nr z File "%s", line %sz %s)r, r5 r7 linecachegetlinestrip)r limitlinesr4 liner r r format s zTraceback.format)z_frames)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)z
Get the traceback where the Python object *obj* was allocated.
Return a Traceback instance.
Return None if the tracemalloc module is not tracing memory allocations or
did not trace the allocation of the object.
N)r r: )objr< r r r get_object_traceback s
rQ c @ s| e Z d 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)Tracez"
Trace of a memory block.
_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 F)r r r )r r r r r s zTrace.__repr__N)z_trace)
r" 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__..)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)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 s8 t j j | } | j d r4 | d d } n | S)N.pyc.pyor6 )rW rX )ospathnormcaseendswith)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 rb r r r r 2 s zFilter.__init__c C s | j S)N)ra )r r r r rc 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^ fnmatchra 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)re )r? r5 r7 )r r r rA K s z*Filter._match_traceback..r )rb anyr` rf )r r r5 r7 r )r r _match_tracebackI s
zFilter._match_traceback) r" r# r$ r r9 rc re rf rh r r r r r_ 1 s
r_ c @ s e Z d 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)SnapshotzB
Snapshot of traces of memory blocks allocated by Python.
c C s t | | _ | | _ d S)N)rT rV traceback_limit)r rV rj r r r r Z s zSnapshot.__init__c C s2 t | d } t j | | t j Wd QXd S)z1
Write the snapshot into a file.
wbN)openpickledumpZHIGHEST_PROTOCOL)r r5 fpr r r rn ` s z
Snapshot.dumpc C s) t | d } t j | SWd QXd S)z.
Load a snapshot from a file.
rbN)rl rm load)r5 ro r r r rq g s z
Snapshot.loadc sf | d | r6 t f d d | D s6 d Sn | rb t f d d | D rb d Sn d S)Nr6 c 3 s | ] } | j Vq d S)N)rh )r? trace_filter)r r r rA r s z)Snapshot._filter_trace..Fc 3 s | ] } | j Vq d S)N)rh )r? rr )r r r rA v s T)rg )r include_filtersexclude_filtersr@ r )r r
_filter_traceo s
zSnapshot._filter_tracec s t | t s+ t d t | j n | r g g x4 | D], } | j rc j | qD j | qD W f d d j j D } n j j j } t
| j S)z
Create a new Snapshot instance with a filtered traces sequence, filters
is a list of Filter instances. If filters is an empty list, return a
new Snapshot instance with a copy of the traces.
z)filters must be a list of filters, not %sc s+ g | ]! } j | r | q Sr )ru )r? r@ )rt rs r r r
s z*Snapshot.filter_traces..) rB r TypeErrortyper" r` r, rV rU copyri rj )r filtersrr Z
new_tracesr )rt rs r r
filter_traces{ s
(zSnapshot.filter_tracesc C sM | d k r" t d | f n | rG | d k rG t d | n i } i } | sZx| j j D] } | \ } } y | | } Wns t k
r | d k r | } n6 | d k r | d d } n | d d d f f } t | } | | | s*
&0 # 5 ! $