Source code for revitron.selection

""" 
A small helper submodule to simplify accessing the list of selected elements.
"""
from System.Collections.Generic import List


[docs]class Selection: """ A selection helper class. Example:: for element in revitron.Selection.get(): ... """
[docs] @staticmethod def get(): """ Get the currently selected elements. Returns: list: The list of selected elements """ import revitron return [ revitron.DOC.GetElement(elId) for elId in revitron.UIDOC.Selection.GetElementIds() ]
[docs] @staticmethod def first(): """ Get the first elements of the list of selected elements. Returns: object: The first element in a list of selected elements """ return Selection.get()[0]
[docs] @staticmethod def set(ids): """ Set the selection to a list of element ids. Args: ids (list): A list of element ids """ import revitron revitron.UIDOC.Selection.SetElementIds(List[revitron.DB.ElementId](ids)) revitron.UIDOC.RefreshActiveView()