JezK
Edit File: tracemalloc.cpython-34.pyc
� � 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) � )�Sequence�Iterable)�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)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)�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 �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)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<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)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 )�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 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 <Frame filename=%r lineno=%r>)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 )�.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" | 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 � linecache�getline�strip)r �limit�linesr4 �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 <Trace size=%s, traceback=%r>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__.<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 s8 t j j | � } | j d � r4 | d d � } n | S)N�.pyc�.pyor6 )rW rX ���)�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 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.<locals>.<genexpr>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)�open�pickle�dumpZHIGHEST_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.<locals>.<genexpr>Fc 3 s | ] } | j � � Vq d S)N)rh )r? rr )r r r rA v s T)rg )r �include_filters�exclude_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 � <listcomp>� s z*Snapshot.filter_traces.<locals>.<listcomp>) rB r � TypeError�typer" 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 | � } | | | <Yn Xy, | | } | j | 7_ | j d 7_ Wqf t k rRt | | d � | | <Yqf Xqf 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_ Wq�t k r@t | | d � | | <Yq�Xq�WqgW| S) Nr r5 r7 zunknown key_type: %rz/cumulative mode cannot by used with key type %rr6 r )z traceback�filename�lineno)r} r| )� 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)zd Group statistics by key_type. Return a sorted list of Statistic instances. �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)z� Compute the differences with an old snapshot old_snapshot. Get statistics as a sorted list of StatisticDiff instances, grouped by group_by. r� 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% r rn � staticmethodrq ru r{ r� r/ r� r r r r ri U s 3 ri c C s7 t � s t d � � n t � } t � } t | | � S)zI Take a snapshot of traces of memory blocks allocated by Python. zLthe tracemalloc module must be tracing memory allocations to take a snapshot)Z is_tracing� RuntimeErrorr Zget_traceback_limitri )rV rj r r r � take_snapshot� s r� )� collectionsr r � functoolsr rd rI Zos.pathrZ rm Z _tracemallocr r r r r' r1 r2 r: rQ rR rT r^ r_ ri r� r r r r �<module> s* &0 # 5 ! $�