Merge pull request #139 from sr-murthy/index-repr

Add index size and custom `__repr__` methods for index classes

rtree/index.py

@@ -147,7 +147,7 @@ def __init__(self, *args, **kwargs):
 
             >>> idx = index.Index(properties=p)
             >>> idx  # doctest: +ELLIPSIS
-            <rtree.index.Index object at 0x...>
+            rtree.index.Index(bounds=[1.7976931348623157e+308, 1.7976931348623157e+308, -1.7976931348623157e+308, -1.7976931348623157e+308], size=0)
 
         Insert an item into the index::
 
@@ -284,6 +284,15 @@ def __init__(self, *args, **kwargs):
                 for item in stream:
                     self.insert(*item)
 
+    def get_size(self):
+        try:
+            return self.count(self.bounds)
+        except core.RTreeError:
+            return 0
+
+    def __repr__(self):
+        return 'rtree.index.Index(bounds={}, size={})'.format(self.bounds, self.get_size())
+
     def __getstate__(self):
         state = self.__dict__.copy()
         del state["handle"]
@@ -1821,7 +1830,7 @@ def __init__(self, *args, **kwargs):
 
             >>> idx = index.RtreeContainer(properties=p)
             >>> idx  # doctest: +ELLIPSIS
-            <rtree.index.RtreeContainer object at 0x...>
+            rtree.index.RtreeContainer(bounds=[1.7976931348623157e+308, 1.7976931348623157e+308, -1.7976931348623157e+308, -1.7976931348623157e+308], size=0)
 
         Insert an item into the index::
 
@@ -1847,6 +1856,15 @@ def __init__(self, *args, **kwargs):
         self._objects = {}
         return super(RtreeContainer, self).__init__(*args, **kwargs)
 
+    def get_size(self):
+        try:
+            return self.count(self.bounds)
+        except core.RTreeError:
+            return 0
+
+    def __repr__(self):
+        return 'rtree.index.RtreeContainer(bounds={}, size={})'.format(self.bounds, self.get_size())
+
     def __contains__(self, obj):
         return id(obj) in self._objects