1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
import types
import io
import memprof
from .mp_utils import *
if PY3:
builtin = (int,float,str,complex)
else:
builtin = (int,float,str,long,complex)
notinteresting = (types.ModuleType,types.FunctionType,types.LambdaType,io.IOBase,type(None),types.MethodType,types.GetSetDescriptorType,types.GeneratorType,types.BuiltinFunctionType,types.BuiltinMethodType,builtin)
def isInteresting(x):
if isinstance(x,notinteresting) or isinstance(x,memprof.MemProf):
return False
return True
def getSize(x):
ids = set()
def size_of(x):
if id(x) in ids:
return 0
ids.add(id(x))
if hasattr(x,"nbytes"):
nbytes = x.nbytes
return nbytes if isinstance(nbytes,int) else 0
size = sys.getsizeof(x)
if isinstance(x,builtin):
return size
items = []
# Go through objects (avoiding modules, MemProf, functions and methods thanks to isInteresting)
if hasattr(x,"__dict__"):
items = x.__dict__.values()
elif isinstance(x,dict):
items = x.values()
# Go through iterables skipping strings (and files, thanks to isInteresting)
elif hasattr(x, '__iter__'):
items = x
for it in items:
if not isinstance(it,builtin) and isInteresting(it):
size += size_of(it)
return size
return size_of(x)
|
