Veronica K. B. Olsen
114 lines
3.0 KiB
Python
114 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""novelWriter Project Index
|
||
|
||
novelWriter – Project Index
|
||
=============================
|
||
Class holding the index of tags
|
||
|
||
File History:
|
||
Created: 2019-05-27 [0.1.4]
|
||
|
||
"""
|
||
|
||
import logging
|
||
import nw
|
||
|
||
from nw.project.document import NWDoc
|
||
from nw.enum import nwItemType
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
class NWIndex():
|
||
|
||
VALID_KEYS = [
|
||
"todo","tag","pov","char","plot","time","location",
|
||
"object","custom","scene","chapter","part"
|
||
]
|
||
|
||
def __init__(self, theProject, theParent):
|
||
|
||
# Internal
|
||
self.theProject = theProject
|
||
self.theParent = theParent
|
||
self.mainConf = self.theParent.mainConf
|
||
|
||
# Indices
|
||
self.itemIndex = {}
|
||
self.tagIndex = {}
|
||
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)
|
||
theItem = self.theProject.getItem(tHandle)
|
||
|
||
if theItem is None:
|
||
return False
|
||
|
||
if theItem.itemType != nwItemType.FILE:
|
||
return 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, theItem)
|
||
|
||
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.tagIndex)
|
||
print(self.keyIndex)
|
||
|
||
return True
|
||
|
||
def indexThis(self, tHandle, aLine, nLine, theItem):
|
||
|
||
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))
|
||
self.tagIndex[tVal] = (nLine, tHandle, theItem.itemClass)
|
||
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
|