Revit API＆amp; 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。问题是找到一种方法来处理所有三种类型的列表/元素。

思想？

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在不同数据结构上运行的一个例子：

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