You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
4 years ago
|
import sys
|
||
|
|
from numbers import Number
|
||
|
|
from collections import Set, Mapping, deque
|
||
|
|
|
||
|
|
zero_depth_bases = (str, bytes, Number, range, bytearray)
|
||
|
|
iteritems = 'items'
|
||
|
|
|
||
|
|
|
||
|
|
def getsize(obj_0):
|
||
|
|
"""Recursively iterate to sum size of object & members."""
|
||
|
|
_seen_ids = set()
|
||
|
|
|
||
|
|
def inner(obj):
|
||
|
|
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)
|