ezdxf 库实战：使用 Python 进行 CAD 图纸自动化处理
ezdxf 是一个纯 Python 实现的 DXF 文件处理工具，无需安装 AutoCAD 等软件即可读写、编辑和生成图纸。对于机械工程师、建筑设计师或数据开发者来说，掌握它能让 CAD 自动化效率倍增。
快速上手
安装只需一行命令，简单到让人难以置信：
pip install ezdxf
验证安装是否成功，直接打印版本号：
import ezdxf
print(f"ezdxf 版本：{ezdxf.__version__}")
绘制基础图形
让我们从一个简单的例子开始，感受 ezdxf 的强大之处。创建新图纸就像打开一张空白画布，进入模型空间后就可以添加基本元素：
import ezdxf
# 创建新图纸 - 指定 DXF 版本
doc = ezdxf.new('R2010')
msp = doc.modelspace()
# 绘制基本图形
msp.add_line((0, 0), (100, 50)) # 斜线
msp.add_circle((50, 50), 25) # 圆形
msp.add_text("Hello CAD!", height=10).set_pos((25, 80)) # 文本
# 保存成果
doc.saveas('my_first_drawing.dxf')
运行这段代码，你就成功创建了第一个 DXF 文件！整个过程就像用 Python 写文字一样简单，只不过这次你'写'的是图形。
图层与块管理
专业的 CAD 图纸离不开良好的图层管理。ezdxf 让你能够轻松创建和管理图层，不同颜色的图层对应不同的业务含义：
# 创建专业图层
doc.layers.add('WALLS', color=2) # 红色图层
doc.layers.add('DOORS', color=3) # 绿色图层
# 在不同图层上绘制
msp.add_line((0, 0), (100, 0), dxfattribs={'layer': 'WALLS'})
msp.add_circle((50, 50), 10, dxfattribs={'layer': 'DOORS'})
在 CAD 设计中，重用是提高效率的关键。ezdxf 的块功能让你能够创建可重复使用的组件，比如定义一个椅子块并多次插入：
# 创建块定义
block = doc.blocks.new('CHAIR')
# 在块中绘制图形
block.add_line((0, 0), (20, 0))
block.add_line((20, 0), (20, 30))
block.add_line((20, 30), (0, 30))
block.add_line((0, 30), (0, 0))
# 在模型空间中插入块
msp.add_blockref('CHAIR', (0, 0))
msp.add_blockref('CHAIR', (50, 0))
批量处理与自动化
当你有成百上千个 DXF 文件需要统一处理时，ezdxf 的批量处理能力将大显身手。想象一下，你需要为整个办公楼的每个房间标注面积，手动操作可能需要数小时，而用脚本只需几分钟：
import os
input_folder = 'project_files'
output_folder = 'processed_files'
for filename in os.listdir(input_folder):
if filename.endswith('.dxf'):
doc = ezdxf.readfile(os.path.join(input_folder, filename))
# 执行统一操作，如标准化图层
doc.saveas(os.path.join(output_folder, filename))
查询特定实体也很方便，比如闭合的多段线：
for room_polyline in msp.query('LWPOLYLINE[closed=True]'):
area = room_polyline.area
center = room_polyline.vertices_center
msp.add_text(f"{area:.1f}㎡", height=2).set_pos(center)
三维建模与视觉效果
ezdxf 不仅限于二维绘图，还能创建令人惊叹的三维模型。虽然不如专业建模软件强大，但在工程图纸中足够用：
import numpy as np
vertices = np.array([
[0, 0, 0], [10, 0, 0], [10, 10, 0], [0, 10, 0],
[0, 0, 10], [10, 0, 10], [10, 10, 10], [0, 10, 10]
])
faces = [
[0, 1, 2, 3], # 底面
[4, 5, 6, 7], # 顶面
[0, 1, 5, 4] # 侧面
]
msp.add_mesh(vertices, faces)
颜色与线宽设置同样重要，让你的图纸更加生动和专业：
# 使用 ACI 颜色系统
msp.add_line((0, 0), (100, 0), dxfattribs={'color': 1}) # 红色
msp.add_line((0, 10), (100, 10), dxfattribs={'color': 5}) # 蓝色
# 设置线宽
msp.add_line((0, 20), (100, 20), dxfattribs={'lineweight': 25})
性能与异常处理
处理大型图纸时，这些技巧将显著提升性能：减少文件保存次数、及时删除不再需要的实体、用实体查询替代遍历所有实体。
编写健壮的 CAD 自动化脚本必须包含错误捕获，防止因文件损坏导致程序崩溃：
try:
doc = ezdxf.readfile('complex_drawing.dxf')
# 处理图纸...
except ezdxf.DXFStructureError as e:
print(f"文件结构错误：{e}")
except Exception as e:
print(f"处理失败：{e}")
官方文档和测试用例也是很好的学习资源。通过上述方法，你可以利用 Python 高效完成 CAD 图纸的自动化任务。
