Yf7 6 d dl Z d dlZd dlZd dlmZ d dlmZ d dlm Z m
Z
m Z m Z d dl
mZmZmZ d dlmZmZ ddlmZ d d
lmZ dad j3 Zej7 d d
j7 dd
j7 dd Z d Z e G d de Z y) N)parse) get_ipython)Magicsline_cell_magic magics_class
no_var_expand)argumentmagic_argumentsparse_argstring)IFramedisplay )Profiler )PrePostAstTransformerz
To enable asyncio mode, use %%pyinstrument --async_mode=enabled
Note that due to IPython limitations this will run in a separate thread!
z`z
z
c t S )zt
Allows the code inserted into the cell to access the pyinstrument Profiler
instance, to start/stop it.
)_active_profiler X/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/pyinstrument/magic/magic.py_get_active_profilerr s
r c e Zd Z fdZ e ededd eddd ed
d d d
ededd ededd e e
dd Z d Z xZ
S )PyinstrumentMagicc t | | t d | _ t d | _ t | j | j | _ y )Nz[
from pyinstrument.magic.magic import _get_active_profiler; _get_active_profiler().start()
z
_get_active_profiler().stop())super__init__r prepostr _transformer)selfshell __class__s r r zPyinstrumentMagic.__init__# sE
k
;< 1$((DII Fr z
--intervalgMbP?zThe minimum time, in seconds, between each stack sample. See: https://pyinstrument.readthedocs.io/en/latest/reference.html#pyinstrument.Profiler.interval)typedefaulthelpz --async_modedisabledzConfigures how this Profiler tracks time in a program that uses async/await. See: https://pyinstrument.readthedocs.io/en/latest/reference.html#pyinstrument.Profiler.async_mode)r& r' z--heightz-hi z
Output heightz
--timelineFzShow output timeline viewcode*z.When used as a line magic, the code to profile)r% nargsr' c t | j | }t }|s t d |xs |}|syt r$t j
rt j
| j |j v r%|j j | j t |j |j a|j j | j |j dk( r|j | }n| j || }|j j | j |j dk( rV|j rJt# |j t r0dt% |j v rt' t( t* dd yt j- |j. }t1 d
t2 j4 j7 | z d |j8 d g
}t j; |j. } t' |j= | dd t j
rJ day)a$
Run a cell with the pyinstrument statistical profiler.
Converts the line/cell's AST to something like:
try:
profiler.start()
run_code
finally:
profiler.stop()
profiler.output_html()
z#couldn't get ipython shell instanceN)interval
async_moder( z event loop is already running)
text/plain text/htmlT)raw)timelinezdata:text/html, z100%zstyle="resize: vertical")srcwidthheightextras)r0 r/ ) r pyinstrumentr RuntimeErrorr
is_runningstopr! ast_transformersremover r- r. appendrun_cellrun_cell_async
error_in_exec
isinstancestrr
_ASYNCIO_TEXT_WARNING_ASYNCIO_HTML_WARNING output_htmlr2 r urllibr quoter5 output_text _repr_html_)
r" linecellargsipr) cell_resulthtml as_iframeas_texts
r r7 z PyinstrumentMagic.pyinstrument, s Z t 0 0$7
] DE E|t 0 ; ; ! ! # 3 3 3 & &t'8'8 9#T]]tW
""4#4#45 ??j ( ++d+K --b$7K
""4#4#45
OOz )));44lC/3{7P7P3QQ
"7!6
++T]]+C"V\\%7%7%==;;./
#..
.Fi 3 3 5WMSWX#.... r c & t j }t j } t j |j
j
t j | |j | }t j || }|j j |j |j t j | S # |j |j t j | w xY w)N)target)
asyncioget_event_loopnew_event_loop threadingThread run_foreverstartset_event_loopr? run_coroutine_threadsaferesultcall_soon_threadsafer: )r" rM r) old_looploopcorofutures r r? z PyinstrumentMagic.run_cell_async s ))+%%' - D$4$4 5 ; ; = " "4 ($$T*D 55dDAF==?)) % %dii 0 " "8 ,
% %dii 0 " "8 ,s BC 2D)N)__name__
__module__ __qualname__r r
r floatboolrB r r r7 r?
__classcell__)r$ s @r r r ! s G
i
(
= D
@D L-r r ) rT rW urllib.parserF astr IPythonr IPython.core.magicr r r r IPython.core.magic_argumentsr r
r IPython.displayr r
r _utilsr r striprD replacerC r r r r r rs s S S S S + )
EG
!!'3/77#FNNvW[\
A- A- A-r