use lib commands

Author: Wurschdhaud

extensions/pyRevitTools.extension/pyRevit.tab/Analysis.panel/Tools.stack/Inspect.pulldown/Count Faces.pushbutton/script.py

@@ -5,12 +5,7 @@
 output = script.get_output()
 output.close_others()
 
-elements = (
-    DB.FilteredElementCollector(doc)
-    .OfClass(DB.FamilyInstance)
-    .WhereElementIsNotElementType()
-    .ToElements()
-)
+elements = revit.query.get_elements_by_class(DB.FamilyInstance, doc=doc)
 
 # Storage
 processed_types = {}  # typeId: {name, coarse, medium, fine}

extensions/pyRevitTools.extension/pyRevit.tab/Selection.panel/select.stack/Select.pulldown/Select All Objects Passing Filter.pushbutton/script.py

@@ -11,7 +11,7 @@
 only_current_view = my_config.get_option("only_current_view", True)
 reverse_filter = my_config.get_option("reverse_filter", False)
 
-filters = list(DB.FilteredElementCollector(doc).OfClass(DB.ParameterFilterElement))
+filters = list(revit.query.get_elements_by_class(DB.ParameterFilterElement, doc=doc))
 
 if not filters:
     forms.alert("No Filters found", exitscript=True)
@@ -56,4 +56,5 @@
     if exclude_nested and isinstance(el, DB.FamilyInstance) and el.SuperComponent:
         continue
     element_ids.append(el.Id)
-uidoc.Selection.SetElementIds(List[DB.ElementId](element_ids))
+
+revit.get_selection().set_to(element_ids)

extensions/pyRevitTools.extension/pyRevit.tab/Toggles.panel/toggles2.stack/SectionBox.pushbutton/script.py

@@ -1,7 +1,5 @@
 """Toggles visibility of section box in current 3D view"""
 
-import pickle
-
 from pyrevit import framework
 from pyrevit import revit, DB, script
 from pyrevit.compat import get_elementid_value_func
@@ -11,7 +9,7 @@
 active_view_id_value = get_elementid_value(active_view.Id)
 
 logger = script.get_logger()
-datafile = script.get_document_data_file("SectionBox", "pym")
+DATA_SLOTNAME = "SectionBox"
 
 my_config = script.get_config()
 scope = my_config.get_option("scope", "Visibility")
@@ -50,21 +48,17 @@ def toggle_sectionbox_active():
         logger.error("Not a 3D view. Operation canceled.")
         return
 
-    sectionbox_active_state = active_view.IsSectionBoxActive
-
     try:
-        with open(datafile, "rb") as f:
-            view_boxes = pickle.load(f)
+        view_boxes = script.load_data(DATA_SLOTNAME)
     except Exception:
         view_boxes = {}
 
-    if sectionbox_active_state:
+    if active_view.IsSectionBoxActive:
         try:
             sectionbox = active_view.GetSectionBox()
             if sectionbox:
                 view_boxes[active_view_id_value] = revit.serialize(sectionbox)
-                with open(datafile, "wb") as f:
-                    pickle.dump(view_boxes, f)
+                script.store_data(DATA_SLOTNAME, view_boxes)
             active_view.IsSectionBoxActive = False
         except Exception as e:
             logger.error("Error saving section box: {}".format(e))