Update view.py

revitron · Jun 18, 2020 · 8bcee99 · 8bcee99
1 parent e2f6ddb
commit 8bcee99
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/revitron/view.py b/revitron/view.py
@@ -10,10 +10,12 @@ def __init__(self, sheets):
         Inits an new ViewSheetList object. 
 
         Args:
-            sheets (list): A list with sheets
+            sheets (list): A list with sheets or sheet ids
         """      
         self.views = []  
         for sheet in sheets:
+            if isinstance(sheet, revitron.DB.ElementId):
+                sheet = revitron.DOC.GetElement(sheet)
             if revitron.Element(sheet).getClassName() == 'ViewSheet':
                 for viewId in sheet.GetAllPlacedViews():
                     item = revitron._helpers.AttrDict()