Preselect group when showing 'Add group' or 'Choose group' dialog

openlp/plugins/images/forms/addgroupform.py

@@ -45,13 +45,17 @@ class AddGroupForm(QtGui.QDialog, Ui_AddGroupDialog):
         QtGui.QDialog.__init__(self, parent)
         self.setupUi(self)
 
-    def exec_(self, clear=True, show_top_level_group=False):
+    def exec_(self, clear=True, show_top_level_group=False, selected_group=None):
         if clear:
             self.name_edit.clear()
         self.name_edit.setFocus()
         if show_top_level_group and not self.parent_group_combobox.top_level_group_added:
             self.parent_group_combobox.insertItem(0, translate('ImagePlugin.MediaItem', '-- Top-level group --'), 0)
             self.parent_group_combobox.top_level_group_added = True
+        if selected_group is not None:
+            for i in range(self.parent_group_combobox.count()):
+                if self.parent_group_combobox.itemData(i) == selected_group:
+                    self.parent_group_combobox.setCurrentIndex(i)
         return QtGui.QDialog.exec_(self)
 
     def accept(self):

openlp/plugins/images/forms/choosegroupform.py

@@ -42,3 +42,10 @@ class ChooseGroupForm(QtGui.QDialog, Ui_ChooseGroupDialog):
         """
         QtGui.QDialog.__init__(self, parent)
         self.setupUi(self)
+
+    def exec_(self, selected_group=None):
+        if selected_group is not None:
+            for i in range(self.group_combobox.count()):
+                if self.group_combobox.itemData(i) == selected_group:
+                    self.group_combobox.setCurrentIndex(i)
+        return QtGui.QDialog.exec_(self)

openlp/plugins/images/lib/mediaitem.py

@@ -325,8 +325,6 @@ class ImageMediaItem(MediaManagerItem):
         ``files``
             The files to be loaded.
         """
-        if target_group is None:
-            self.listView.clear()
         self.loadList(files, target_group)
         last_dir = os.path.split(unicode(files[0]))[0]
         Settings().setValue(self.settingsSection + u'/last directory', last_dir)
@@ -336,20 +334,33 @@ class ImageMediaItem(MediaManagerItem):
         Add new images to the database. This method is called when adding images using the Add button or DnD.
         """
         if target_group is None:
-            # Ask which group the images should be saved in
+            # Find out if a group must be pre-selected
-            if self.manager.get_object_count(ImageGroups) == 0:
+            preselect_group = None
+            selected_items = self.listView.selectedItems()
+            if len(selected_items) > 0:
+                selected_item = selected_items[0]
+                if isinstance(selected_item.data(0, QtCore.Qt.UserRole), ImageFilenames):
+                    selected_item = selected_item.parent()
+                if isinstance(selected_item, QtGui.QTreeWidgetItem):
+                    if isinstance(selected_item.data(0, QtCore.Qt.UserRole), ImageGroups):
+                        preselect_group = selected_item.data(0, QtCore.Qt.UserRole).id
+            # Enable and disable parts of the 'choose group' form
+            if preselect_group is None:
                 self.choose_group_form.nogroup_radio_button.setChecked(True)
                 self.choose_group_form.existing_radio_button.setChecked(False)
                 self.choose_group_form.new_radio_button.setChecked(False)
+            else:
+                self.choose_group_form.nogroup_radio_button.setChecked(False)
+                self.choose_group_form.existing_radio_button.setChecked(True)
+                self.choose_group_form.new_radio_button.setChecked(False)
+            if self.manager.get_object_count(ImageGroups) == 0:
                 self.choose_group_form.existing_radio_button.setDisabled(True)
                 self.choose_group_form.group_combobox.setDisabled(True)
             else:
-                self.choose_group_form.nogroup_radio_button.setChecked(True)
-                self.choose_group_form.existing_radio_button.setChecked(False)
-                self.choose_group_form.new_radio_button.setChecked(False)
                 self.choose_group_form.existing_radio_button.setDisabled(False)
                 self.choose_group_form.group_combobox.setDisabled(False)
-            if self.choose_group_form.exec_():
+            # Ask which group the images should be saved in
+            if self.choose_group_form.exec_(selected_group=preselect_group):
                 if self.choose_group_form.nogroup_radio_button.isChecked():
                     # User chose 'No group'
                     parent_group = ImageGroups()
@@ -368,6 +379,10 @@ class ImageMediaItem(MediaManagerItem):
         else:
             parent_group = target_group.data(0, QtCore.Qt.UserRole)
             if isinstance(parent_group, ImageFilenames):
+                if parent_group.group_id == 0:
+                    parent_group = ImageGroups()
+                    parent_group.id = 0
+                else:
                     parent_group = target_group.parent().data(0, QtCore.Qt.UserRole)
         # If no valid parent group is found, do nothing
         if not isinstance(parent_group, ImageGroups):
@@ -501,7 +516,18 @@ class ImageMediaItem(MediaManagerItem):
         """
         Called to add a new group
         """
-        if self.add_group_form.exec_(show_top_level_group=True):
+        # Find out if a group must be pre-selected
+        preselect_group = 0
+        selected_items = self.listView.selectedItems()
+        if len(selected_items) > 0:
+            selected_item = selected_items[0]
+            if isinstance(selected_item.data(0, QtCore.Qt.UserRole), ImageFilenames):
+                selected_item = selected_item.parent()
+            if isinstance(selected_item, QtGui.QTreeWidgetItem):
+                if isinstance(selected_item.data(0, QtCore.Qt.UserRole), ImageGroups):
+                    preselect_group = selected_item.data(0, QtCore.Qt.UserRole).id
+        # Show 'add group' dialog
+        if self.add_group_form.exec_(show_top_level_group=True, selected_group=preselect_group):
             new_group = ImageGroups.populate(parent_id=self.add_group_form.parent_group_combobox.itemData(
                 self.add_group_form.parent_group_combobox.currentIndex(), QtCore.Qt.UserRole),
                 group_name=self.add_group_form.name_edit.text())