Building a simple index now works

nw/project/index.py

@@ -13,16 +13,87 @@
 import logging
 import nw
 
+from nw.project.document import NWDoc
+
 logger = logging.getLogger(__name__)
 
 class NWIndex():
 
-    def __init__(self, theParent):
+    VALID_KEYS = ["todo","tag","pov","char","plot","time","location"]
+
+    def __init__(self, theProject, theParent):
 
         # Internal
-        self.theParent = theParent
-        self.mainConf  = self.theParent.mainConf
+        self.theProject = theProject
+        self.theParent  = theParent
+        self.mainConf   = self.theParent.mainConf
+
+        # Indices
+        self.itemIndex  = {}
+        self.keyIndex   = {}
+        for aKey in self.VALID_KEYS:
+            self.keyIndex[aKey] = []
 
         return
 
+    def scanFile(self, tHandle):
+
+        theDocument = NWDoc(self.theProject, self.theParent)
+        theText = theDocument.openDocument(tHandle, False)
+
+        self.itemIndex[tHandle] = {}
+        for aKey in self.VALID_KEYS:
+            self.itemIndex[tHandle][aKey] = []
+
+        nLine = 0
+        for aLine in theText.splitlines():
+            aLine  = aLine.strip()
+            nLine += 1
+            nChar  = len(aLine)
+            if nChar > 0 and aLine[0] == "@":
+                self.indexThis(tHandle, aLine, nLine)
+
+        for aKey in self.VALID_KEYS:
+            if len(self.itemIndex[tHandle][aKey]) > 0:
+                if tHandle not in self.keyIndex[aKey]:
+                    self.keyIndex[aKey].append(tHandle)
+            else:
+                if tHandle in self.keyIndex[aKey]:
+                    self.keyIndex[aKey].remove(tHandle)
+
+        print(self.itemIndex)
+        print(self.keyIndex)
+
+        return
+
+    def indexThis(self, tHandle, aLine, nLine):
+
+        nChar = len(aLine)
+        nPos  = aLine.find(":")
+        if nPos < 2 or nChar < nPos+2:
+            return False
+
+        aKey = aLine[1:nPos].strip().lower()
+        tVal = aLine[nPos+1:].strip()
+        if aKey not in self.VALID_KEYS:
+            return False
+
+        if aKey == "todo":
+            self.itemIndex[tHandle]["todo"].append((nLine, tVal))
+        elif aKey == "tag":
+            if tVal.find(",") >- 0:
+                return False
+            self.itemIndex[tHandle]["tag"].append((nLine, tVal))
+        else:
+            kVal = tVal.split(",")
+            cVal = []            
+            for aVal in kVal:
+                cVal.append(aVal.strip().lower())
+            if len(cVal) > 0:
+                self.itemIndex[tHandle][aKey].append((nLine, cVal))
+            else:
+                return False
+
+        return True
+
 # END Class NWIndex