Python+DRVT 从外部调用 Revit：批量创建门

from typing import List
import math
# drvt_pybind 支持多会话、多文档，先从简单的单会话、单文档开始
# MyContext是在Python中用户自定义的单会话、单文档的上下文管理类
from packs.core import MyContext, m2feet, feet2m, isNone# 导入驱动 Revit 的核心模块
from drvt_pybind.Autodesk import Revit
from drvt_pybind.Autodesk.Revit import DB
from drvt_pybind.Autodesk.Revit.DB import Structure# 遍历标高，返回按Elevation从低到高的List[DB.Level]
def getLevels(ctx : MyContext) -> List[DB.Level]:tmp : List[DB.Level] = []levelList = DB.FilteredElementCollector.ctor(ctx.session, ctx.doc).OfClass(ctx.session.typeof(DB.Level)).ToElements()for i in range(levelList.getCount()):l = DB.asLevel(levelList.getItem(i))if isNone(l):continuetmp.append(l)ret = sorted(tmp, key=lambda x:x.getElevation())return ret# 遍历WallType，返回List[DB.WallType]
def getWallTypes(ctx : MyContext) -> List[DB.WallType]:wallTypeList : List[DB.WallType] = []listWt = DB.FilteredElementCollector.ctor(ctx.session, ctx.doc).OfClass(ctx.session.typeof(DB.WallType)).ToElements()for i in range(listWt.getCount()):wt = DB.asWallType(listWt.getItem(i))if isNone(wt):continuewallTypeList.append(wt)return wallTypeList# 遍历门FamilySymbol，返回List[DB.FamilySymbol]
def getDoorFamilySymobls(ctx : MyContext) -> List[DB.FamilySymbol]:doorFamilySymobls : List[DB.FamilySymbol] = []listSy = DB.FilteredElementCollector.ctor(ctx.session, ctx.doc).OfClass(ctx.session.typeof(DB.FamilySymbol)).ToElements()for i in range(listSy.getCount()):fs = DB.asFamilySymbol(listSy.getItem(i))if isNone(fs):continuecat = fs.getCategory()if isNone(cat) or cat.getBuiltInCategory() != DB.BuiltInCategory.OST_Doors:continue;doorFamilySymobls.append(fs)return doorFamilySymobls# 示例创建墙并在墙上创建门
def DoorCreation(ctx: MyContext) -> None:"""本示例展示如何使用Python+DRVT，在外部让Revit批量创建墙并在墙上创建门"""if ctx.doc is not None:ctx.closeDoc(True)template = "建筑样板.rte"fileName = "DoorCreation.rvt"err = ctx.createAndActiveDoc(template, fileName)if err != 0:raise AssertionError(f"创建项目文档失败: 错误码 {err}")session = ctx.sessiondoc = ctx.doc# 取所有标高，缺省是两个levelList = getLevels(ctx)if len(levelList) < 2:raise AssertionError(f"标高过少，本示例用到2个标高")level0 = levelList[0].getId()level1 = levelList[1].getId()ele0 = levelList[0].getElevation()try:# 取所有墙类型wallTypeList = getWallTypes(ctx)# 取所有门类型doorFamilySymoblList = getDoorFamilySymobls(ctx)# 涉及到修改操作的，需要锁定并用事务包装ctx.lockAndStartTransaction("批量创建示例")docCreate = doc.getCreate()for i in range(0,10):# 取一个WallTypewallType = wallTypeList[i % len(wallTypeList)]# 设置WallTypectx.doc.SetDefaultElementTypeId(DB.ElementTypeGroup.WallType, wallType.getId())# 这里使用轴线法# 创建墙的轴线x = m2feet(i*3)y1 = m2feet(3)y0 = -y1p0 = DB.XYZ.ctor(session, x, y0, ele0)p1 = DB.XYZ.ctor(session, x, y1, ele0)line = DB.Line.CreateBound(session, p0, p1)# 创建墙wall = DB.Wall.Create(session, doc, line, level0, False)# 取“顶部约束”参数p = wall.getParameter(DB.BuiltInParameter.WALL_HEIGHT_TYPE)# 更新其值为“标高2”的IDp.Set(level1)# 选择门FamyliSymboldoorFamilySymbol = doorFamilySymoblList[i % len(doorFamilySymoblList)]if not doorFamilySymbol.getIsActive():doorFamilySymbol.Activate()# 确定门的位置doorPos = DB.XYZ.ctor(session, x, 0, ele0)# 创建门实例docCreate.NewFamilyInstance(doorPos, doorFamilySymbol, wall, levelList[0], Structure.StructuralType.NonStructural)# 提交事务并解锁，若没有错误则修改操作将会生效ctx.commitTransactionAndUnlock()except Exception as e:print(f"【异常】{e}")# 回滚事务并解锁ctx.rollbackTransactionAndUnlock()# 下面这段代码，看过“创建新项目”示例的会很熟悉，就是照搬过来的
# 创建新的上下文（可以按需创建多个，意味着能直接管理多个 Revit 会话）
ctx = MyContext("在这里指定会话名称")
# 初始化会话（启动 Revit 进程实例）
ctx.initSession()# 调用DoorCreation
DoorCreation(ctx)# 保存文档并关闭
ctx.closeDoc(True)# 结束会话（Revit 进程实例将退出）
ctx.dispose()