U 祡c(@sTdZddlZddlZddlmZmZddlmZGdddeZ GdddeZ dS) a|The tracker module allows you to track changes in the memory usage over time. Using the SummaryTracker, you can create summaries and compare them with each other. Stored summaries can be ignored during comparison, avoiding the observer effect. The ObjectTracker allows to monitor object creation. You create objects from one time and compare with objects from an earlier time. N)muppysummary)compatc@sHeZdZdZdddZddZddd Zdd d Zdd d ZddZ dS)SummaryTrackerab Helper class to track changes between two summaries taken. Detailed information on single objects will be lost, e.g. object size or object id. But often summaries are sufficient to monitor the memory usage over the lifetime of an application. On initialisation, a first summary is taken. Every time `diff` is called, a new summary will be created. Thus, a diff between the new and the last summary can be extracted. Be aware that filtering out previous summaries is time-intensive. You should therefore restrict yourself to the number of summaries you really need. TcCs tt|_i|_||_dS)aXConstructor. The number of summaries managed by the tracker has a performance impact on new summaries, iff you decide to exclude them from further summaries. Therefore it is suggested to use them economically. Keyword arguments: ignore_self -- summaries managed by this object will be ignored. N)r summarizer get_objectss0 summaries ignore_self)selfr r 3/tmp/pip-unpacked-wheel-8ad_c8mj/pympler/tracker.py__init__#s zSummaryTracker.__init__cs|jstt}ngifdd}||j|jD]\}}||t||q@tt}D]"}tt |dkrpt ||qpD].}tt |t |dkrt ||q|S)zReturn a summary. See also the notes on ignore_self in the class as well as the initializer documentation. cs<|t|kr,t|d7<n dt|<dS)N)appendid)oZ all_of_themZ ref_counterr r store_infoMs  z1SummaryTracker.create_summary..store_info) r rrrrr itemsZ _traverselengcZ get_referrersZ _subtractr)r resrkvZ_idrr rr create_summary1s" zSummaryTracker.create_summaryNcCspd}|dkrH||_|dkr0t|j|j}nt||j}|j|_n|dk r^t||}ntdt|S)a9Compute diff between to summaries. If no summary is provided, the diff from the last to the current summary is used. If summary1 is provided the diff from summary1 to the current summary is used. If summary1 and summary2 are provided, the diff between these two is used. Nz-You cannot provide summary2 without summary1.)rs1rget_diffr ValueErrorZ_sweep)r summary1summary2rr r r diffis   zSummaryTracker.diffcCst|j||ddS)aECompute diff between to summaries and print it. If no summary is provided, the diff from the last to the current summary is used. If summary1 is provided the diff from summary1 to the current summary is used. If summary1 and summary2 are provided, the diff between these two is used. r!r"N)rprint_r#r r!r"r r r print_diffszSummaryTracker.print_diffcCst|j||dS)aeCompute diff between to summaries and return a list of formatted lines. If no summary is provided, the diff from the last to the current summary is used. If summary1 is provided the diff from summary1 to the current summary is used. If summary1 and summary2 are provided, the diff between these two is used. r$)rformat_r#r&r r r format_diffs zSummaryTracker.format_diffcCs||j|<dS)z*Store a current summary in self.summaries.N)rr )r keyr r r store_summaryszSummaryTracker.store_summary)T)NN)NN)NN) __name__ __module__ __qualname____doc__rrr#r'r)r+r r r r rs 8  rc@s@eZdZdZddZdddZdddZdd d Zdd d Zd S) ObjectTrackeraq Helper class to track changes in the set of existing objects. Each time you invoke a diff with this tracker, the objects which existed during the last invocation are compared with the objects which exist during the current invocation. Please note that in order to do so, strong references to all objects will be stored. This means that none of these objects can be garbage collected. A use case for the ObjectTracker is the monitoring of a state which should be stable, but you see new objects being created nevertheless. With the ObjectTracker you can identify these new objects. cCs|jtfd|_dS)zOn initialisation, the current state of objects is stored. Note that all objects which exist at this point in time will not be released until you destroy this ObjectTracker instance. )ignoreN) _get_objectsinspect currentframeo0)r r r r rszObjectTracker.__init__r cCsddd}t}|t|||f7}t|dr<||jf7}t|drR||jf7}|||}g}|D],}t|}|D]}t |sv| |qvqd| |t |}|dk r|||}~|S)a>Get all currently existing objects. XXX - ToDo: This method is a copy&paste from muppy.get_objects, but some modifications are applied. Specifically, it allows to ignore objects (which includes the current frame). keyword arguments ignore -- list of objects to ignore r cSs(g}|D]}t||s||q|S)N)rZobject_in_listr)Zobjectsr1rrr r r remove_ignores   z1ObjectTracker._get_objects..remove_ignorer5o1N)r ) rrr3r4hasattrr5r7rZ get_referentsZ is_trackedrextendZ_remove_duplicates)r r1r6tmprrrefsrefr r r r2s(           zObjectTracker._get_objectscCs2||tf|_t|j|j}|j|_|S)zGet the diff to the last time the state of objects was measured. keyword arguments ignore -- list of objects to ignore )r2r3r4r7rrr5)r r1r#r r r rszObjectTracker.get_diffcCs&||tfD] }t|qdS)zPrint the diff to the last time the state of objects was measured. keyword arguments ignore -- list of objects to ignore N)r)r3r4print)r r1liner r r r'szObjectTracker.print_diffcCsxg}||tf}|dtt|dD]}||q6|dtt|dD]}||qd|S)zFormat the diff to the last time the state of objects was measured. keyword arguments ignore -- list of objects to ignore zAdded objects:+zRemoved objects:-)rr3r4rrr(r)r r1linesr#r>r r r r)s    zObjectTracker.format_diffN)r )r )r )r ) r,r-r.r/rr2rr'r)r r r r r0s  - r0) r/rr3ZpymplerrrZ pympler.utilrobjectrr0r r r r s