_w @sdZddlmZddlZddlZddlZddlmZyddl m Z Wn"e k rddl m Z YnXGddde ZdS)a9 Application Profiler ==================== This module provides a middleware that profiles each request with the :mod:`cProfile` module. This can help identify bottlenecks in your code that may be slowing down your application. .. autoclass:: ProfilerMiddleware :copyright: 2007 Pallets :license: BSD-3-Clause )print_functionN)Stats)Profilec@s@eZdZdZejd fddddZdd ZdS) ProfilerMiddlewareaWrap a WSGI application and profile the execution of each request. Responses are buffered so that timings are more exact. If ``stream`` is given, :class:`pstats.Stats` are written to it after each request. If ``profile_dir`` is given, :mod:`cProfile` data files are saved to that directory, one file per request. The filename can be customized by passing ``filename_format``. If it is a string, it will be formatted using :meth:`str.format` with the following fields available: - ``{method}`` - The request method; GET, POST, etc. - ``{path}`` - The request path or 'root' should one not exist. - ``{elapsed}`` - The elapsed time of the request. - ``{time}`` - The time of the request. If it is a callable, it will be called with the WSGI ``environ`` dict and should return a filename. :param app: The WSGI application to wrap. :param stream: Write stats to this stream. Disable with ``None``. :param sort_by: A tuple of columns to sort stats by. See :meth:`pstats.Stats.sort_stats`. :param restrictions: A tuple of restrictions to filter stats by. See :meth:`pstats.Stats.print_stats`. :param profile_dir: Save profile data files to this directory. :param filename_format: Format string for profile data file names, or a callable returning a name. See explanation above. .. code-block:: python from werkzeug.middleware.profiler import ProfilerMiddleware app = ProfilerMiddleware(app) .. versionchanged:: 0.15 Stats are written even if ``profile_dir`` is given, and can be disable by passing ``stream=None``. .. versionadded:: 0.15 Added ``filename_format``. .. versionadded:: 0.9 Added ``restrictions`` and ``profile_dir``. timecallsNz/{method}.{path}.{elapsed:.0f}ms.{time:.0f}.profcCs:||_||_||_||_||_||_dS)N)_app_stream_sort_by _restrictions _profile_dir_filename_format)selfZappstreamZsort_byZ restrictionsZ profile_dirZfilename_formatr@/tmp/pip-build-5gj8f0j9/Werkzeug/werkzeug/middleware/profiler.py__init__Is     zProfilerMiddleware.__init__c sgdfddfdd}t}tj}|j|dj}tj|}jdk r(tjrj}nYjjdddjd j d j d d pd d |ddtj}t j jj|}|j |jdk rt|dj} | jjtdddjtdjjd ddj| jjtddddj|gS)Ncs|||jS)N)append)statusheadersexc_info) response_bodystart_responserrcatching_start_response\sz.catching_start_responsecs<j}j|t|dr8|jdS)Nclose)rextendhasattrr)Zapp_iter)renvironrrrrrunapp`s z+ProfilerMiddleware.__call__..runappmethodREQUEST_METHODpathZ PATH_INFO/.rootelapsedg@@rr-Pfilez PATH: {!r} )rrZruncalljoinr callabler formatgetstripreplaceosr"Z dump_statsr rZ sort_statsr printZ print_statsr ) rrrrprofilestartbodyr&filenamestatsr)rrrrrr__call__Ys4     *  (zProfilerMiddleware.__call__)rr)__name__ __module__ __qualname____doc__sysstdoutrr9rrrrrs , r)r= __future__ros.pathr2r>rZpstatsrZcProfiler ImportErrorr4objectrrrrr s