From e7e50039d7bebeb880d6a34032cbf36b11f96302 Mon Sep 17 00:00:00 2001
From: Griatch <griatch@gmail.com>
Date: Thu, 15 May 2014 21:27:30 +0200
Subject: [PATCH] Added deepsize function for size evaluation of object and its
 referends

---
 src/utils/idmapper/base.py |  1 -
 src/utils/utils.py         | 28 ++++++++++++++++++++++++++++
 2 files changed, 28 insertions(+), 1 deletion(-)

diff --git a/src/utils/idmapper/base.py b/src/utils/idmapper/base.py
index a421a89f5c..cae83e6992 100755
--- a/src/utils/idmapper/base.py
+++ b/src/utils/idmapper/base.py
@@ -378,4 +378,3 @@ def cache_size(mb=True):
     get_recurse(SharedMemoryModel.__subclasses__())
     sizedict["_total"] = tuple(sizedict["_total"])
     return sizedict
-
diff --git a/src/utils/utils.py b/src/utils/utils.py
index bc4a875410..5d640751d7 100644
--- a/src/utils/utils.py
+++ b/src/utils/utils.py
@@ -1029,6 +1029,34 @@ def get_evennia_pids():
         return int(server_pid), int(portal_pid)
     return None, None
 
+from gc import get_referents
+from sys import getsizeof
+def deepsize(obj, max_depth=4):
+    """
+    Get not only size of the given object, but also the
+    size of objects referenced by the object, down to
+    max_depth distance from the object.
+
+    Note that this measure is necessarily approximate
+    since some memory is shared between objects. The
+    max_depth of 4 is roughly tested to give reasonable
+    size information about database models and their handlers.
+
+    Returns size in Bytes
+    """
+    def _recurse(o, dct, depth):
+        if max_depth >= 0 and depth > max_depth:
+            return
+        for ref in get_referents(o):
+            idr = id(ref)
+            if not idr in dct:
+                dct[idr] = (ref, getsizeof(ref, default=0))
+                _recurse(ref, dct, depth+1)
+    sizedict = {}
+    _recurse(obj, sizedict, 0)
+    #count = len(sizedict) + 1
+    size = getsizeof(obj) + sum([p[1] for p in sizedict.values()])
+    return size
 
 # lazy load handlers