U 祡ca#@srddlmZmZmZmZmZmZmZddlZddl m Z ddl m Z ddl mZmZddlmZddlmZeedd d Zd+eeeed ddZeeedddZeeeeeeeefdddZeeeedddZd,eeeeeeeedddZd-eeeedddZeeeed d!d"Zeed#d$d%Z eeeedd&d'Z!dd(d)d*Z"dS).)AnyCallableDictListOptionalSetTupleN)summary)compat)isframestack getsizeof)_Py_TPFLAGS_HAVE_GC)objreturncCs&z t|WStk r YdSXdS)NT)r ReferenceError)rr1/tmp/pip-unpacked-wheel-8ad_c8mj/pympler/muppy.py ignore_objects rTF) remove_dupsinclude_framesrcCstt}dd|D}g}|D]*}t|}|D]}t|s6||q6q&|||rht|}|rtddD]}||dqz|S)a9Return a list of all known objects excluding frame objects. If (outer) frame objects shall be included, pass `include_frames=True`. In order to prevent building reference cycles, the current frame object (of the caller of get_objects) is ignored. This will not prevent creating reference cycles if the object list is passed up the call-stack. Therefore, frame objects are not included by default. Keyword arguments: remove_dups -- if True, all duplicate objects will be removed. include_frames -- if True, includes frame objects. cSsg|]}t|s|qSr)r.0orrr *szget_objects..Nr) gcZcollect get_objects get_referentsZ is_trackedappendextend_remove_duplicatesr )rrtmpresrrefsrefZsfrrrrs   robjectsrc CsTd}|D]F}z|t|7}Wqtk rLtdtt|t|fYqXq|S)z2Compute the total size of all elements in objects.rzIGNORING: type=%s; o=%s)rAttributeErrorprintstrtype)r(r$rrrrget_size@s$r-)leftrightrcCsggd}ttttttfddd}ttttttfttddd}||}||}||||d<||||d <|S) zGet the difference of both lists. The result will be a dict with this form {'+': [], '-': []}. Items listed in '+' exist only in the right list, items listed in '-' exist only in the left list. )+-r'cSs<i}|D].}t|}t||kr(g||<|||q|S)z!Partition the passed object list.)r,r )r(r$rtrrr partitionUs zget_diff..partition)foobarrcSsLg}|D]>}tt||s(||qt||t|s||q|S)zCompare objects from foo with objects defined in the values of bar (set of partitions). Returns a list of all objects included in list, but not dict values. )r Zobject_in_listr,r )r4r5r$rrrrget_not_included_s  z"get_diff..get_not_includedr0r1)rrrr,)r.r/r$r3r6Z left_objectsZ right_objectsrrrget_diffKs   r7cCst|td}|S)zSort objects by size in bytes.)key)sortedr)r(rrrsortws r:)r(Typeminmaxrcsng}krdkrtddk r6fdd|D}dkrPfdd|D}dkrjfdd|D}|S)zFilter objects. The filter can be by type, minimum size, and/or maximum size. Keyword arguments: Type -- object type to filter by min -- minimum object size max -- maximum object size r;z$minimum must be smaller than maximumNcsg|]}t|r|qSr) isinstancer)r<rrrs zfilter..csg|]}t|kr|qSrr r)r=rrrs csg|]}t|kr|qSrr r)r>rrrs ) ValueError)r(r<r=r>r$r)r<r>r=rfilter}s rA)objectlevelrcCs@t|}|d8}|dkr4|D]}|t||qt|}|S)a@Get all referents of an object up to a certain level. The referents will not be returned in a specific order and will not contain duplicate objects. Duplicate objects will be removed. Keyword arguments: level -- level of indirection to which referents considered. This function is recursive. rBr)rrr!r")rCrDr$rrrrrs rfunctionargsrcsd}tttdddtttdfdd }dddd }||}||}||f|}||f|}t||}t|}t|d kr|}|S) a7Test if more memory is used after the function has been called. The function will be invoked twice and only the second measurement will be considered. Thus, memory used in initialisation (e.g. loading modules) will not be included in the result. The goal is to identify memory leaks caused by functions which use more and more memory. Any arguments next to the function will be passed on to the function on invocation. Note that this function is currently experimental, because it is not tested thoroughly and performs poorly. NrEcWs(tt}||tt}||fS)z}Get a 2-tuple containing one summary from before, and one summary from after the function has been invoked. )r summarizer)rFrGs_befores_afterrrr_get_summariess  z"_get_usage.._get_summariesc s|f|\}}g}||kr(|||D]F}tt|dkrL|||D] }tt|dkrP||qPq,|D]}t||}qxt||}t|S)zGet the usage of a function call. This function is to be used only internally. The 'real' get_usage function is a wrapper around _get_usage, but the workload is done here. r)r lenrZ get_referrersr Z _subtractr7_sweep) rFrGrIrJignorerowitemrr$rKrr _get_usages   z_get_usage.._get_usagercSsdS)Nrrrrrnoopsz_get_usage..noopr)rrrrr r7rMrL)rFrGr$rRrToffsetr#rrQrrRs    rR)rrcCsttt|ddt@S)z(Is the passed object a container object. __flags__r)boolgetattrr,r)rrrr_is_containerobjectsrYcCs>t}g}|D]*}t|}||kr$q||||q|S)zaRemove duplicate objects. Inspired by http://www.peterbe.com/plog/uniqifiers-benchmark )setidaddr )r(seenresultrPmarkerrrrr"s  r"rScCstttdS)z%Print a summary of all known objects.N)r print_rHrrrrr print_summarysra)TF)Nr;r;)rB)#typingrrrrrrrrZpymplerr Z pympler.utilr inspectr r sysrZpympler.asizeofrrWrrintr-r+r7r:r,rArrRrYr"rarrrrs0$     * &, P