Add context menu option to rename item to first heading (#1692)

vkbo · Feb 11, 2024 · 81a9d51 · 81a9d51
2 parents dc661b7 + e6efd15
commit 81a9d51
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 6 deletions.
diff --git a/novelwriter/gui/projtree.py b/novelwriter/gui/projtree.py
@@ -1785,6 +1785,7 @@ def buildSingleSelectMenu(self, hasChild: bool) -> None:
         action = self.addAction(self.tr("Rename"))
         action.triggered.connect(lambda: self.projTree.renameTreeItem(self._handle))
         if isFile:
+            self._itemHeader()
             self._itemActive(False)
         self._itemStatusImport(False)
 
@@ -1833,6 +1834,15 @@ def _itemCreation(self) -> None:
         menu.addAction(self.projView.projBar.aAddFolder)
         return
 
+    def _itemHeader(self) -> None:
+        """Check if there is a header that can be used for rename."""
+        if hItem := SHARED.project.index.getItemHeader(self._handle, "T0001"):
+            action = self.addAction(self.tr("Rename to Heading"))
+            action.triggered.connect(
+                lambda: self.projTree.renameTreeItem(self._handle, hItem.title)
+            )
+        return
+
     def _itemActive(self, multi: bool) -> None:
         """Add Active/Inactive actions."""
         if multi:

diff --git a/tests/test_gui/test_gui_projtree.py b/tests/test_gui/test_gui_projtree.py
@@ -1223,8 +1223,8 @@ def getTransformSubMenu(menu: QMenu) -> list[str]:
     ctxMenu.buildSingleSelectMenu(True)
     actions = [x.text() for x in ctxMenu.actions() if x.text()]
     assert actions == [
-        "Open Document", "View Document", "Create New ...", "Rename", "Toggle Active",
-        "Set Status to ...", "Transform ...", "Expand All", "Collapse All",
+        "Open Document", "View Document", "Create New ...", "Rename", "Rename to Heading",
+        "Toggle Active", "Set Status to ...", "Transform ...", "Expand All", "Collapse All",
         "Duplicate from Here", "Move to Trash",
     ]
     assert getTransformSubMenu(ctxMenu) == [
@@ -1239,8 +1239,9 @@ def getTransformSubMenu(menu: QMenu) -> list[str]:
     ctxMenu.buildSingleSelectMenu(False)
     actions = [x.text() for x in ctxMenu.actions() if x.text()]
     assert actions == [
-        "Open Document", "View Document", "Create New ...", "Rename", "Toggle Active",
-        "Set Importance to ...", "Transform ...", "Duplicate Document", "Move to Trash",
+        "Open Document", "View Document", "Create New ...", "Rename", "Rename to Heading",
+        "Toggle Active", "Set Importance to ...", "Transform ...", "Duplicate Document",
+        "Move to Trash",
     ]
     assert getTransformSubMenu(ctxMenu) == [
         "Split Document by Headers",
@@ -1253,8 +1254,9 @@ def getTransformSubMenu(menu: QMenu) -> list[str]:
     ctxMenu.buildSingleSelectMenu(False)
     actions = [x.text() for x in ctxMenu.actions() if x.text()]
     assert actions == [
-        "Open Document", "View Document", "Create New ...", "Rename", "Toggle Active",
-        "Set Status to ...", "Transform ...", "Duplicate Document", "Move to Trash",
+        "Open Document", "View Document", "Create New ...", "Rename", "Rename to Heading",
+        "Toggle Active", "Set Status to ...", "Transform ...", "Duplicate Document",
+        "Move to Trash",
     ]
     assert getTransformSubMenu(ctxMenu) == [
         "Convert to Novel Document", "Split Document by Headers",