Revit API & Dynamo，处理变量输入类型

Question

根据 Dynamo 论坛上的这个帖子：http://dynamobim.org/forums/topic/how-to-handle-variable-list-lengthnesting-in-custom-nodes-python/，我想找到一种方法来处理自定义节点的变量输入。

这是一个更改元素族类型的示例脚本：

import clr
clr.AddReference('ProtoGeometry')
from Autodesk.DesignScript.Geometry import *

# Import ToDSType(bool) extension method
clr.AddReference("RevitNodes")
import Revit
clr.ImportExtensions(Revit.Elements)

# Import DocumentManager and TransactionManager
clr.AddReference("RevitServices")
import RevitServices
from RevitServices.Persistence import DocumentManager
from RevitServices.Transactions import TransactionManager

from System.Collections.Generic import *

# Import RevitAPI
clr.AddReference("RevitAPI")
import Autodesk

doc = DocumentManager.Instance.CurrentDBDocument
uiapp = DocumentManager.Instance.CurrentUIApplication
app = uiapp.Application


doc =  DocumentManager.Instance.CurrentDBDocument
app =  DocumentManager.Instance.CurrentUIApplication.Application


inputItems = []

for i in IN[0]:
    inputItems.append(UnwrapElement(i))

toggle = IN[1]
familyType = UnwrapElement(IN[2]).Id
elementlist = []


TransactionManager.Instance.EnsureInTransaction(doc)

if toggle == True:
    for item in inputItems:
        item.ChangeTypeId(familyType)
        elementlist.append(item)

else:
    None

TransactionManager.Instance.TransactionTaskDone()

OUT = elementlist

我想找到一种方法来处理作为列表、嵌套列表或单个项目（非列表）的输入，如下图所示：

后面两个我直接用一个list.flatten就可以了。问题是找到一种方法来处理所有三种类型的 lists/elements.

想法？

Answer 1

在这种情况下，如果您的目标是使用 Python 来处理所有不同的输入结构，我会建议如下：

如果您只想对某个项目执行操作（比如加 +1）而不考虑数据结构，我会使用递归和简单的 isinstance (IN[0], list) 检查，如下所示：

#The inputs to this node will be stored as a list in the IN variable.
dataEnteringNode = IN

def ProcessList(_func, _list):
    return map( lambda x: ProcessList(_func, x) if type(x)==list else _func(x), _list )

def AddOne(item):
    return item + 1

if isinstance (IN[0], list):
    output = ProcessList(AddOne, IN[0])
else:
    output = AddOne(IN[0])

#Assign your output to the OUT variable.
OUT = output

下面是 Dynamo 中对不同数据结构进行操作的示例：

这种方法非常好，因为它允许您输出与输入完全相同的结构。我经常用这个。

Revit API & Dynamo，处理变量输入类型

Revit API & Dynamo, Handling Variable Input Types

revit-api

revitpythonshell