ricrogz · October 14, 2017 23:14 · Oct 14, 2017 · Jul 21, 2016 · Jul 21, 2016 · Jul 21, 2016
diff --git a/pysize.py → py_real_obj_size.py b/pysize.py → py_real_obj_size.py
diff --git a/pysize.py b/pysize.py
@@ -16,6 +16,6 @@ def get_size(obj, seen=None):
         size += sum([get_size(k, seen) for k in obj.keys()])
     elif hasattr(obj, '__dict__'):
         size += get_size(obj.__dict__, seen)
-    elif hasattr(obj, '__iter__'):
+    elif hasattr(obj, '__iter__') and not isinstance(obj, (str, bytes, bytearray)):
         size += sum([get_size(i, seen) for i in obj])
     return size
diff --git a/pysize → pysize.py b/pysize → pysize.py
diff --git a/pysize b/pysize
@@ -0,0 +1,21 @@
+import sys
+
+def get_size(obj, seen=None):
+    """Recursively finds size of objects"""
+    size = sys.getsizeof(obj)
+    if seen is None:
+        seen = set()
+    obj_id = id(obj)
+    if obj_id in seen:
+        return 0
+    # Important mark as seen *before* entering recursion to gracefully handle
+    # self-referential objects
+    seen.add(obj_id)
+    if isinstance(obj, dict):
+        size += sum([get_size(v, seen) for v in obj.values()])
+        size += sum([get_size(k, seen) for k in obj.keys()])
+    elif hasattr(obj, '__dict__'):
+        size += get_size(obj.__dict__, seen)
+    elif hasattr(obj, '__iter__'):
+        size += sum([get_size(i, seen) for i in obj])
+    return size
No results found