Skip to content

Commit

Permalink
Update filter.py
Browse files Browse the repository at this point in the history
  • Loading branch information
marcantondahmen committed Dec 2, 2020
1 parent 86c8ee4 commit 9038caa
Showing 1 changed file with 15 additions and 12 deletions.
27 changes: 15 additions & 12 deletions revitron/filter.py
Original file line number Diff line number Diff line change
Expand Up @@ -127,25 +127,28 @@ def byRegex(self, paramName, regex, invert = False):
object: The Filter instance
"""
import revitron
from revitron import _

passed = []
failed = []

for elementId in self.getElementIds():
value = str(_(elementId).get(paramName))
if re.search(regex, value, re.IGNORECASE):
passed.append(elementId)
else:
failed.append(elementId)
for element in self.getElements():
value = revitron.Parameter(element, paramName).getString()
if not value:
value = revitron.Parameter(element, paramName).getValueString()
if value:
if re.search(regex, value, re.IGNORECASE):
passed.append(element)
print(value)
else:
failed.append(element)

if invert:
excluded = passed
if not invert:
elements = passed
else:
excluded = failed
elements = failed

if excluded:
self.collector = self.collector.Excluding(List[revitron.DB.ElementId](excluded))
if elements:
self.collector = Filter(elements).collector

return self

Expand Down

0 comments on commit 9038caa

Please sign in to comment.