o PDi*@sddlZddlZddlZddlZddlmZedZddZddZ ddd Z d d Z d d Z ddZ ddZddZddZddZddZdS)N) silk_profilezsilk.profiling.dynamiccCshd|vr(|d}d|dd}|d}t|tt|gd}t||}|St|ttgd}|S)zY Given a module name in form 'path.to.module' return module object for 'module'. .Nr)splitjoin __import__globalslocalsgetattr) module_namespltimpfrmmodulerO/var/www/Datamplify/venv/lib/python3.10/site-packages/silk/profiling/dynamic.py _get_module s  rcCsPd}d}d|vr|d\}}|rt||}t||}||fSt||}||fS)z Given a module and a function name, return the function. func_name can be of the forms: - 'foo': return a function - 'Class.foo': return a method Nr)rr )r func_namecls_nameclsfuncrrr _get_funcs   rcCsft|tr t|}t|dd}|}t||\}}||}|r+t||dd|dSt|||dS)z Programmatically apply a decorator to a function in a given module [+ class] @param module: module object or module name in form 'path.to.module' @param func: function object or function name in form 'foo' or 'Class.method' T)_dynamicrrN) isinstancestrrrrsetattrr)rrname decoratorrrwrapped_targetrrrprofile_function_or_method.s  rc Cs\tj}|jd}t|dkr,|ddD]}zt||}Wqty+||}Yqw|S)Nrr)sysmodules__name__rlenr AttributeError)rparentr r rrr_get_parent_moduleAs    r'c Csd|}d}t|Z}d}t|D]J\}} ||krBtd| } z| d}Wn ty2d}Ynw|||7}||d| 7}q||krL|krWnn ||d| 7}q|| 7}qWd|S1sgwY|S)N(with silk_profile('%s', _dynamic=True): 1from silk.profiling.profiler import silk_profile ^(\s+).*$rz )open enumerateresearchgroups IndexError) end_line file_pathr start_line inject_codecodefwsilinexrrr_get_context_manager_sourceMs,      r<cCs8td|}z |d}W|Styd}Y|Sw)z8 Return whitespace at the beginning of a string r+rr*)r.r/r0r%)txtmfwsrrr_get_wscs  r@cCsHt|d}t|d}tdt|D] }|||dd||<q|S)Nrr*r )inspectgetsourcelinesr@ranger$replace)rsourcer?r9rrr_get_source_linesos  rFc Csjd|}tt}|dd}i}|j}|j}|}||t d|t |||||j S)z Create new function defined in source but maintain context from func @param func: The function whose global + local context we will use @param source: Python source code containing def statement r*rzNew src_str: %s) rrAgetouterframes currentframe __globals__f_localscopyupdateLoggerdebugexecr#) rErsrc_strframes calling_framecontextrr combinedrrr_new_func_from_sourcews     rVc Cst|}|d7}|d7}t||}t||D]}z d||||<Wqty3td|j|fw|||d||d|d|t||S)a injects a context manager into the given function e.g given: x = 5 def foo(): print x print '1' print '2' print '3' inject_context_manager_func(foo, 0, 2, 'cm') foo will now have the definition: def foo(): with silk_profile('cm'): print x print '1' print '2' print '3' closures, globals & locals are honoured @param func: object of type or type @param start_line: line at which to inject 'with' statement. line num. is relative to the func, not the module. @param end_line: line at which to exit the context @param name: name of the profiler r z z!Function %s does not have line %dr)r()rFr@rCr1r#insertrV)rr4r2rrEr8r9rrr_inject_context_manager_funcs   rXcCs t|tSN)rr)orrr is_str_typs r[cCsht|rt|}d}t|r|}t||\}}n|j}t||||}|r,t|||dSt|||dSrY)r[rrr#rXr)rrr4r2rrrnew_funcrrrinject_context_manager_funcsr]rY)rAloggingr.r!silk.profiling.profilerr getLoggerrNrrrr'r<r@rFrVrXr[r]rrrrs"     "-