PyQt/PySide中实现QFileDialog选择现有及非现有目录的教程

from PyQt5.QtWidgets import QFileDialog, QLineEdit, QDialogButtonBox, QDialog
from PyQt5.QtCore import QDir, QFileInfo
from PyQt5.QtGui import QWindow # 仅用于Qt6的类型提示，Qt5可省略

class SelectDirDialog(QFileDialog):
    def __init__(self, parent=None, caption='', path=''):
        super().__init__(parent)
        # 必须在任何其他设置之前调用，强制使用Qt风格的对话框
        self.setOptions(QFileDialog.DontUseNativeDialog)

        # 设置文件模式为目录，并将其接受模式设置为AcceptSave
        # AcceptSave模式会自动将标题更改为“另存为”或其翻译
        # 但我们希望能够自定义标题，所以先设置FileMode
        self.setFileMode(QFileDialog.Directory)
        title = caption or self.windowTitle() # 获取或设置自定义标题
        self.setAcceptMode(QFileDialog.AcceptSave)
        self.setWindowTitle(title)

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

        # 通过objectName访问内部控件
        # 文件名编辑框的objectName是'fileNameEdit'
        self.fileNameEdit = self.findChild(QLineEdit, 'fileNameEdit')
        # 连接textChanged信号到自定义的槽函数，用于检查OK按钮状态
        if self.fileNameEdit:
            self.fileNameEdit.textChanged.connect(self.checkOkButton)

        # 接受按钮（通常是QDialogButtonBox中的Save按钮）
        # 注意：QDialogButtonBox可能包含多个按钮，需要指定是Save按钮
        button_box = self.findChild(QDialogButtonBox)
        if button_box:
            self.okButton = button_box.button(QDialogButtonBox.Save)
        else:
            self.okButton = None # 处理找不到按钮的情况

    def accept(self):
        """
        重写accept方法，以允许选择不存在的目录。
        """
        files = self.selectedFiles()
        if not files:
            # 如果没有文件被选中，调用父类的accept方法（通常是QFileDialog的默认行为）
            super().accept()
            return

        # 获取第一个选中的路径信息
        info = QFileInfo(files[0])
        # 如果是目录或路径不存在，则接受对话框
        if info.isDir() or not info.exists():
            # 重要：不能调用QFileDialog的accept()，因为它会使用默认行为，
            # 不接受不存在的目录。应调用QDialog的accept()。
            QDialog.accept(self)
        else:
            # 否则，调用QFileDialog的默认accept()行为（例如，如果选择了文件而不是目录）
            super().accept()

    def checkOkButton(self):
        """
        根据当前输入框的文本，动态启用或禁用OK按钮。
        """
        if self.okButton is None:
            return

        # 如果OK按钮已经启用，则无需进一步检查
        if self.okButton.isEnabled():
            return

        # 获取当前文件名编辑框的文本
        current_text = self.fileNameEdit.text()
        info = QFileInfo(current_text)

        # 如果文本为空，或者对应的路径是目录，或者路径不存在，则启用OK按钮
        self.okButton.setEnabled(current_text == '' or info.isDir() or not info.exists())

    def selectedPath(self):
        """
        获取用户选择的最终路径。
        """
        files = self.selectedFiles()
        return files[0] if files else ''

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QPushButton, QVBoxLayout, QWidget, QLabel

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("自定义目录选择示例")
        self.setGeometry(100, 100, 400, 200)

        central_widget = QWidget()
        self.setCentralWidget(central_widget)
        layout = QVBoxLayout(central_widget)

        self.select_button = QPushButton("选择目录")
        self.select_button.clicked.connect(self.open_custom_dialog)
        layout.addWidget(self.select_button)

        self.path_label = QLabel("选中的路径: 无")
        layout.addWidget(self.path_label)

    def open_custom_dialog(self):
        dlg = SelectDirDialog(self, "请选择一个目录...", QDir.currentPath())
        # 还可以添加ShowDirsOnly选项，确保只显示目录
        dlg.setOptions(dlg.options() | QFileDialog.ShowDirsOnly) 

        if dlg.exec_(): # 注意Qt5中使用exec_()
            selected_path = dlg.selectedPath()
            print("选中的路径:", selected_path)
            self.path_label.setText(f"选中的路径: {selected_path}")
            # 在这里你可以使用os.makedirs(selected_path, exist_ok=True)来创建目录
            # import os
            # os.makedirs(selected_path, exist_ok=True)
        else:
            print("取消选择")
            self.path_label.setText("选中的路径: 取消")

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