Linux下PyQt5实现仅限USB盘访问的受保护文件对话框需求咨询

需求背景 #

在Linux系统上使用PyQt5开发应用，需要实现文件选择/保存对话框，仅允许用户在指定U盘挂载目录（如/mnt/myUSB）内浏览操作，禁止进入系统目录（如/usr/local/bin）。默认的QFileDialog无法限制用户向上导航，存在安全风险。

解决方法 #

方法1：自定义对话框+限制文件系统模型（推荐）

通过自定义QFileSystemModel，直接将对话框的根目录锁定为U盘挂载点，从根源上阻止用户访问外部路径。

from PyQt5 import QtWidgets, QtGui

class RestrictedFileDialog(QtWidgets.QFileDialog):
    def __init__(self, usb_root, parent=None):
        super().__init__(parent)
        # 初始化U盘目录
        self.usb_root = usb_root
        self.setDirectory(usb_root)
        
        # 创建受限文件模型，仅显示U盘目录内容
        self.file_model = QtGui.QFileSystemModel()
        self.file_model.setRootPath(usb_root)
        # 设置对话框的根索引为U盘目录
        root_idx = self.file_model.index(usb_root)
        self.setModel(self.file_model)
        self.setRootIndex(root_idx)
        
        # 可选：禁用"返回上级目录"按钮，强化限制
        back_btn = self.findChild(QtWidgets.QToolButton, 'backButton')
        if back_btn:
            back_btn.setEnabled(False)

# 使用示例
# 假设U盘挂载路径为/mnt/myUSB
usb_mount_path = '/mnt/myUSB'
# 打开文件选择对话框
dialog = RestrictedFileDialog(usb_mount_path, self)
dialog.setFileMode(QtWidgets.QFileDialog.AnyFile)
# 若需保存对话框，切换接受模式
# dialog.setAcceptMode(QtWidgets.QFileDialog.AcceptSave)

if dialog.exec_() == QtWidgets.QDialog.Accepted:
    selected_files = dialog.selectedFiles()
    # 处理选中文件逻辑

原理说明：QFileSystemModel的setRootPath会限制模型仅加载指定目录下的文件，setRootIndex则让对话框只能从该索引开始浏览，用户无法通过任何方式导航到U盘目录以外的路径。

方法2：监听目录变化，强制拉回U盘目录

若不想自定义对话框，可通过监听directoryEntered信号，检查当前目录是否在U盘范围内，若超出则强制切换回去。

from PyQt5 import QtWidgets

usb_mount_path = '/mnt/myUSB'
dialog = QtWidgets.QFileDialog(self)
dialog.setDirectory(usb_mount_path)
dialog.setFileMode(QtWidgets.QFileDialog.AnyFile)

# 监听目录切换事件
def enforce_usb_directory(current_path):
    # 检查当前路径是否为U盘目录的子路径
    if not current_path.startswith(usb_mount_path):
        dialog.setDirectory(usb_mount_path)

dialog.directoryEntered.connect(enforce_usb_directory)

if dialog.exec_() == QtWidgets.QDialog.Accepted:
    selected_files = dialog.selectedFiles()
    # 处理选中文件逻辑

优缺点：无需自定义类，但用户可能会看到短暂的目录跳转，体验不如方法1；极端情况下可能存在绕过风险。

额外注意事项 #

• 确保U盘挂载路径固定，或在运行时通过读取/etc/mtab、监听udev事件动态获取挂载点
• 可结合setNameFilters限制允许操作的文件类型，例如：dialog.setNameFilters(["升级包 (*.bin)", "配置文件 (*.conf)"])
• 对于保存对话框，方法1同样能阻止用户在U盘外创建文件

内容的提问来源于stack exchange，提问作者Stefano Remigio