U 祡c*@sdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddlm Z ddl mZdd lmZdd lmZdd lmZdd lmZmZdd lmZz ddlZWn ek rddlmZYnXGdddejZeZddZ ddZ!ej"#ej"$e%Z&ej"'e&dZ(ej)*e(e+de,dddZ-e+de,dddZ.e+de,ddd Z/e+d!e,d"d#d"Z0e+d$d%d&Z1e+d'e,d(d)d*Z2e+d+e,d,d-d.Z3e+d/d0d1Z4d2d3Z5e+d4e,d5d6d5Z6e+d7e,d8d9d:Z7d;d<Z8e+d=d>d?Z9e+d@dAdBZ:GdCdDdDej;ZdLdMZ?dS)Oak This module provides a web-based memory profiling interface. The Pympler web frontend exposes process information, tracker statistics, and garbage graphs. The web frontend uses `Bottle `_, a lightweight Python web framework. Bottle is packaged with Pympler. The web server can be invoked almost as easily as setting a breakpoint using *pdb*:: from pympler.web import start_profiler start_profiler() Calling ``start_profiler`` suspends the current thread and executes the Pympler web server, exposing profiling data and various facilities of the Pympler library via a graphic interface. N)getouterframes)dumps)rmtree)mkdtemp)Thread)WeakValueDictionary) make_server)asizeof) GarbageGraph)get_current_threadsProcessMemoryInfo) safe_repr)bottlec@s eZdZdZddZddZdS) ServerStatez Represents the state of a running server. Needs to be thread local so multiple servers can be started in different threads without interfering with each other. Cache internal structures (garbage graphs, tracker statistics). cCs&d|_d|_d|_t|_t|_dSN)serverstatsgarbage_graphsrid2refdictid2objselfr//tmp/pip-unpacked-wheel-8ad_c8mj/pympler/web.py__init__4s zServerState.__init__cCs d|_dSrrrrrr clear_cache;szServerState.clear_cacheN)__name__ __module__ __qualname____doc__rrrrrrr,srcCs>t|}z|tj|<Wntk r4|tj|<YnXt|S)a* Get string reference to object. Stores a weak reference in a dictionary using the object's id as the key. If the object cannot be weakly referenced (e.g. dictionaries, frame objects), store a strong references in a classic dictionary. Returns the object's id as a string. )idrr TypeErrorrstr)objoidrrrget_refBs r'cCst|}tj|ptj|S)z!Get object from string reference.)intrrgetr)refr&rrrget_objSsr+Z templates/indexcCst}t|dS)z Get overview.)Z processinfo)r r)pmirrrroot_sr/z/processprocesscCst}t}t||dS)zGet process overview.)infothreads)r r r)r.r2rrrr0gs/trackertrackercCsZtj}|rL|jrL|g}|jD]>}g}|jD] }||j|iddq4|||fq&dd|jD}|d|f|jdjj rdd|jD}|d|fdd|jD}|d |fd d|jD}|d |fd d|jD}|d |fndd|jD}|d|fdd|D}t |jt |dSt gdSdS)zGet tracker overview.sumrcSsg|] }|jqSr)overhead.0srrr ~sz!tracker_index..zProfiling overheadcSs g|]}|jj|j|jqSr) system_total data_segment tracked_totalr6r7rrrr:sz Data segmentcSsg|] }|jjqSr)r;Z code_segmentr7rrrr:sz Code segmentcSsg|] }|jjqSr)r;Z stack_segmentr7rrrr:sz Stack segmentcSsg|] }|jjqSr)r;Zshared_segmentr7rrrr:sz Shared memorycSsg|]}|j|j|jqSr)totalr=r6r7rrrr:sZOthercSs$g|]\}}t|tt|dqS))labeldata)rlist enumerate)r8r?r@rrrr:s) snapshots timeseries)rCN) rrrCannotateZtracked_classesappendclassesr)r;r<rr)rrDclsZseriesZsnapshotrrr tracker_indexps>  rIz/tracker/class/ tracker_classcCs(tj}|std|t||dS)zGet class instance details.r3)rclsname)rrrredirectrEr)rKrrrrrJs  z/refreshcCsttddS)zClear all cached information.r,N)rrrrLrrrrrefreshsrMz/traceback/Z stacktracecCsRt|}t}||krB||}t|d}|dd|D}ng}t||dS)NcSs(g|] }t|djf|ddqS)rN)r'f_locals)r8frrrr:sz!get_traceback..)stackthreadid)r(sys_current_framesrreverser)rSframesframerRrrr get_tracebacks rYz/objects/ referentscCs|i}t|}t|tkr$t|}nt|}dd|D}|D]0\}}t|t|jt|ddt|f||<q@t|dS)NcSsg|]}tt||fqSr)reprtype)r8xrrrr:sz%get_obj_referents..0)Zclip)rZ) r+r\rr Z named_refsZ _getreferentsr'rr )r&rZr%Z named_objectsrefsnameorrrget_obj_referentss      rbz/static/cCstj|tdS)zGet static files (CSS-files).r/)r static_file static_files)filenamerrrrdsrdcCstjdkrtt_tjS)zN Retrieve garbage graph objects from cache, compute if cache is cold. N)rrr Zsplit_and_sortrrrr_compute_garbage_graphss  rgz/garbage garbage_indexcCst}t|dS)zGet garbage overview.)Zgraphs)rgrrrrrrhsz/garbage/garbagecCs8tt|}||j}|jdddt||dS)zGet reference cycle details.cSs|j Sr)size)r]rrrzgarbage_cycle..)key)objectsr-)rgr(reduce_to_cyclesmetadatasortr)r-graphrnrrr garbage_cycles rsc Csfz |j}WnPtk rZz"|jtjtj|dd|}Wntk rTd}YnXYnX||_|S)zRetrieve or render a graph.Zpng)formatN) rendered_fileAttributeErrorrenderospathjoinrtmpdirOSError)rrrfrenderedrrr _get_graphs r~z/garbage/graph/cCsdtt|}tjjdd}|r*|}|s2dSd||f}t||}|r\tj|t j dSdSdS)z,Get graph representation of reference cycle.reduceNzgarbage%so%s.pngrc) rgr(rrequestGETr)ror~rdrr{)r-rrZ reduce_graphrfrurrr garbage_graphs  rz/helpcCstddS)z!Redirect to online documentation.z)https://pympler.readthedocs.io/en/latest/N)rrLrrrrshow_documentation src@seZdZdZddZdS) PymplerServerzSimple WSGI server.cCs t|j|j||_|jdSr)rhostportr serve_forever)rhandlerrrrrunszPymplerServer.runN)rrr r!rrrrrrsr localhostFcKsh|r|s|jt_n|t_z>tddt_tf||d|t_t|tjtjdW5ttjXdS)a Start the web server to show profiling data. The function suspends the Python application (the current thread) until the web server is stopped. The only way to stop the server is to signal the running thread, e.g. press Ctrl+C in the console. If this isn't feasible for your application use `start_in_background` instead. During the execution of the web server, profiling data is (lazily) cached to improve performance. For example, garbage graphs are rendered when the garbage profiling data is requested and are simply retransmitted upon later requests. The web server can display profiling data from previously taken snapshots when `tracker` or `stats` is specified. The former is useful for profiling a running application, the latter for off-line analysis. Requires existing snapshots taken with :py:meth:`~pympler.classtracker.ClassTracker.create_snapshot` or :py:meth:`~pympler.classtracker.ClassTracker.start_periodic_snapshots`. :param host: the host where the server shall run, default is localhost :param port: server listens on the specified port, default is 8090 to allow coexistance with common web applications :param tracker: `ClassTracker` instance, browse profiling data (on-line analysis) :param stats: `Stats` instance, analyze `ClassTracker` profiling dumps (useful for off-line analysis) pympler)prefix)rr)rN) rrrr{rrrdebugr)rrr4rrkwargsrrrstart_profilers   rcs*eZdZdZdfdd ZddZZS) ProfilerThreadz,Encapsulates a thread to run the web server.NPympler web frontendc s&tt|j|||d||_d|_dS)N)grouptargetr`T)superrrrdaemon)rrrr`r __class__rrrCs  zProfilerThread.__init__cCstf|jdSr)rrrrrrrKszProfilerThread.run)NNr)rrr r!rr __classcell__rrrrrAsrcKstf|}||S)a  Start the web server in the background. A new thread is created which serves the profiling interface without suspending the current application. For the documentation of the parameters see `start_profiler`. Returns the created thread object. )rstart)rthreadrrrstart_in_backgroundOs r)rrNNF)@r!rTrx threadinginspectrjsonrshutilrtempfilerrweakrefrZwsgiref.simple_serverrrr Zpympler.garbagegraphr Zpympler.processr r Zpympler.util.stringutilsr r ImportErrorZ pympler.utillocalrrr'r+rydirnameabspath__file__Z pympler_pathrzreZ TEMPLATE_PATHrFZrouteviewr/r0rIrJrMrYrbrdrgrhrsr~rrZ ServerAdapterrrrrrrrrs               %           +