Add some clear data functions

novelwriter/core/index.py

@@ -116,7 +116,7 @@ class NWIndex:
     #  Public Methods
     ##
 
-    def clearIndex(self) -> None:
+    def clear(self) -> None:
         """Clear the index dictionaries and time stamps."""
         self._tagsIndex.clear()
         self._itemIndex.clear()
@@ -125,9 +125,9 @@ class NWIndex:
         SHARED.indexSignalProxy({"event": "clearIndex"})
         return
 
-    def rebuildIndex(self) -> None:
+    def rebuild(self) -> None:
         """Rebuild the entire index from scratch."""
-        self.clearIndex()
+        self.clear()
         for nwItem in self._project.tree:
             if nwItem.isFileType():
                 text = self._project.storage.getDocumentText(nwItem.itemHandle)

novelwriter/core/itemmodel.py

@@ -138,6 +138,7 @@ class ProjectNode:
 
     def updateCount(self, propagate: bool = True) -> None:
         """Update counts, and propagate upwards in the tree."""
+        # print("Counting", self._item.itemHandle)
         self._count = self._item.wordCount + sum(c._count for c in self._children)
         self._cache[C_COUNT_TEXT] = f"{self._count:n}"
         if propagate and (parent := self._parent):
@@ -462,6 +463,11 @@ class ProjectModel(QAbstractItemModel):
     #  Other Methods
     ##
 
+    def clear(self) -> None:
+        """Clear the project model."""
+        self._root._children.clear()
+        return
+
     def allExpanded(self) -> list[QModelIndex]:
         """Return a list of all expanded items."""
         expanded = []

novelwriter/core/project.py

@@ -89,6 +89,12 @@ class NWProject:
         logger.debug("Delete: NWProject")
         return
 
+    def clear(self) -> None:
+        """Clear the project."""
+        self._tree.clear()
+        self._index.clear()
+        return
+
     ##
     #  Properties
     ##
@@ -352,7 +358,7 @@ class NWProject:
         self._index.loadIndex()
         if xmlReader.state == XMLReadState.WAS_LEGACY:
             # Often, the index needs to be rebuilt when updating format
-            self._index.rebuildIndex()
+            self._index.rebuild()
 
         self.updateWordCounts()
         self._session.startSession()
@@ -417,7 +423,7 @@ class NWProject:
     def closeProject(self, idleTime: float = 0.0) -> None:
         """Close the project."""
         logger.info("Closing project")
-        self._index.clearIndex()  # Triggers clear signal, see #1718
+        self._index.clear()  # Triggers clear signal, see #1718
         self._options.saveSettings()
         self._tree.writeToCFile()
         self._session.appendSession(idleTime)

novelwriter/core/tree.py

@@ -109,6 +109,7 @@ class NWTree:
     def clear(self) -> None:
         """Clear the item tree entirely."""
         oldModel = self._model
+        oldModel.clear()
         self._model = ProjectModel(self)
         self._items.clear()
         self._nodes.clear()

novelwriter/guimain.py

@@ -738,7 +738,7 @@ class GuiMain(QMainWindow):
             QApplication.setOverrideCursor(QCursor(Qt.CursorShape.WaitCursor))
             tStart = time()
 
-            SHARED.project.index.rebuildIndex()
+            SHARED.project.index.rebuild()
             SHARED.project.tree.refreshAllItems()
             self.novelView.refreshTree()
 

novelwriter/shared.py

@@ -398,6 +398,7 @@ class SharedData(QObject):
         """Create a new project and spell checking instance."""
         from novelwriter.core.project import NWProject
         if isinstance(self._project, NWProject):
+            self._project.clear()
             del self._project
             del self._spelling
         self._project = NWProject()