U de@sdZddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z zddl mZWnek rdZYnXd 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))gpBZTB)gAZGB)g0AZMB)g@ZKB)gbcCs ddl}td||dS)Nrz!Blocking detection timed-out at: )inspect RuntimeError getframeinfo)signumframerr6/tmp/pip-unpacked-wheel-mu1yl971/celery/utils/debug.py _on_blocking"src csX|s dVnHtd}|tkr dn|}ttd<zt|VW5|rJ|td<tXdS)zqContext that raises an exception if process is blocking. Uses ``SIGALRM`` to detect blocking functions. NZALRM)rrZ reset_alarmZ arm_alarm)timeoutZ old_handlerrrrr)srcCst}t||S)z\Sample RSS memory usage. Statistics can then be output by calling :func:`memdump`. )r _mem_sampleappend)Z current_rssrrrr ?s r cCsLt}t||krt|nt||}gtdd<ddl}|t}||fS)Nr)rlenlistr gcZcollectr )samplesSprevr after_collectrrr_memdumpIs r$cCshtt|d}tdkr"|ddSt|\}}|rT|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 r&Zsayr"r#Zmemrrrr Ss   r c csPt||}t|D]6}z||VWntk r@YqLYnX||7}qdS)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)rrange 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)fr3irrrhfloatwsr8cstfddtDS)z4Convert bytes to human-readable form (e.g., KB, MB).c3s6|].\}}|krt|r |n|VqdSN)r8).0divunitsrr szhumanbytes..)nextUNITSr=rr=rr s r cCs t}|dk rtt|jSdS)z.Return RSS memory usage as a humanized string.N)rr _process_memory_infoZrssr2rrrr sr cCs tdkrtdk rttatS)zReturn the global :class:`psutil.Process` instance. Note: Returns :const:`None` if :pypi:`psutil` is not installed. N)_processrosgetpidrrrrrs rcCs,z |WStk r&|YSXdSr9)Z memory_infoAttributeErrorZget_memory_info)processrrrrBs rB=1c Csddl}|dkrtn|}tt|d}dd|D}||}tD]h\}}||} | sdqL|| j ||t j ||d|||d||t |j |d|dqL|S) zpReturn stack-trace of all active threads. See Also: Taken from https://gist.github.com/737056. rNr%cSsi|] }|j|qSr)ident)r:trrr szcry..zLOCAL VARIABLES)stream ) threadingrrr) enumeratesys_current_framesitemsgetname traceback print_stackrf_localsgetvalue) outZsepchrZseplenrOPZtmapseptidrthreadrrrrs$    r)r)rN)r)r1)NrHrI) __doc__rDrQrV contextlibr functoolsrrZcelery.platformsrZcelery.utils.textrZpsutilr ImportError__all__rArCrrrr r$r r r8r r rrBrrrrrs: