[#179] Refactored folder deletion so that it uses selection in the core's directory_tree instead of using the one from the GUI layer.

qt/base/directories_dialog.py

@@ -27,9 +27,11 @@ class DirectoriesDialog(QMainWindow):
         self.app = app
         self.lastAddedFolder = platform.INITIAL_FOLDER_IN_DIALOGS
         self.recentFolders = Recent(self.app, 'recentFolders')
-        self.directoriesModel = DirectoriesModel(self.app)
-        self.directoriesDelegate = DirectoriesDelegate()
         self._setupUi()
+        self.directoriesModel = DirectoriesModel(self.app, view=self.treeView)
+        self.directoriesDelegate = DirectoriesDelegate()
+        self.treeView.setItemDelegate(self.directoriesDelegate)
+        self._setupColumns()
         self.app.recentResults.addMenu(self.menuLoadRecent)
         self.app.recentResults.addMenu(self.menuRecentResults)
         self.recentFolders.addMenu(self.menuRecentFolders)
@@ -106,8 +108,6 @@ class DirectoriesDialog(QMainWindow):
         self.promptLabel = QLabel(tr("Select folders to scan and press \"Scan\"."), self.centralwidget)
         self.verticalLayout.addWidget(self.promptLabel)
         self.treeView = QTreeView(self.centralwidget)
-        self.treeView.setItemDelegate(self.directoriesDelegate)
-        self.treeView.setModel(self.directoriesModel)
         self.treeView.setAcceptDrops(True)
         triggers = QAbstractItemView.DoubleClicked|QAbstractItemView.EditKeyPressed\
             |QAbstractItemView.SelectedClicked
@@ -115,11 +115,6 @@ class DirectoriesDialog(QMainWindow):
         self.treeView.setDragDropOverwriteMode(True)
         self.treeView.setDragDropMode(QAbstractItemView.DropOnly)
         self.treeView.setUniformRowHeights(True)
-        header = self.treeView.header()
-        header.setStretchLastSection(False)
-        header.setResizeMode(0, QHeaderView.Stretch)
-        header.setResizeMode(1, QHeaderView.Fixed)
-        header.resizeSection(1, 100)
         self.verticalLayout.addWidget(self.treeView)
         self.horizontalLayout = QHBoxLayout()
         self.removeFolderButton = QPushButton(self.centralwidget)
@@ -149,6 +144,13 @@ class DirectoriesDialog(QMainWindow):
         else:
             moveToScreenCenter(self)
     
+    def _setupColumns(self):
+        header = self.treeView.header()
+        header.setStretchLastSection(False)
+        header.setResizeMode(0, QHeaderView.Stretch)
+        header.setResizeMode(1, QHeaderView.Fixed)
+        header.resizeSection(1, 100)
+    
     def _updateAddButton(self):
         if self.recentFolders.isEmpty():
             self.addFolderButton.setMenu(None)
@@ -212,8 +214,7 @@ class DirectoriesDialog(QMainWindow):
         index = indexes[0]
         node = index.internalPointer()
         if node.parent is None:
-            row = index.row()
-            self.app.model.remove_directory(row)
+            self.directoriesModel.model.remove_selected()
     
     def scanButtonClicked(self):
         if self.app.model.results.is_modified:

qt/base/directories_model.py

@@ -10,7 +10,7 @@ import urllib.parse
 
 from PyQt4.QtCore import pyqtSignal, Qt, QRect, QUrl, QModelIndex
 from PyQt4.QtGui import (QComboBox, QStyledItemDelegate, QApplication, QBrush, QStyle,
-    QStyleOptionComboBox, QStyleOptionViewItemV4)
+    QStyleOptionComboBox, QStyleOptionViewItemV4, QItemSelection)
 
 from hscommon.trans import trget
 from qtlib.tree_model import RefNode, TreeModel
@@ -60,10 +60,14 @@ class DirectoriesDelegate(QStyledItemDelegate):
     
 
 class DirectoriesModel(TreeModel):
-    def __init__(self, app):
+    def __init__(self, app, view):
         TreeModel.__init__(self)
         self.model = DirectoryTree(self, app.model)
+        self.view = view
+        self.view.setModel(self)
         self.model.connect()
+        
+        self.view.selectionModel().selectionChanged[(QItemSelection, QItemSelection)].connect(self.selectionChanged)
     
     def _createNode(self, ref, row):
         return RefNode(self, None, ref, row)
@@ -139,8 +143,14 @@ class DirectoriesModel(TreeModel):
         # work with ActionMove either. So screw that, and accept anything.
         return Qt.ActionMask
     
+    #--- Events
+    def selectionChanged(self, selected, deselected):
+        newNodes = [modelIndex.internalPointer().ref for modelIndex in self.view.selectionModel().selectedRows()]
+        self.model.selected_nodes = newNodes
+    
     #--- Signals
     foldersAdded = pyqtSignal(list)
+    
     #--- model --> view
     def refresh(self):
         self.reset()