U 祡c;@sfdZddlZddlZddlZddlmZddlmZddlmZGddde Z Gdd d e Z Gd d d e Z Gd d d e Z Gddde ZzPejdkrddlmZn ddlmZGdddejZGdddejejZWnek rdZYnXddZGddde ZgZdZedD]ZegZ e!e q"ddZ"dd Z#d!d"Z$e%d#krbe$dS)$a>Tree-like exploration of object referrers. This module provides a base implementation for tree-like referrers browsing. The two non-interactive classes ConsoleBrowser and FileBrowser output a tree to the console or a file. One graphical user interface for referrers browsing is provided as well. Further types can be subclassed. All types share a similar initialisation. That is, you provide a root object and may specify further settings such as the initial depth of the tree or an output function. Afterwards you can print the tree which will be arranged based on your previous settings. The interactive browser is based on a TreeWidget implemented in IDLE. It is available only if you have Tcl/Tk installed. If you try to instantiate the interactive browser without having Tkinter installed, an ImportError will be raised. N)muppy)summary)tkinterc@s"eZdZdZdddZddZdS)_NodeaA node as it is used in the tree structure. Each node contains the object it represents and a list of children. Children can be other nodes or arbitrary other objects. Any object in a tree which is not of the type _Node is considered a leaf. NcCs||_g|_||_dS)a!You have to define the object this node represents. Also you can define an output function which will be used to represent this node. If no function is defined, the default str representation is used. keyword arguments str_func -- output function N)ochildrenstr_func)selfrrr 6/tmp/pip-unpacked-wheel-8ad_c8mj/pympler/refbrowser.py__init__&s z_Node.__init__cCs$|jdk r||jSt|jSdS)z,Override str(self.o) if str_func is defined.N)rrstrr r r r __str__3s  z _Node.__str__)N)__name__ __module__ __qualname____doc__r rr r r r rs rc@s4eZdZdZdejddfddZddZd d ZdS) RefBrowserzBase class to other RefBrowser implementations. This base class provides means to extract a tree from a given root object and holds information on already known objects (to avoid repetition if requested). TNcCs>||_||_||_||_||_g|_t|_|j|jdS)aYou have to provide the root object used in the refbrowser. keyword arguments maxdepth -- maximum depth of the initial tree str_func -- function used when calling str(node) repeat -- should nodes appear repeatedly in the tree, or should be referred to existing nodes stream -- output stream (used in derived classes) N) rootmaxdepthrrepeatstreamignoresetalready_includedappend)r rootobjectrrrrr r r r Ds zRefBrowser.__init__cCs |jt||j|jS)z+Get a tree of referrers of the root object.)rrinspect currentframe _get_treerrrr r r get_tree\szRefBrowser.get_treecCst|}t||j}|jt||dkr2|S|jt |j||D]}t |t rzt ddt|DrzqRt|}|js||jkr||}|jd||fqRt |tsR||jkrR|j|||dqR|S)a`Workhorse of the get_tree implementation. This is a recursive method which is why we have a wrapper method. root is the current root object of the tree which should be returned. Note that root is not of the type _Node. maxdepth defines how much further down the from the root the tree should be build. rcss|]}t|ttfVqdSN) isinstancerr).0refr r r usz'RefBrowser._get_tree..z%s (already included, id %s))gcZ get_referrersrrraddidrrrr r$dictanyrrr!)r rrZobjectsresrZ_idsr r r r!as.     zRefBrowser._get_tree) rrrrr_reprr r"r!r r r r r;s rc@s2eZdZdZdZdZdZdZd ddZd d Z dS) StreamBrowserzRefBrowser implementation which prints the tree to the console. If you don't like the looks, you can change it a little bit. The class attributes 'hline', 'vline', 'cross', and 'space' can be modified to your needs. -|+ NcCs"|dkr|}||dddS)z Print referrers tree to console. keyword arguments tree -- if not None, the passed tree will be printed. Otherwise it is based on the rootobject. N)r"_print)r treer r r print_treeszStreamBrowser.print_treecCs||j||j}d}t|tr0t|j}|t|7}||jtt|7}||j ksnt|trn|dkr|j |ddS||j 7}||j7}t|jdkr||j7}||j7}||j 7}||j7}|dkr| |jd||td|D]}|dd|j|j }||dkr4|ddd|j}| |j|||||dkrt|ddkrndS|j |dddqdS)aCompute and print a new line of the tree. This is a recursive function. arguments tree -- tree to print prefix -- prefix to the current line to print carryon -- prefix which is used to carry on the vertical lines r Nr(r5)countcrossvliner$rlenrr spacerrwritehliner7rangestriprstrip)r r8prefixZcarryonlevelZ len_childrenbr r r r7s:            zStreamBrowser._print)N) rrrrrCr?r>rAr9r7r r r r r1s r1cs eZdZdZfddZZS)ConsoleBrowserz/RefBrowser that prints to the console (stdout).cs$tt|j|||js tj|_dSr#)superrJr rsysstdout)r argskwargs __class__r r r szConsoleBrowser.__init__)rrrrr __classcell__r r rPr rJsrJcs"eZdZdZdfdd ZZS) FileBrowserz:RefBrowser implementation which prints the tree to a file.Ncs@|j}t|d|_ztt|j|dW5|j||_XdS)z Print referrers tree to file (in text format). keyword arguments tree -- if not None, the passed tree will be printed. wr8N)ropencloserKrSr9)r filenamer8Z old_streamrPr r r9s   zFileBrowser.print_tree)N)rrrrr9rRr r rPr rSsrS)rr<) TreeWidgetrUc@s(eZdZdZddZddZddZdS) _TreeNodezTreeNode used by the InteractiveBrowser. Not to be confused with _Node. This one is used in the GUI context. cCs8|jj|jjjd|j_|j||dS)z(Reload all referrers for this _TreeNode.r(N)itemreftreer!noder_clear_childrenexpandupdaterr r r reload_referrerss z_TreeNode.reload_referrerscCst|jjjdS)z8Print object which this _TreeNode represents to console.N)printr\r^rrr r r print_objectsz_TreeNode.print_objectcstj|tj|jddjd|jdjd|jd jd|j d jddfd d }|j d |dd d}ddd}ddd}d S)z|Override drawtext from _TreeWidget.TreeNode. This seems to be a good place to add the popup menu. r)Ztearoffzreload referrers)labelcommandrcr`zClose Popup Menu)recs|j|jdSr#)postZx_rootZy_root)eventmenur r do_popupsz$_TreeNode.drawtext..do_popupz NcSsdSr#r r rhr r r edit%sz _TreeNode.drawtext..editcSsdSr#r rlr r r edit_finish(sz'_TreeNode.drawtext..edit_finishcSsdSr#r rlr r r edit_cancel+sz'_TreeNode.drawtext..edit_cancel)N)N)N) _TreeWidgetTreeNodedrawtextrZMenucanvasZ add_commandrbrdZ add_separatorr`rebind)r rkrmrnror rir rr s     z_TreeNode.drawtextN)rrrrrbrdrrr r r r r[sr[c@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)_ReferrerTreeItemz&Tree item wrapper around _Node object.cCs0tj|tj||||_||_||_dS)zYou need to provide the parent window, the node this TreeItem represents, as well as the tree (_Node) which the node belongs to. N)rpTreeItemr rLabelr^ parentwindowr])r rxr^r]r r r r 1s  z_ReferrerTreeItem.__init__cCs2g}|jjD]}t|ts ||q ||j_dS)zClear children list from any TreeNode instances. Normally these objects are not required for memory profiling, as they are part of the profiler. N)r^rr$r[r)r Z new_childrenchildr r r r_=s    z!_ReferrerTreeItem._clear_childrencCs t|jSr#)r r^rr r r GetTextJsz_ReferrerTreeItem.GetTextcCs|s dSdS)z_Different icon when object cannot be expanded, i.e. has no referrers. pythonN) IsExpandablerr r r GetIconNameMsz_ReferrerTreeItem.GetIconNamecCs6t|jtsdSt|jjdkr$dSt|jjSdS)zqAn object is expandable when it is a node which has children and is a container object. FrTN)r$r^rr@rr_is_containerobjectrrr r r r|Us  z_ReferrerTreeItem.IsExpandablecCsrg}|jj}t|dkrJt|jjrJ|j|jjd|_||jj}|D]}t |j ||j}| |qN|S)zThis method is the point where further referrers are computed. Thus, the computation is done on-demand and only when needed. rr() r^rr@rr~rr]r!r_rurxr)r Zsublistrryr\r r r GetSubListbs   z_ReferrerTreeItem.GetSubListN) rrrrr r_rzr}r|rr r r r ru.s   rucCst|dt|S)z,Default str function for InteractiveBrowser.z(id=%s))rr0r+rr r r gui_default_str_function{src@s*eZdZdZdedfddZd ddZd S) InteractiveBrowsera!Interactive referrers browser. The interactive browser is based on a TreeWidget implemented in IDLE. It is available only if you have Tcl/Tk installed. If you try to instantiate the interactive browser without having Tkinter installed, an ImportError will be raised. rTcCs&tdkrtdt|||||dS)aGYou have to provide the root object used in the refbrowser. keyword arguments maxdepth -- maximum depth of the initial tree str_func -- function used when calling str(node) repeat -- should nodes appear repeatedly in the tree, or should be referred to existing nodes Nz4InteractiveBrowser requires Tkinter to be installed.)r ImportErrorrr )r rrrrr r r r s zInteractiveBrowser.__init__FcCs`t}tj|dddd}|jjdddt|||}t|j d|}| |r\| dS)zCreate interactive browser window. keyword arguments standalone -- Set to true, if the browser is not attached to other windows Zwhiterr()bgZhighlightthicknessZ takefocusZboth)r`fillN) rZTkrpZScrolledCanvasframepackrur"r[rsr`Zmainloop)r standaloneZwindowZscr\r^r r r mainszInteractiveBrowser.mainN)F)rrrrrr rr r r r rs  rrrcCs tt|Sr#)r typerr r r foosrcCstttd}|dS)Nr)rJrrr9)cbr r r print_samples rcCstttd}|ddS)Nrz sample.txt)rSrrr9)Zfbr r r write_samples r__main__)&rr)rrLZpymplerrrZpympler.util.compatrobjectrrr1rJrS version_infoZidlelibrZrpr8rqr[rvrwrurrrZ superlistrrDitmprrrrrr r r r s>   HM   4I -