New folders and files can now be added.

nw/project/item.py

@@ -27,8 +27,9 @@ class NWItem():
 
     CLASS_NONE      = 0
     CLASS_NOVEL     = 1
-    CLASS_CHARACTER = 2
-    CLASS_WORLD     = 3
+    CLASS_CHAPTER   = 2
+    CLASS_CHARACTER = 3
+    CLASS_WORLD     = 4
 
     def __init__(self):
 
@@ -88,6 +89,8 @@ class NWItem():
             self.itemClass = self.CLASS_NONE
         elif theClass == "NOVEL":
             self.itemClass = self.CLASS_NOVEL
+        elif theClass == "CHAPTER":
+            self.itemClass = self.CLASS_CHAPTER
         elif theClass == "CHARACTER":
             self.itemClass = self.CLASS_CHARACTER
         elif theClass == "WORLD":
@@ -113,6 +116,7 @@ class NWItem():
     def getClass(self):
         if self.itemClass == self.CLASS_NONE:      return "NONE"
         if self.itemClass == self.CLASS_NOVEL:     return "NOVEL"
+        if self.itemClass == self.CLASS_CHAPTER:   return "CHAPTER"
         if self.itemClass == self.CLASS_CHARACTER: return "CHARACTER"
         if self.itemClass == self.CLASS_WORLD:     return "WORLD"
         return "NONE"

nw/project/project.py

@@ -44,27 +44,27 @@ class NWProject():
     def buildProjectTree(self):
         return True
 
-    def newRoot(self, rootName, rootType):
+    def newRoot(self, rootName, rootClass):
         newItem = NWItem()
         newItem.setName(rootName)
         newItem.setType(NWItem.TYPE_ROOT)
-        newItem.setClass(rootType)
+        newItem.setClass(rootClass)
         self._appendItem(None,None,newItem)
         return newItem.itemHandle
 
-    def newFolder(self, folderName, pHandle):
+    def newFolder(self, folderName, folderClass, pHandle):
         newItem = NWItem()
         newItem.setName(folderName)
         newItem.setType(NWItem.TYPE_FOLDER)
-        newItem.setClass(NWItem.CLASS_NONE)
+        newItem.setClass(folderClass)
         self._appendItem(None,pHandle,newItem)
         return newItem.itemHandle
 
-    def newFile(self, fileName, pHandle):
+    def newFile(self, fileName, fileClass, pHandle):
         newItem = NWItem()
         newItem.setName(fileName)
         newItem.setType(NWItem.TYPE_FILE)
-        newItem.setClass(NWItem.CLASS_NONE)
+        newItem.setClass(fileClass)
         self._appendItem(None,pHandle,newItem)
         return newItem.itemHandle
 
@@ -79,8 +79,8 @@ class NWProject():
         hNovel = self.newRoot("Novel", NWItem.CLASS_NOVEL)
         hChars = self.newRoot("Characters",NWItem.CLASS_CHARACTER)
         hWorld = self.newRoot("World", NWItem.CLASS_WORLD)
-        hChapt = self.newFolder("New Chapter", hNovel)
-        hScene = self.newFile("New Scene", hChapt)
+        hChapt = self.newFolder("New Chapter", NWItem.CLASS_CHAPTER ,hNovel)
+        hScene = self.newFile("New Scene", NWItem.CLASS_NONE, hChapt)
         return
 
     def openProject(self, fileName):