发帖 回复
返回列表
  • 534阅读
  • 0回复

[讨论]QFileSystemModel+QTreeView如何让目录树只显示本地磁盘，不显示外部存储设备？ [复制链接]

 上一主题 下一主题
离线lwei24
骑士
 
只看楼主 倒序阅读 楼主  发表于: 2023-05-05
如题，QFileSystemModel+QTreeView如何实现只显示本地磁盘，不显示外部存储设备的目录树。各位大佬们，如下
//frmain.cpp
复制代码
  1. FrmMain::FrmMain(QWidget *parent) : QMainWindow(parent)
  2. {
  3.     m_pTreeView = new QTreeView(this);
  5.     dir = new QFileSystemModel(this);
  6.     dir->setReadOnly(false);
  7.     dir->setRootPath(QDir::currentPath());
  9.     m_usbModel = new USBDriveFilterProxyModel(this);
  10.     m_usbModel->setSourceModel(dir);
  11.     m_usbModel->setDynamicSortFilter(true);
  12.     m_pTreeView->setModel(m_usbModel);
  14.     m_pTreeView->header()->setStretchLastSection(true);
  15.     m_pTreeView->header()->setSortIndicatorShown(true);
  16.     m_pTreeView->setHeaderHidden(true);
  17.     m_pTreeView->hideColumn(1);
  18.     m_pTreeView->hideColumn(2);
  19.     m_pTreeView->hideColumn(3);
  20.     m_pTreeView->sortByColumn(0, Qt::DescendingOrder);
  21.     m_pTreeView->collapseAll();
  22.     m_pTreeView->resizeColumnToContents(0);
  24.     this->setCentralWidget(m_pTreeView);
  25. }
//usbdrivefilter.cpp
复制代码
  1. #include "usbdrivefilter.h"
  3. #include <windows.h>
  5. USBDriveFilterProxyModel::USBDriveFilterProxyModel(QObject *parent) :
  6.     QSortFilterProxyModel(parent)
  7. {
  8.     getMountedRemovables();
  9.     // will eventually also register for changes to mounted removables
  10.     // but need to get passed my current issue of not displaying only USBs.
  11. }
  13. bool USBDriveFilterProxyModel::filterAcceptsRow(int sourceRow,
  14.                                                 const QModelIndex &sourceParent) const
  15. {
  16.     QModelIndex index0 = sourceModel()->index(sourceRow, 0, sourceParent);
  18.     // Since drive string can have more than just "<DriveLetter>:", need
  19.     // to check each entry in the usb list for whether it is contained in
  20.     // the current drive string.
  23.     for (int i = 0; i < drivers.size(); i++)
  24.     {
  25.         if (sourceModel()->data(index0).toString().contains(drivers[i]))
  26.         {
  27.             return true;
  28.         }
  29.     }
  32.     return false;
  33. }
  35. void USBDriveFilterProxyModel::getMountedRemovables()
  36. {
  37.     DWORD test = GetLogicalDrives();
  39.     DWORD mask = 1;
  40.     UINT type = 0;
  41.     WCHAR wdrive[] = L"C:\\"; // use as a drive letter template
  42.     for (int i = 0; i < 32; i++)
  43.     {
  44.         if (test & mask)
  45.         {
  46.             wdrive[0] = (char)('A' + i); // change letter in template
  47.             type = GetDriveType(wdrive);
  48.             switch (type) {
  49.             case DRIVE_REMOVABLE:
  50.             {
  51.                 QString qdrive = QString((char)('A' + i)) + ":";
  52.                 removables.append(qdrive);
  53.                 break;
  54.             }
  55.             case DRIVE_FIXED:
  56.             {
  57.                 QString qdrive = QString((char)('A' + i)) + ":";
  58.                 drivers.append(qdrive);
  59.             }
  60.                 break;
  61.             default:
  62.                 break;
  63.             }
  64.         }
  65.         mask = mask << 1;
  66.     }
  67. }
//main.cpp
复制代码
  1. #include <QApplication>
  3. #include "frmmain.h"
  5. int main(int argc, char **argv)
  6. {
  7.     QApplication app(argc, argv);
  9.     FrmMain w;
  10.     w.resize(500, 500);
  11.     w.show();
  12.     return app.exec();
  13. }
如上代码，为什么虽然你能实现显示本地磁盘，但是点击本地磁盘后，就没有目录树了。各位大佬，有什么拌粉实现只显示本地磁盘，不显示外部存储设备的目录树呢？欢迎大佬们指点一二，小弟在此感激不尽！！
回复
举报
分享到 淘江湖 新浪 QQ微博 QQ空间 开心 人人 豆瓣 网易微博 百度 鲜果 白社会 飞信
发帖 回复
返回列表
快速回复
限100 字节
 
上一个 下一个