PyQt/PySide：实现QFileDialog选择现有或非现有目录的教程

from PyQt5.QtWidgets import (
    QFileDialog, QLineEdit, QDialogButtonBox, QDialog, QApplication, QWidget, QVBoxLayout, QPushButton
)
from PyQt5.QtCore import QDir, QFileInfo
import sys
import os

class SelectDirDialog(QFileDialog):
    """
    一个自定义的QFileDialog，允许用户选择一个目录，
    该目录可以已经存在，也可以是新输入的（非现有）。
    """
    def __init__(self, parent=None, caption='', path=''):
        super().__init__(parent)
        # 1. 禁用原生对话框：必须在设置其他选项之前调用
        # 对于Qt6，请使用 self.Option.DontUseNativeDialog
        self.setOptions(self.DontUseNativeDialog)

        # 2. 设置文件模式和接受模式
        # 设置为目录模式，并使用AcceptSave来获取“保存”按钮
        self.setFileMode(self.Directory)
        title = caption or self.windowTitle() # 获取默认标题或使用自定义标题
        self.setAcceptMode(self.AcceptSave)
        self.setWindowTitle(title)

        # 初始目录设置
        self.setDirectory(path or QDir.currentPath())

        # 查找对话框内部的控件，以便进行自定义控制
        # 'fileNameEdit' 是路径输入框的objectName
        self.fileNameEdit = self.findChild(QLineEdit, 'fileNameEdit')
        if self.fileNameEdit:
            # 连接textChanged信号到自定义槽函数，以便动态控制“保存”按钮
            self.fileNameEdit.textChanged.connect(self.checkOkButton)

        # 查找QDialogButtonBox中的“保存”按钮
        # 对于Qt6，QDialogButtonBox.Save 可能需要是 QDialogButtonBox.StandardButton.Save
        button_box = self.findChild(QDialogButtonBox)
        if button_box:
            self.okButton = button_box.button(QDialogButtonBox.Save)
        else:
            self.okButton = None # 如果找不到按钮，则置为None

        # 确保在初始化时调用一次checkOkButton，以设置按钮的初始状态
        self.checkOkButton()

    def accept(self):
        """
        重写accept方法，以允许接受非现有目录。
        """
        files = self.selectedFiles()
        if not files:
            # 如果没有选择任何文件（即路径输入框为空），则调用父类的accept
            super().accept()
            return

        # 获取用户输入的路径
        selected_path = files[0]
        info = QFileInfo(selected_path)

        # 如果路径是一个现有目录，或者路径不存在，则视为有效
        if info.isDir() or not info.exists():
            # 关键：不能调用QFileDialog的accept()，因为它会执行默认的路径验证。
            # 而是调用QDialog的accept()，它只负责关闭对话框并返回Accepted状态。
            QDialog.accept(self)
        else:
            # 如果路径存在但不是目录（例如是一个文件），则禁用“保存”按钮
            # 或者可以显示一个警告信息
            if self.okButton:
                self.okButton.setEnabled(False)

    def checkOkButton(self):
        """
        根据路径输入框的内容动态启用或禁用“保存”按钮。
        """
        if not self.okButton:
            return

        # 获取当前输入框中的文本
        current_text = self.fileNameEdit.text()
        info = QFileInfo(current_text)

        # 如果当前文本为空，或者它是一个现有目录，或者它不存在，则启用按钮
        # 否则（即文本指向一个现有文件但不是目录），禁用按钮
        self.okButton.setEnabled(current_text == "" or info.isDir() or not info.exists())

    def selectedPath(self):
        """
        提供一个便捷方法来获取最终选择的路径。
        """
        files = self.selectedFiles()
        return files[0] if files else ''

# 示例用法
class MyApp(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.setWindowTitle('自定义目录选择器')
        self.setGeometry(300, 300, 300, 200)

        layout = QVBoxLayout()
        self.btn = QPushButton('选择目录', self)
        self.btn.clicked.connect(self.openDirDialog)
        layout.addWidget(self.btn)
        self.setLayout(layout)

    def openDirDialog(self):
        # 实例化自定义的目录选择对话框
        dlg = SelectDirDialog(self, "请选择或输入一个目录", QDir.homePath())
        # 可选：如果你只想显示目录而不显示文件，可以添加ShowDirsOnly选项
        # dlg.setOption(QFileDialog.ShowDirsOnly) # 对于Qt6是QFileDialog.Option.ShowDirsOnly

        if dlg.exec_(): # 使用exec_()来显示模态对话框
            selected_dir = dlg.selectedPath()
            print(f"用户选择的目录是: {selected_dir}")
            # 在这里，你可以检查目录是否存在，如果不存在则创建它
            if selected_dir and not os.path.exists(selected_dir):
                try:
                    os.makedirs(selected_dir)
                    print(f"已成功创建目录: {selected_dir}")
                except OSError as e:
                    print(f"创建目录失败: {e}")
            elif selected_dir:
                print(f"目录已存在: {selected_dir}")

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