Added layout flag to items

nw/enum.py

@@ -35,6 +35,17 @@ class nwItemClass(Enum):
 
 # END Enum nwItemClass
 
+class nwItemLayout(Enum):
+
+    NO_LAYOUT   = 0
+    TITLE_PAGE  = 1
+    SIMPLE_PAGE = 2
+    PARTITION   = 3
+    CHAPTER     = 4
+    SCENE       = 5
+
+# END Enum nwItemLayout
+
 class nwItemAction(Enum):
 
     NO_ACTION   = 0

nw/project/item.py

@@ -17,7 +17,7 @@ from os       import path, mkdir
 from lxml     import etree
 from datetime import datetime
 
-from nw.enum  import nwItemType, nwItemClass
+from nw.enum  import nwItemType, nwItemClass, nwItemLayout
 
 logger = logging.getLogger(__name__)
 
@@ -33,6 +33,23 @@ class NWItem():
         nwItemClass.TIMELINE  : "Timeline",
         nwItemClass.OBJECT    : "Objects",
     }
+    CLASS_FLAG = {
+        nwItemClass.NO_CLASS  : "X",
+        nwItemClass.NOVEL     : "N",
+        nwItemClass.PLOT      : "P",
+        nwItemClass.CHARACTER : "C",
+        nwItemClass.WORLD     : "L",
+        nwItemClass.TIMELINE  : "T",
+        nwItemClass.OBJECT    : "O",
+    }
+    LAYOUT_FLAG = {
+        nwItemLayout.NO_LAYOUT   : "Xo",
+        nwItemLayout.TITLE_PAGE  : "Tt",
+        nwItemLayout.SIMPLE_PAGE : "Pg",
+        nwItemLayout.PARTITION   : "Pt",
+        nwItemLayout.CHAPTER     : "Ch",
+        nwItemLayout.SCENE       : "Sc",
+    }
 
     def __init__(self):
 
@@ -42,6 +59,7 @@ class NWItem():
         self.itemOrder   = None
         self.itemType    = nwItemType.NO_TYPE
         self.itemClass   = nwItemClass.NO_CLASS
+        self.itemLayout  = nwItemLayout.NO_LAYOUT
         self.itemDepth   = None
         self.hasChildren = False
         self.isExpanded  = False
@@ -68,6 +86,8 @@ class NWItem():
         xSub = self._subPack(xPack,"depth",     text=str(self.itemDepth))
         xSub = self._subPack(xPack,"children",  text=str(self.hasChildren))
         xSub = self._subPack(xPack,"expanded",  text=str(self.isExpanded))
+        if self.itemClass == nwItemClass.NOVEL and self.itemLayout is not nwItemLayout.NO_LAYOUT:
+            xSub = self._subPack(xPack,"layout",    text=str(self.itemLayout.name))
         if self.itemType == nwItemType.FILE:
             xSub = self._subPack(xPack,"charCount", text=str(self.charCount), none=False)
             xSub = self._subPack(xPack,"wordCount", text=str(self.wordCount), none=False)
@@ -92,6 +112,7 @@ class NWItem():
         elif tagName == "order":     self.setOrder(tagValue)
         elif tagName == "type":      self.setType(tagValue)
         elif tagName == "class":     self.setClass(tagValue)
+        elif tagName == "layout":    self.setLayout(tagValue)
         elif tagName == "depth":     self.setDepth(tagValue)
         elif tagName == "children":  self.setChildren(tagValue)
         elif tagName == "expanded":  self.setExpanded(tagValue)
@@ -148,6 +169,19 @@ class NWItem():
         self.itemClass = nwItemClass.NO_CLASS
         return
 
+    def setLayout(self, theLayout):
+        if isinstance(theLayout, nwItemLayout):
+            self.itemLayout = theLayout
+            return
+        else:
+            for itemLayout in nwItemLayout:
+                if theLayout == itemLayout.name:
+                    self.itemLayout = itemLayout
+                    return
+        logger.error("Unrecognised item layout '%s'" % theLayout)
+        self.itemLayout = nwItemLayout.NO_LAYOUT
+        return
+
     def setDepth(self, theDepth):
         theDepth = self._checkInt(theDepth,-1)
         if theDepth >= 0 and theDepth <= self.MAX_DEPTH: