Fixes #32 - Added __getitem__ to selection

notes.md

@@ -1,5 +1,9 @@
 # Release Notes
 
+### 1.7.3
+* Added Get by Index to Selection (issues #32)
+* Converted ElementSet internal storage to list instead of set to allow ordered sequence
+
 ### 1.7.2
 * Fixed: Collector.get_element_ids() Typo
 * Fix: XYZ can now take regular DB.XYZ

rpw/__init__.py

@@ -26,7 +26,7 @@
 """
 
 __title__ = 'revitpythonwrapper'
-__version__ = '1.7.2'
+__version__ = '1.7.3'
 __maintainer__ = ['Gui Talarico', 'Ehsan Iran-Nejad']
 __license__ = 'MIT'
 __contact__ = 'github.com/gtalarico/revitpythonwrapper'

rpw/db/collection.py

@@ -35,7 +35,7 @@ class ElementSet(BaseObject):
 
     def __init__(self, elements_or_ids=None, doc=revit.doc):
         self.doc = doc
-        self._element_id_set = set()
+        self._element_id_set = []
         if elements_or_ids:
             self.add(elements_or_ids)
 
@@ -49,7 +49,8 @@ def add(self, elements_or_ids):
         """
         element_ids = to_element_ids(elements_or_ids)
         for id_ in element_ids:
-            self._element_id_set.add(id_)
+            if id_ not in self._element_id_set:
+                self._element_id_set.append(id_)
 
     def pop(self, element_reference, wrapped=True):
         """
@@ -69,7 +70,7 @@ def pop(self, element_reference, wrapped=True):
 
     def clear(self):
         """ Clears Set """
-        self._element_id_set = set()
+        self._element_id_set = []
 
     @property
     def _elements(self):
@@ -161,7 +162,7 @@ def __iter__(self):
 
     def __getitem__(self, element_reference):
         """
-        Get Element from set
+        Get Element from set from an element ElementId
 
         Args:
             element_reference (DB.Element, DB.ElementID)
@@ -207,8 +208,6 @@ class ElementCollection(BaseObject):
     True
     >>> element_set.clear()
 
-    NOTE: Similar to DB.ElementCollection, doesnt wrap since there is no advantage
-
     Args:
         (`DB.Element`, `DB.ElementID`, optional): Elements or Element Ids.
     """
@@ -365,7 +364,9 @@ def __bool__(self):
         return bool(self._elements)
 
     def __repr__(self, data=None):
-        return super(ElementCollection, self).__repr__(data={'count': len(self)})
+        return super(ElementCollection, self).__repr__(
+                                                    data={'count': len(self)})
+
 
 class XyzCollection(BaseObject):
     """

rpw/ui/selection.py

@@ -100,6 +100,18 @@ def clear(self):
         ElementSet.clear(self)
         self.update()
 
+    def __getitem__(self, index):
+        """
+        Overrides ElementSet __getitem__ to retrieve from selection
+        based on index.
+        """
+        # https://github.com/gtalarico/revitpythonwrapper/issues/32
+        for n, element in enumerate(self.__iter__()):
+            if n ==index:
+                return element
+        else:
+            raise IndexError('Index is out of range')
+
     def __bool__(self):
         """
         Returns: