Yfe-ddlmZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZddlmZmZddlmZdd lmZmZmZmZdd lmZdd lmZmZGd d ZedZGddZ y)) annotationsN)Path) process_time)IOAny) renderers)AWAIT_FRAME_IDENTIFIEROUT_OF_CONTEXT_FRAME_IDENTIFIER)Session) AsyncState StackSamplerbuild_call_stackget_stack_sampler) LiteralStr)file_supports_colorfile_supports_unicodec0eZdZUded< ddZy)ActiveProfilerSessionzlist[tuple[list[str], float]] frame_recordsc<||_||_||_g|_yN) start_timestart_process_timestart_call_stackr)selfrrrs U/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/pyinstrument/profiler.py__init__zActiveProfilerSession.__init__s# %"4 0N)rfloatrrr list[str]returnNone)__name__ __module__ __qualname____annotations__rrrrrs300    "  $    rr)enableddisabledstrictc`eZdZUdZded<ded<ded<ded <dd d Zed!d Zed"d Zed#d Z d$d%dZ d&dZ edZ dZ dZd'dZ d(dZej$fddddd d)dZ d* d+dZd,d-dZ d, d.dZd/d0dZd1dZd&dZy)2ProfilerzB The profiler - this is the main way to use pyinstrument. Session | None _last_sessionzActiveProfilerSession | None_active_sessionr _interval AsyncMode _async_modec<||_d|_d|_||_y)z Note the profiling will not start until :func:`start` is called. :param interval: See :attr:`interval`. :param async_mode: See :attr:`async_mode`. N)r0r.r/r2)rinterval async_modes rrzProfiler.__init__2s#"!#%rc|jS)z The minimum time, in seconds, between each stack sample. This translates into the resolution of the sampling. )r0rs rr4zProfiler.interval>s ~~rc|jS)a Configures how this Profiler tracks time in a program that uses async/await. ``enabled`` When this profiler sees an ``await``, time is logged in the function that awaited, rather than observing other coroutines or the event loop. ``disabled`` This profiler doesn't attempt to track ``await``. In a program that uses async/await, this will interleave other coroutines and event loop machinery in the profile. Use this option if async support is causing issues in your use case, or if you want to run multiple profilers at once. ``strict`` Instructs the profiler to only profile the current `async context `_. Frames that are observed in an other context are ignored, tracked instead as ````. )r2r7s rr5zProfiler.async_modeFs0rc|jS)z@ The previous session recorded by the Profiler. )r.r7s r last_sessionzProfiler.last_session`s !!!rNc L|tjj} tt jt t |dd|_|jdk7}tj|j|j|y#d|_xYw)a Instructs the profiler to start - to begin observing the program's execution and recording frames. The normal way to invoke ``start()`` is with a new instance, but you can restart a Profiler that was previously running, too. The sessions are combined. :param caller_frame: Set this to override the default behaviour of treating the caller of ``start()`` as the 'start_call_stack' - the instigator of the profile. Most renderers will trim the 'root' from the call stack up to this frame, to present a simpler output. You might want to set this to ``inspect.currentframe().f_back`` if you are writing a library that wraps pyinstrument. Ninitial)rrrr)) inspect currentframef_backrtimerrr/r5r subscribe_sampler_saw_call_stackr4)r caller_frameuse_async_contexts rstartzProfiler.startgs"  "//188L #899;#/>!1, 4!P$D !%: =    ) ),,dmm=N  #'D s A6B B#c |js td tj|jt|jjz }t|jj|jjtj|jjz t|jjdjtj |jj"|}d|_|j$ tj&|j$|}||_|S#t j $r tdwxYw)z Stops the profiler observing, and sets :attr:`last_session` to the captured session. :return: The captured session. z'This profiler is not currently running.zUFailed to stop profiling. Make sure that you start/stop profiling on the same thread. )rrduration sample_countprogramrcpu_timeN)r/ RuntimeErrorr unsubscriberBr SubscriberNotFoundrrr rrr@lenjoinsysargvrr:combiner.)rrKsessions rstopz Profiler.stops'##HI I    + +D,H,H I  >D$8$8$K$KK..<<++66YY[4#7#7#B#BBT11??@HHSXX&!11BB  $    (ood&7&7AG$1.. g  s #E E)c|jduS)zd Returns `True` if this profiler is running - i.e. observing the program execution. N)r/r7s r is_runningzProfiler.is_runnings ##4//rcJ|jr|jd|_y)zC Resets the Profiler, clearing the `last_session`. N)rWrUr.r7s rresetzProfiler.resets ?? IIK!rcb|jtjj|S)aE Context manager support. Profilers can be used in `with` blocks! See this example: .. code-block:: python with Profiler() as p: # your code here... do_some_work() # profiling has ended. let's print the output. p.print() )rC)rEr=r>r?r7s r __enter__zProfiler.__enter__s&  4 4 6 = = > rc$|jyr)rU)rargss r__exit__zProfiler.__exit__s  rc|js td|rY|jdk(rJ|jdvr<|j}|jj j |tgz|fy|rZ|jdk(rK|jdk(r<|j}|jj j |tgz|fy|jj j ||fy)NzReceived a call stack without an active session. Please file an issue on pyinstrument Github describing how you made this happen!out_of_context_awaited)r(r*out_of_context_unknownr*) r/rLstater2inforappendr r )r call_stacktime_since_last_sample async_stateawaiting_coroutine_stackcontext_exit_frames rrBz Profiler._sampler_saw_call_stacks ##T  !!%==  $99'2'7'7 $  . . 5 5,0F/GG*  !!%==  H,!,!1!1   . . 5 5&*I)JJ*   . . 5 5zCY6Z [rFunicodecolorshow_alltimelinecx| t|}| t|}t|j|||||y)aprint(file=sys.stdout, *, unicode=None, color=None, show_all=False, timeline=False) Print the captured profile to the console. :param file: the IO stream to write to. Could be a file descriptor or sys.stdout, sys.stderr. Defaults to sys.stdout. :param unicode: Override unicode support detection. :param color: Override ANSI color support detection. :param show_all: Sets the ``show_all`` parameter on the renderer. :param timeline: Sets the ``timeline`` parameter on the renderer. Nrj)file)rrprint output_text)rrprkrlrmrns rrqzProfiler.printsN& ?+D1G ='-E   !!    rcT|jtj||||S)z\ Return the profile output as text, as rendered by :class:`ConsoleRenderer` rjrenderer)outputrConsoleRenderer)rrkrlrmrns rrrzProfiler.output_texts1{{..ux(  rcP|jtj||S)zY Return the profile output as HTML, as rendered by :class:`HTMLRenderer` rnrmrt)rvr HTMLRenderer)rrnrms r output_htmlzProfiler.output_html/s#{{I$:$:HW_$`{aarct|}|j|jtj||dy)zc Writes the profile output as HTML to a file, as rendered by :class:`HTMLRenderer` ryrtzutf-8)encodingN)r write_textrvrrz)rpathrnrmrps r write_htmlzProfiler.write_html5s= Dz  KK!7!7T\!]K ^  rcl|j}tj|j|S)zE Opens the last profile session in your web browser. )rn)_get_last_session_or_failrrzopen_in_browser)rrnrTs rrzProfiler.open_in_browserAs.002%%x8HHQQrcD|j}|j|S)z Returns the last profile session, as rendered by ``renderer``. :param renderer: The renderer to use. )rrender)rrurTs rrvzProfiler.outputIs! 002w''rcv|jr td|j td|jS)NzBcan't render profile output because this profiler is still runningzYcan't render profile output because this profiler has not completed a profile session yet)rW Exceptionr:r7s rrz"Profiler._get_last_session_or_failSs? ??`a a    $k    r)gMbP?r()r4rr5r1)r!r)r!r1)r!r-r)rCztypes.FrameType | None)r!r )r]r)rer rfrrgzAsyncState | None) rpzIO[str]rk bool | Nonerlrrmboolrnr)FFFF) rkrrlrrmrrnrr!str)FF)rnrrmrr!r)rzstr | os.PathLike[str]rnrrmr)F)rnr)ruzrenderers.Rendererr!r)r#r$r%__doc__r&rpropertyr4r5r:rErUrWrYr[r^rBrQstdoutrqrrr{rrrvrr'rrr,r,(s"!11 &  2"" !F$L00 "$%\%\!&%\' %\R    $!            H         bV[  *  6:  NR  R( !rr,)! __future__rr=osrQr@typespathlibrrtypingrr pyinstrumentrpyinstrument.framer r pyinstrument.sessionr pyinstrument.stack_samplerr r rrpyinstrument.typingrpyinstrument.utilrrrr1r,r'rrrsX" "V(dd*H    6 7 t!t!r