o
    iw                     @   sx   d Z ddlmZ ddlZddlZddlZddlmZ zddl	m
Z
 W n ey1   ddlm
Z
 Y nw G dd deZdS )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.   e Zd ZdZejddddfddZdd	 ZdS )
ProfilerMiddlewarea  Wrap 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``.
    )timecalls Nz/{method}.{path}.{elapsed:.0f}ms.{time:.0f}.profc                 C   s(   || _ || _|| _|| _|| _|| _d S N)_app_stream_sort_by_restrictions_profile_dir_filename_format)selfappstreamsort_byrestrictionsprofile_dirfilename_formatr   r   W/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/werkzeug/middleware/profiler.py__init__I   s   	
zProfilerMiddleware.__init__c           
         s4  g dfdd	  fdd}t  }t }|| d}t | }jd urctjr:}njjd d	d
dd	pMd
|d t d}tjj|}|| jd urt|jd}	|	jj  tdjd tdddjd |	jj  tdjd |gS )Nc                    s   | ||  j S r	   )append)statusheadersexc_info)response_bodystart_responser   r   catching_start_response\   s   z<ProfilerMiddleware.__call__.<locals>.catching_start_responsec                     s0     } |  t| dr|   d S d S )Nclose)r
   extendhasattrr    )app_iter)r   environr   r   r   r   runapp`   s
   

z+ProfilerMiddleware.__call__.<locals>.runapp    REQUEST_METHOD	PATH_INFO/.rootg     @@)methodpathelapsedr   )r   zP--------------------------------------------------------------------------------)filez
PATH: {!r} zQ--------------------------------------------------------------------------------
r	   )r   r   runcalljoinr   callabler   formatgetstripreplaceosr-   
dump_statsr   r   
sort_statsr   printprint_statsr   )
r   r$   r   r%   profilestartbodyr.   filenamestatsr   )r   r$   r   r   r   r   __call__Y   s6   





zProfilerMiddleware.__call__)__name__
__module____qualname____doc__sysstdoutr   rB   r   r   r   r   r      s    0
r   )rF   
__future__r   os.pathr8   rG   r   pstatsr   cProfiler   ImportErrorr=   objectr   r   r   r   r   <module>   s    