&`i"1ddlZddlZddlZddlZddlZddlZddlmZddlmZddl m Z ej e Z dZdZdZde eeffd Zd efd Zd efd ZGd dZGddZdS)N)datetime)Path)Unionz#sudo chown root: `which {profiler}`z'sudo chown root:root `which {profiler}`aB Note that this command requires `{profiler}` to be installed with root permissions. You can install `{profiler}` and give it root permissions as follows: $ pip install {profiler} $ {set_chown_command} $ sudo chmod u+s `which {profiler}` Alternatively, you can start Ray with passwordless sudo / root permissions. stringcZt|tr|dS|S)Nzutf-8) isinstancebytesdecode)rs /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/dashboard/modules/reporter/profile_manager.pyr r s+&%  &}}W%%% Mreturnc ft|}d}d|vrbtjdkrt|nt |}t||}d|d|dt|d t|d S) N permissiondarwin)profiler)rset_chown_commandzFailed to execute `z`. z === stderr === z === stdout ===  )r lowersysplatformDARWIN_SET_CHOWN_CMDformatLINUX_SET_CHOWN_CMD"PROFILER_PERMISSIONS_ERROR_MESSAGE)cmdrstdoutstderr stderr_str extra_messagers r _format_failed_profiler_commandr!"sJMz''))))|x'' ! ' ' ' : : :$++X+>>  ;AA1BB   3           r cK tjdddtjtjd{V}|d{V\}}|jdkS#t $rYdSwxYw)Nsudo-ntruerrrF)asynciocreate_subprocess_exec subprocessPIPE communicate returncodeFileNotFoundError)process_s r _can_passwordless_sudor0>s '6   ??          ((********1!Q&& uus3A A-,A-c TeZdZdefdZd dededeeffdZ dded ededeeffd Z d S)CpuProfilingManagerprofile_dir_pathcrt||_|jdd|_dS)NTexist_okzpy-spyrr3mkdir profiler_nameselfr3s r __init__zCpuProfilingManager.__init__Os; $%5 6 6 ##T#222%r Fpidnativer cKtj|j}|dS|ddt|g}tjdkr|r|dtd{Vrddg|z}tj |tj tj d d{V}| d{V\}}|j d krd t||j||fSd t|fS) a Capture and dump a trace for a specified process. Args: pid: The process ID (PID) of the target process for trace capture. native (bool, optional): If True, includes native (C/C++) stack frames. Default is False. Returns: Tuple[bool, str]: A tuple containing a boolean indicating the success of the trace capture operation and a string with the trace data or an error message. NFz*Failed to execute: py-spy is not installeddump-plinux--nativer#r$r&rFT)shutilwhichr9strrrappendr0r'r(r)r*r+r,r!r )r;r=r>pyspyrr.rrs r trace_dumpzCpuProfilingManager.trace_dumpTs= T/00 =FFfdCHH- <7 " "v " JJz " " "')) ) ) ) ) ) ) '4.3&C6 ??          '2244444444   " "9T' ' 'r flamegraphdurationc Ktj|j}|dS|dvr dd|ddzfS|dkrd }nd }|j|d |d |z }|d d|dt |dt |d|g }t jdkr|r|dtd{Vrddg|z}tj |tj tj dd{V} | d{V\} } | jdkrdt||j| | fSdt!|dfS)a Perform CPU profiling on a specified process. Args: pid: The process ID (PID) of the target process to be profiled. format (str, optional): The format of the CPU profile output. Default is "flamegraph". duration (float, optional): The duration of the profiling session in seconds. Default is 5 seconds. native (bool, optional): If True, includes native (C/C++) stack frames. Default is False. Returns: Tuple[bool, str]: A tuple containing a boolean indicating the success of the profiling operation and a string with the profile data or an error message. Nr@)rKraw speedscopeFz"Failed to execute: Invalid format z, z%must be [flamegraph, raw, speedscope]rKsvgtxtr/z_cpu_profiling.record-orBz-d-frCrDr#r$r&rTrb)rErFr9r3rGrrrHr0r'r(r)r*r+r,r!openread) r;r=rrMr>rI extensionprofile_file_pathrr.rrs r cpu_profilezCpuProfilingManager.cpu_profileys( T/00 =FF < < <?V???9:  \ ! !III  !v$O$O$O$OI$O$O O       HH  MM     <7 " "v " JJz " " "')) ) ) ) ) ) ) '4.3&C6 ??          '2244444444   " "9T' /66;;=== =r NF)rKrLF) __name__ __module__ __qualname__rGr<intboolrJfloatr[r r r2r2Ns&&&&& #(#(C#(#(4+#(#(#(#(LRWA>A>A>7<A>JNA> A>A>A>A>A>A>r r2c eZdZdefdZ ddedededed eeff d Z dded ed ed ed eeff dZ dded ed eeffdZ dS)MemoryProfilingManagerr3cxt|dz |_|jdd|_dS)NmemrayTr5r7r:s r r<zMemoryProfilingManager.__init__s@ $%5 6 6 A ##T#222%r rKFr=profiler_filenamerleaksr cKtj|j}|dS|j|z }t |sdd|dfSt j|\}}|j|dz } |dkr|dd| d g} n|d kr|d d| d g} ndd |d fS|r| d | |tj | tj tj dd{V} | d{V\} } | jdkrdt!| |j| | fSdt#| dfS)a Convert the Memray profile result to specified format. Args: pid: The process ID (PID) associated with the profiling operation. profiler_filename: The filename of the profiler output to be processed. format (str, optional): The format of the profile result. Default is "flamegraph". leaks (bool, optional): If True, include memory leak information in the profile result. Returns: Tuple[bool, str]: A tuple containing a boolean indicating the success of the operation and a string with the processed profile result or an error message. NF*Failed to execute: memray is not installedFzFailed to execute: process z has not been profiledz.htmlrKrTrUtablez'Failed to execute: Report with format: z is not supportedz--leaksr&rTrV)rErFr9r3ris_fileospathsplitextrHr'r(r)r*r+r,r!rWrX)r;r=rhrrirgrZr9r/profile_visualize_path visualize_cmdr.rrs r get_profile_resultz)MemoryProfilingManager.get_profile_results0d011 >FF 14EE%&&..00 TSSSSS S7++,=>> q!%!6M9P9P9P!P \ ! !& MMw  & MMS&SSS   ,   + + +.///6 ??          '2244444444   " "9t166 T0$77<<>>>>r r>trace_python_allocatorsverbosecKtj|j}|dStjd}|d|d}|j|z }|dt|d|g} |r| d|r| d |r| d td{Vrd d g| z} tj | tj tj d d{V} | d{V\} } | jdkrddt!| |j| | fSd|d|fS)a Attach a Memray profiler to a specified process. Args: pid: The process ID (PID) of the target process which the profiler attached to. native (bool, optional): If True, includes native (C/C++) stack frames. Default is False. trace_python_allocators (bool, optional): If True, includes Python stack frames. Default is False. verbose (bool, optional): If True, enables verbose output. Default is False. Returns: Tuple[bool, str]: A tuple containing a boolean indicating the success of the operation and a string of a success message or an error message. N)FNrlz %Y%m%d%H%M%S_memory_profiling_z.binattachrTrDz--trace-python-allocators --verboser#r$r&rFTz$Success attaching memray to process )rErFr9rnowstrftimer3rGrHr0r'r(r)r*r+r,r!) r;r=r>rurvrg timestamprhrZrr.rrs r attach_profilerz&MemoryProfilingManager.attach_profiler s0d011 >LLLNN++N;; "EEiEEE 14EExS41BC  # JJz " " " " 4 JJ2 3 3 3  $ JJ{ # # #')) ) ) ) ) ) ) '4.3&C6 ??          '2244444444   " "/+VV !FFx   $ JJ{ # # # 3s88')) ) ) ) ) ) ) '4.3&C6 ??          '2244444444   " "9T' G#GGG Gr N)rKF)FFFr\) r]r^r_rGr<r`rartr~rrcr r reres0&&&&&# G?G? G?G? G?  G?  G?G?G?G?X(- >> >>"& >  >  >>>>F)H)H )H)H  )H)H)H)H)H)Hr re)r'loggingrorEr)rrpathlibrtypingr getLoggerr]loggerrrrrGr r r!rar0r2rercr r rsk    8 $ $<? &"5e$ c8 'd ' ' ' ' l>l>l>l>l>l>l>l>^xHxHxHxHxHxHxHxHxHxHr