Skip to content

Instantly share code, notes, and snippets.

@j1fig
Last active March 27, 2017 10:59
Show Gist options
  • Save j1fig/9599fade1333566d24c52d7347841238 to your computer and use it in GitHub Desktop.
Save j1fig/9599fade1333566d24c52d7347841238 to your computer and use it in GitHub Desktop.
recursively gets size of object, guaranteeing not counting same obj twice. literally copied from http://stackoverflow.com/a/30316760 but might contain further improvements.
import sys
from numbers import Number
from collections import Set, Mapping, deque
try: # Python 2
zero_depth_bases = (basestring, Number, xrange, bytearray)
iteritems = 'iteritems'
except NameError: # Python 3
zero_depth_bases = (str, bytes, Number, range, bytearray)
iteritems = 'items'
def getsize(obj_0):
"""Recursively iterate to sum size of object & members."""
def inner(obj, _seen_ids = set()):
obj_id = id(obj)
if obj_id in _seen_ids:
return 0
_seen_ids.add(obj_id)
size = sys.getsizeof(obj)
if isinstance(obj, zero_depth_bases):
pass # bypass remaining control flow and return
elif isinstance(obj, (tuple, list, Set, deque)):
size += sum(inner(i) for i in obj)
elif isinstance(obj, Mapping) or hasattr(obj, iteritems):
size += sum(inner(k) + inner(v) for k, v in getattr(obj, iteritems)())
# Check for custom object instances - may subclass above too
if hasattr(obj, '__dict__'):
size += inner(vars(obj))
if hasattr(obj, '__slots__'): # can have __slots__ with __dict__
size += sum(inner(getattr(obj, s)) for s in obj.__slots__ if hasattr(obj, s))
return size
return inner(obj_0)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment