使用ezdxf在PyQt5应用中集成DWG/DXF文件查看器

pip install ezdxf PyQt5

import sys
from pathlib import Path
from PyQt5.QtWidgets import QApplication, QMainWindow, QVBoxLayout, QWidget, QFileDialog, QAction
from PyQt5.QtCore import Qt

# 从 ezdxf.addons.drawing 导入 CADViewer
from ezdxf.addons.drawing.qtviewer import CADViewer
import ezdxf

class DXFViewerApp(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("ezdxf DXF Viewer")
        self.setGeometry(100, 100, 1024, 768)

        self._create_menu()

        # 创建 CADViewer 实例，它是一个 QWidget
        self.viewer = CADViewer(self)

        # 将 CADViewer 设置为中心部件
        central_widget = QWidget()
        layout = QVBoxLayout(central_widget)
        layout.addWidget(self.viewer)
        self.setCentralWidget(central_widget)

        self.current_dxf_path = None

    def _create_menu(self):
        menubar = self.menuBar()
        file_menu = menubar.addMenu("文件")

        open_action = QAction("打开 DXF...", self)
        open_action.setShortcut("Ctrl+O")
        open_action.triggered.connect(self._open_dxf_file_dialog)
        file_menu.addAction(open_action)

        exit_action = QAction("退出", self)
        exit_action.setShortcut("Ctrl+Q")
        exit_action.triggered.connect(self.close)
        file_menu.addAction(exit_action)

    def _open_dxf_file_dialog(self):
        """打开文件对话框并加载选定的DXF文件"""
        options = QFileDialog.Options()
        file_path, _ = QFileDialog.getOpenFileName(
            self, "打开 DXF 文件", "", "DXF 文件 (*.dxf);;所有文件 (*)", options=options
        )
        if file_path:
            self.load_dxf_file(file_path)

    def load_dxf_file(self, filename: str):
        """加载并显示指定的DXF文件"""
        if not Path(filename).exists():
            print(f"错误: 文件 '{filename}' 不存在。")
            return

        try:
            # 使用 ezdxf 读取 DXF 文件
            doc = ezdxf.readfile(filename)
            # 将文档设置给 CADViewer
            self.viewer.set_document(doc)
            self.current_dxf_path = filename
            self.setWindowTitle(f"ezdxf DXF Viewer - {Path(filename).name}")
            print(f"成功加载 DXF 文件: {filename}")
        except ezdxf.DXFStructureError as e:
            print(f"DXF 文件结构错误: {e}")
        except Exception as e:
            print(f"加载 DXF 文件时发生未知错误: {e}")

if __name__ == '__main__':
    app = QApplication(sys.argv)
    main_win = DXFViewerApp()
    main_win.show()
    sys.exit(app.exec_())