o he@sdZddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z zddl mZWn ey?dZYnwd Zd ZdagZd d Zed dZddZd'ddZd(ddZd)ddZd*ddZddZddZdd Zd!d"Zd+d%d&ZdS),z:Utilities for debugging memory usage, blocking calls, etc.N)contextmanager)partial)pprint)signals) WhateverIO)Process)blockdetection sample_memmemdumpsample humanbytesmem_rsspscry))gpBTB)gAGB)g0AMB)g@KB)gbcCsddl}td||)Nrz!Blocking detection timed-out at: )inspect RuntimeError getframeinfo)signumframerrD/var/www/html/env/lib/python3.10/site-packages/celery/utils/debug.py _on_blocking"src csn|sdVdStd}|tkrdn|}ttd<zt|VW|r&|td<tdS|r2|td<tw)zqContext that raises an exception if process is blocking. Uses ``SIGALRM`` to detect blocking functions. NALRM)rr arm_alarm reset_alarm)timeout old_handlerrrrr)s   rcCst}t||S)z\Sample RSS memory usage. Statistics can then be output by calling :func:`memdump`. )r _mem_sampleappend) current_rssrrrr ?s r cCsLt}t||kr t|nt||}gtdd<ddl}|t}||fS)Nr)r"lenlistr gccollectr )samplesSprevr( after_collectrrr_memdumpIs r.cCshtt|d}tdur|ddSt|\}}|r*|d|D] }|d|dq|d|ddS) zDump memory statistics. Will print a sample of all RSS memory samples added by calling :func:`sample_mem`, and in addition print used RSS memory after :func:`gc.collect`. fileNz- rss: (psutil not installed).z- rss (sample):z- > ,z - rss (end): .)rprintrr.)r*r0sayr,r-memrrrr Ss   r c csNt||}t|D]}z||VWn tyYdSw||7}q dS)zGiven a list `x` a sample of length ``n`` of that list is returned. For example, if `n` is 10, and `x` has 100 items, a list of every tenth. item is returned. ``k`` can be used as offset. N)r&range IndexError)xnkj_rrrr fs    r cCs"t|}||kr |Sdj||dS)zConvert float to value suitable for humans. Arguments: f (float): The floating point number. p (int): Floating point precision (default is 5). z{0:.{p}}p)intformat)fr?irrrhfloatwsrDcstfddtDS)z4Convert bytes to human-readable form (e.g., KB, MB).c3s8|]\}}|krt|r|n|VqdSN)rD).0divunitsrr s zhumanbytes..)nextUNITSrIrrIrr s r cCs t}|durtt|jSdS)z.Return RSS memory usage as a humanized string.N)rr _process_memory_inforssr>rrrr sr cCs tdurtdurttatS)zReturn the global :class:`psutil.Process` instance. Note: Returns :const:`None` if :pypi:`psutil` is not installed. N)_processrosgetpidrrrrrs rcCs&z|WSty|YSwrE) memory_infoAttributeErrorget_memory_info)processrrrrNs    rN=1c Csddl}|dur tn|}tt|d}dd|D}||}tD]4\}}||} | s2q&|| j ||t j ||d|||d||t |j |d|dq&|S) zpReturn stack-trace of all active threads. See Also: Taken from https://gist.github.com/737056. rNr/cSsi|]}|j|qSr)ident)rFtrrr szcry..zLOCAL VARIABLES)stream ) threadingrrr3 enumeratesys_current_framesitemsgetname traceback print_stackrf_localsgetvalue) outsepchrseplenr^Ptmapseptidrthreadrrrrs$    r)r%)r%N)r)r=)NrWrX) __doc__rQr`re contextlibr functoolsrrcelery.platformsrcelery.utils.textrpsutilr ImportError__all__rMrPr"rrr r.r r rDr r rrNrrrrrs>