a
DfX! @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
ddl m Z ddl m
Z
dd lmZ d d d Zd
d Zd ddZdd Zd ddZdg fddZed!ddZd"d d ZdS )# N)contextmanager)Profilewraps )config)escape )stateFc C s ddl m} ddlm} |||d}|r2| d }n&| d }| dd D ]}|||}qFz||}W n ty| d}Y n0 t|dfS ) Nr ) HTMLRenderer)Sessiontimelineshow_allr z5Rendering pyinstrument session failed
)Zpyinstrument.renderersr Zpyinstrument.sessionr combinerender Exceptionr ) sessionsr r r r rsessionsrendered r ]/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/panel/io/profile.pyrender_pyinstrument s
r c
C s ddl m} dd l}ddlm}m} ddlm} tj
tj |j dd}t
| }|| } W d n1 st0 Y ||d }
|dd D ]} |
| q| j| ||
||
dd d
d } t| d fS )
Nr )Stats)
json_stats
table_rows)TemplateZ templateszviz.htmlr )Z profile_namer Zcalleeszutf-8z/static/z/snakeviz/static/zbackground-color: white;)pstatsr snakevizZsnakeviz.statsr r Ztornado.templater ospathjoindirname__file__openreadaddgeneratedecodereplacer )
namer r r" r r r Z
SNAKEVIZ_PATHfZSNAKEVIZ_TEMPLATEr! r r r r r render_snakeviz s$
* r0 Ttreec C sl ddl m} ddlm} ddlm} ddlm} ddlm } |||| d
|}
|d } t } |
| | || j}
|r|
j|d ur|nd d
}n|
j|d ur|nd d
}d |
jji}|d v rt|
|d
< |
j|fi |}W d n1 s0 Y t }|dkr.|||
j|| n(|dkrJ|||
j| n |j|d |d | dfS )Nr )
FileReader)FlameGraphReporter)
StatsReporter)
TableReporter) TreeReporter)
flamegraphstatstabler1 r T)
merge_threadsZ
native_tracesr7 r9 Zmemory_recordsr7 r9 )filer ) memrayr2 Zmemray.reporters.flamegraphr3 Zmemray.reporters.statsr4 Zmemray.reporters.tabler5 Zmemray.reporters.treer6 gettempfileNamedTemporaryFilewriteflushr. Z get_leaked_allocation_recordsZ%get_high_watermark_allocation_recordsmetadataZhas_native_tracestupleZget_memory_snapshotsZ
from_snapshotioStringIOr seekr) )r. r Zshow_memory_leaksr: reporterr2 r3 r4 r5 r6 Z reporter_clsr nfreaderZsnapshotkwargsZ
reporter_instoutr r r
render_memray3 sP
$
rM c
K s ddl m}m} g }| D ]\\}}}|s.q |dkrt||fi |\}} |dddvrddlm}
|
dd d
d } | | | || f q nt
|}|d krt |fi |\}} n|d
krt ||\}} |d| d| ddd dd } | || f q |s| d|df |S )Nr )HTMLMarkdownr= rH r1 r; )TerminalZ stretch_bothr iX ) sizing_modemarginZ
min_height pyinstrumentr" zi r z No profiling output available)
ZpanerN rO itemsrM r> widgetsrP rA appendr r r0 )
Z profilersrK rN rO Zprofilesr$ enginer srcstylerP Ztermhtmlr r r get_profilesa s8
r[ c s fddt j D S )Nc s> i | ]6\\}}} r t |rr0t |s||f|qS r )rematch).0r. eZpsallowdenyr r
s z get_sessions..)r
_profilesrT r` r r` r get_sessions s re c s( ddl m}m}m}m} ddlm}m} |tt ddd fdd| j
d t j
d
krfd d } |d
ddj
| d |dddj
| d |dd}
nHt j
dkrfdd} |dg dddj
| d }
n
|dd}
||d|
fg dd dd d ddS ) Nr ) AccordionColumnRowTabs)CheckboxSelect)r Z
stretch_width)rR rQ c s" t t fi |d d < d S )N)r[ re )argsrK )ra rb tabsr r update_profiles s
z'profiling_tabs..update_profilesrd rS c s j j d d S )Nr
valuerm )r r ro r r update_pyinstrument s z+profiling_tabs..update_pyinstrumentzEnable timeline)rl r )r. rR rq zShow All)rQ r= c s j d d S )N)rH rp rr )rH ro r r
update_memray s z%profiling_tabs..update_memrayZReporter)r7 r9 r1 r1 )r. optionsrq ZConfigz#444444z#333333r )activeZactive_header_backgroundZheader_backgroundrQ rR )Zlayoutrf rg rh ri rU rj rk r[ re paramZwatchr Zprofiler) r
ra rb rf rg rh ri rj rk rs Z config_panelrt r )ra rb rH r rn r ro r profiling_tabs sL
rx rS c c s" | dkrNddl m} z| }| W q tyJ |dd}| Y q0 n\| dkrft }| nD| dkrddl}t d t
j }| |}|
n| du rg }|V | dkr|| nR| dkr| || n6| dkr |ddd |t|d
t| dS ) ap
A context manager which profiles the body of the with statement
with the supplied profiling engine and returns the profiling object
in a list.
Arguments
---------
engine: str
The profiling engine, e.g. 'pyinstrument', 'snakeviz' or 'memray'
Returns
-------
sessions: list
A list containing the profiling session.
rS r )Profilerdisabled)Z
async_moder" r= Nz/tmprb)rS ry start RuntimeErrorr enabler= r?
gettempdiruuiduuid4hexZTracker __enter__rV stopdisable__exit__r( r) r# remove)rW ry Zprofr= Ztmp_filetrackerr r r r profile_ctx s8
r c s$ t tstd fdd}|S )a
A decorator which may be added to any function to record profiling
output.
Arguments
---------
name: str
A unique name for the profiling session.
engine: str
The profiling engine, e.g. 'pyinstrument', 'snakeviz' or 'memray'
z Profiler must be given a name.c s t fdd}|S )Nc s| t jr t jt jv r | i |S t }| i |}W d n1 sL0 Y t j f |7 < t jd |S )Nrd )r
ZcurdocZ
_launchingr rd rw trigger)rm rK r ret)rW funcr. r r wrapped s
, z)profile..wrapper..wrappedr )r r rW r. )r r wrapper s zprofile..wrapper)
isinstancestr
ValueError)r. rW r r r r profile s
r )FF)TTr1 )NN)rS )rS )rE r# r\ r? r
contextlibr ZcProfiler functoolsr r utilr r
r r0 rM r[ re rx r r r r r r s&
.
3 /