Skip to content

Instantly share code, notes, and snippets.

@ksobon
Created December 3, 2015 17:21
Show Gist options
  • Save ksobon/21a068de16d9bf0d4ac0 to your computer and use it in GitHub Desktop.
Save ksobon/21a068de16d9bf0d4ac0 to your computer and use it in GitHub Desktop.
# Copyright(c) 2015, Konrad K Sobon
# @arch_laboratory, http://archi-lab.net
# Import Element wrapper extension methods
import clr
clr.AddReference("RevitNodes")
import Revit
clr.ImportExtensions(Revit.Elements)
# Import RevitAPI
clr.AddReference("RevitAPI")
import Autodesk
from Autodesk.Revit.DB import *
# Import DocumentManager and TransactionManager
clr.AddReference("RevitServices")
import RevitServices
from RevitServices.Persistence import DocumentManager
doc = DocumentManager.Instance.CurrentDBDocument
import sys
pyt_path = r'C:\Program Files (x86)\IronPython 2.7\Lib'
sys.path.append(pyt_path)
#The inputs to this node will be stored as a list in the IN variables.
dataEnteringNode = IN
def ProcessList(_func, _list):
return map(lambda x: ProcessList(_func, x) if type(x)==list else _func(x), _list)
def TryGetAssetId(mat):
doc = DocumentManager.Instance.CurrentDBDocument
appearanceId = mat.AppearanceAssetId
if appearanceId != ElementId.InvalidElementId:
appearanceElemn = doc.GetElement(appearanceId)
return appearanceElemn
else:
return ElementId.InvalidElementId
def Unwrap(item):
return UnwrapElement(item)
if isinstance(IN[0], list):
elements = ProcessList(Unwrap, IN[0])
else:
elements = [Unwrap(IN[0])]
try:
errorReport = None
output = ProcessList(TryGetAssetId, elements)
except:
# if error accurs anywhere in the process catch it
import traceback
errorReport = traceback.format_exc()
#Assign your output to the OUT variable
if errorReport == None:
OUT = output
else:
OUT = errorReport
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment