o wvXj@sdZddlmZmZmZddlZddlZddlZddlm Z ddl m Z ddl m Z ddl mZmZmZddlmZzdd lmZWn eyMdZYnwd Zd ZdagZd d Ze ddZddZd(ddZd)ddZd*ddZd+ddZ ddZ!ddZ"d d!Z#d"d#Z$d,d&d'Z%dS)-z:Utilities for debugging memory usage, blocking calls, etc.)absolute_importprint_functionunicode_literalsN)contextmanager)partial)pprint) WhateverIOitemsrange)signals)Process)blockdetection sample_memmemdumpsample humanbytesmem_rsspscry))gpBTB)gAGB)g0AMB)g@KB)gbcCsddl}td||)Nrz$Blocking detection timed-out at: {0})inspect RuntimeErrorformat getframeinfo)signumframerr O/var/www/html/myproject/venv/lib/python3.10/site-packages/celery/utils/debug.py _on_blocking%s r"c 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)r r" arm_alarm reset_alarm)timeout old_handlerr r r!r .s   r cCst}t||S)z\Sample RSS memory usage. Statistics can then be output by calling :func:`memdump`. )r _mem_sampleappend) current_rssr r r!rDs r cCsLt}t||kr t|nt||}gtdd<ddl}|t}||fS)Nr)r(lenlistrgccollectr)samplesSprevr. after_collectr r r!_memdumpNs r4cCsdtt|d}tdur|ddSt|\}}|r)|d|D] }|d|q|d|dS)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 - > {0},z- rss (end): {0}.)rprintrr4r)r0r6sayr2r3memr r r!rXs   rc 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,r IndexError)xnkj_r r r!rks    rcCs"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)intr)frBir r r!hfloat|srFcstfddtDS)z4Convert bytes to human-readable form (e.g., KB, MB).c3s8|]\}}|krdt|r|n|VqdS)z{0}{1}N)rrF).0divunitsr r! s zhumanbytes..)nextUNITSrJr rJr!rs rcCs t}|durtt|jSdS)z.Return RSS memory usage as a humanized string.N)rr_process_memory_inforssrAr r r!rsrcCs tdurtdurttatS)zReturn the global :class:`psutil.Process` instance. Note: Returns :const:`None` if :pypi:`psutil` is not installed. N)_processr osgetpidr r r r!rs rcCs&z|WSty|YSw)N) memory_infoAttributeErrorget_memory_info)processr r r!rOs    rO=1c Csddl}|dur tn|}tt|d}dd|D}||}ttD]5\}}||} | s2q&|d | ||t j ||d|||d||t |j |d|d q&|S) zpReturn stack-trace of all active threads. See Also: Taken from https://gist.github.com/737056. rNr5cSsi|]}|j|qSr )ident)rGtr r r! szcry..z{0.name}zLOCAL VARIABLES)stream ) threadingrrr7 enumerater sys_current_framesgetr traceback print_stackrf_localsgetvalue) outsepchrseplenr_Ptmapseptidrthreadr r r!rs$   r)r+)r+N)r)r@)NrXrY)&__doc__ __future__rrrrRrard contextlibr functoolsrr celery.fiverr r celery.platformsr psutilr ImportError__all__rNrQr(r"r rr4rrrFrrrrOrr r r r!s@