Update filter.py

revitron · Feb 8, 2021 · aecbaf2 · aecbaf2
1 parent 5243f8a
commit aecbaf2
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/revitron/filter.py b/revitron/filter.py
@@ -30,10 +30,9 @@ def __init__(self, scope = None, doc = None):
 			scope (Element ID or list of elements, optional): The optional scope. It can be either a view Id or a list of elements. Defaults to None.
 		"""   
 		import revitron
-
 		if not doc:
 			doc = revitron.DOC 
-
+		self.doc = doc
 		self.scope = scope         
 		if scope:
 			if type(scope) == list:
@@ -97,7 +96,7 @@ def applyStringFilter(self, paramName, value, evaluator, invert = False):
 		for valueProvider in revitron.ParameterValueProviders(paramName).get():
 			rule = revitron.DB.FilterStringRule(valueProvider, evaluator, value, True)
 			_filter = Filter()
-			_filter.collector = revitron.DB.FilteredElementCollector(revitron.DOC, self.getElementIds())
+			_filter.collector = revitron.DB.FilteredElementCollector(self.doc, self.getElementIds())
 			_filter.applyParameterFilter(rule, invert) 
 			filters.append(_filter)
 
@@ -232,7 +231,7 @@ def byStringContainsOneInCsv(self, paramName, csv, invert = False):
 		filters = []
 		for item in csv.split(','):		
 			_filter = Filter()
-			_filter.collector = revitron.DB.FilteredElementCollector(revitron.DOC, self.getElementIds())
+			_filter.collector = revitron.DB.FilteredElementCollector(self.doc, self.getElementIds())
 			_filter.byStringContains(paramName, item.strip(), invert)
 			filters.append(_filter)