Veronica Berglyd Olsen
1c45331b0a
* Fix bug in early error reporting in main init * Update docstrings and optimise code in GuiMain * Update docstrings and optimise code in Config * Update docstrings and optimise code in common module * Update docstrings and optimise code in main project classes * Update the OptionState class * Update the spell checker class * Update the file converter classes and extend tests * Update the about, merge, split and item editor classes and extend tests * Update item editor test * Update about dialog tests * Some minor test cleanup * Fix typo and add clarification in contributing guide
328 lines
11 KiB
Python
328 lines
11 KiB
Python
"""
|
||
novelWriter – Project Item Class
|
||
================================
|
||
Data class for a project tree item
|
||
|
||
File History:
|
||
Created: 2018-10-27 [0.0.1]
|
||
|
||
This file is a part of novelWriter
|
||
Copyright 2018–2021, Veronica Berglyd Olsen
|
||
|
||
This program is free software: you can redistribute it and/or modify
|
||
it under the terms of the GNU General Public License as published by
|
||
the Free Software Foundation, either version 3 of the License, or
|
||
(at your option) any later version.
|
||
|
||
This program is distributed in the hope that it will be useful, but
|
||
WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
General Public License for more details.
|
||
|
||
You should have received a copy of the GNU General Public License
|
||
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
"""
|
||
|
||
import logging
|
||
|
||
from lxml import etree
|
||
|
||
from novelwriter.enum import nwItemType, nwItemClass, nwItemLayout
|
||
from novelwriter.common import (
|
||
checkInt, isHandle, isItemClass, isItemLayout, isItemType
|
||
)
|
||
from novelwriter.constants import nwLabels, nwLists, trConst
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class NWItem():
|
||
|
||
def __init__(self, theProject):
|
||
|
||
self.theProject = theProject
|
||
|
||
self.itemName = ""
|
||
self.itemHandle = None
|
||
self.itemParent = None
|
||
self.itemOrder = 0
|
||
self.itemType = nwItemType.NO_TYPE
|
||
self.itemClass = nwItemClass.NO_CLASS
|
||
self.itemLayout = nwItemLayout.NO_LAYOUT
|
||
self.itemStatus = None
|
||
self.isExpanded = False
|
||
self.isExported = True
|
||
|
||
# Document Meta Data
|
||
self.charCount = 0 # Current character count
|
||
self.wordCount = 0 # Current word count
|
||
self.paraCount = 0 # Current paragraph count
|
||
self.initCount = 0 # Initial word count
|
||
self.cursorPos = 0 # Last cursor position
|
||
|
||
return
|
||
|
||
##
|
||
# XML Pack/Unpack
|
||
##
|
||
|
||
def packXML(self, xParent):
|
||
"""Packs all the data in the class instance into an XML object.
|
||
"""
|
||
xPack = etree.SubElement(xParent, "item", attrib={
|
||
"handle": str(self.itemHandle),
|
||
"order": str(self.itemOrder),
|
||
"parent": str(self.itemParent),
|
||
})
|
||
self._subPack(xPack, "name", text=str(self.itemName))
|
||
self._subPack(xPack, "type", text=str(self.itemType.name))
|
||
self._subPack(xPack, "class", text=str(self.itemClass.name))
|
||
self._subPack(xPack, "status", text=str(self.itemStatus))
|
||
if self.itemType == nwItemType.FILE:
|
||
self._subPack(xPack, "exported", text=str(self.isExported))
|
||
self._subPack(xPack, "layout", text=str(self.itemLayout.name))
|
||
self._subPack(xPack, "charCount", text=str(self.charCount), none=False)
|
||
self._subPack(xPack, "wordCount", text=str(self.wordCount), none=False)
|
||
self._subPack(xPack, "paraCount", text=str(self.paraCount), none=False)
|
||
self._subPack(xPack, "cursorPos", text=str(self.cursorPos), none=False)
|
||
else:
|
||
self._subPack(xPack, "expanded", text=str(self.isExpanded))
|
||
|
||
return
|
||
|
||
def unpackXML(self, xItem):
|
||
"""Sets the values from an XML entry of type 'item'.
|
||
"""
|
||
if xItem.tag != "item":
|
||
logger.error("XML entry is not an NWItem")
|
||
return False
|
||
|
||
if "handle" in xItem.attrib:
|
||
self.setHandle(xItem.attrib["handle"])
|
||
else:
|
||
logger.error("XML item entry does not have a handle")
|
||
return False
|
||
|
||
self.setParent(xItem.attrib.get("parent", None))
|
||
self.setOrder(xItem.attrib.get("order", 0))
|
||
|
||
tmpStatus = ""
|
||
for xValue in xItem:
|
||
if xValue.tag == "name":
|
||
self.setName(xValue.text)
|
||
elif xValue.tag == "type":
|
||
self.setType(xValue.text)
|
||
elif xValue.tag == "class":
|
||
self.setClass(xValue.text)
|
||
elif xValue.tag == "layout":
|
||
self.setLayout(xValue.text)
|
||
elif xValue.tag == "status":
|
||
tmpStatus = xValue.text
|
||
elif xValue.tag == "expanded":
|
||
self.setExpanded(xValue.text)
|
||
elif xValue.tag == "exported":
|
||
self.setExported(xValue.text)
|
||
elif xValue.tag == "charCount":
|
||
self.setCharCount(xValue.text)
|
||
elif xValue.tag == "wordCount":
|
||
self.setWordCount(xValue.text)
|
||
elif xValue.tag == "paraCount":
|
||
self.setParaCount(xValue.text)
|
||
elif xValue.tag == "cursorPos":
|
||
self.setCursorPos(xValue.text)
|
||
else:
|
||
# Sliently skip as we may otherwise cause orphaned
|
||
# items if an otherwise valid file is opened by a
|
||
# version of novelWriter that doesn't know the tag.
|
||
logger.error("Unknown tag '%s'", xValue.tag)
|
||
|
||
# Guarantees that <status> is parsed after <class>
|
||
self.setStatus(tmpStatus)
|
||
|
||
return True
|
||
|
||
@staticmethod
|
||
def _subPack(xParent, name, attrib=None, text=None, none=True):
|
||
"""Packs the values into an xml element.
|
||
"""
|
||
if not none and (text is None or text == "None"):
|
||
return None
|
||
xAttr = {} if attrib is None else attrib
|
||
xSub = etree.SubElement(xParent, name, attrib=xAttr)
|
||
if text is not None:
|
||
xSub.text = text
|
||
|
||
return
|
||
|
||
##
|
||
# Methods
|
||
##
|
||
|
||
def describeMe(self, hLevel=None):
|
||
"""Return a string description of the item.
|
||
"""
|
||
descKey = "none"
|
||
if self.itemType == nwItemType.ROOT:
|
||
descKey = "root"
|
||
elif self.itemType == nwItemType.FOLDER:
|
||
descKey = "folder"
|
||
elif self.itemType == nwItemType.FILE:
|
||
if self.itemLayout == nwItemLayout.DOCUMENT:
|
||
if hLevel == "H1":
|
||
descKey = "doc_h1"
|
||
elif hLevel == "H2":
|
||
descKey = "doc_h2"
|
||
elif hLevel == "H3":
|
||
descKey = "doc_h3"
|
||
else:
|
||
descKey = "document"
|
||
elif self.itemLayout == nwItemLayout.NOTE:
|
||
descKey = "note"
|
||
|
||
return trConst(nwLabels.ITEM_DESCRIPTION.get(descKey, ""))
|
||
|
||
##
|
||
# Set Item Values
|
||
##
|
||
|
||
def setName(self, theName):
|
||
"""Set the item name.
|
||
"""
|
||
if isinstance(theName, str):
|
||
self.itemName = theName.strip()
|
||
else:
|
||
self.itemName = ""
|
||
return
|
||
|
||
def setHandle(self, theHandle):
|
||
"""Set the item handle, and ensure it is valid.
|
||
"""
|
||
if isHandle(theHandle):
|
||
self.itemHandle = theHandle
|
||
else:
|
||
self.itemHandle = None
|
||
return
|
||
|
||
def setParent(self, theParent):
|
||
"""Set the parent handle, and ensure that it is valid.
|
||
"""
|
||
if theParent is None:
|
||
self.itemParent = None
|
||
elif isHandle(theParent):
|
||
self.itemParent = theParent
|
||
else:
|
||
self.itemParent = None
|
||
return
|
||
|
||
def setOrder(self, theOrder):
|
||
"""Set the item order, and ensure that it is valid. This value
|
||
is purely a meta value, not actually used by novelWriter.
|
||
"""
|
||
self.itemOrder = checkInt(theOrder, 0)
|
||
return
|
||
|
||
def setType(self, theType):
|
||
"""Set the item type from either a proper nwItemType, or set it
|
||
from a string representing a nwItemType.
|
||
"""
|
||
if isinstance(theType, nwItemType):
|
||
self.itemType = theType
|
||
elif isItemType(theType):
|
||
self.itemType = nwItemType[theType]
|
||
else:
|
||
logger.error("Unrecognised item type '%s'", theType)
|
||
self.itemType = nwItemType.NO_TYPE
|
||
return
|
||
|
||
def setClass(self, theClass):
|
||
"""Set the item class from either a proper nwItemClass, or set
|
||
it from a string representing a nwItemClass.
|
||
"""
|
||
if isinstance(theClass, nwItemClass):
|
||
self.itemClass = theClass
|
||
elif isItemClass(theClass):
|
||
self.itemClass = nwItemClass[theClass]
|
||
else:
|
||
logger.error("Unrecognised item class '%s'", theClass)
|
||
self.itemClass = nwItemClass.NO_CLASS
|
||
return
|
||
|
||
def setLayout(self, theLayout):
|
||
"""Set the item layout from either a proper nwItemLayout, or set
|
||
it from a string representing a nwItemLayout.
|
||
"""
|
||
if isinstance(theLayout, nwItemLayout):
|
||
self.itemLayout = theLayout
|
||
elif isItemLayout(theLayout):
|
||
self.itemLayout = nwItemLayout[theLayout]
|
||
elif theLayout in nwLists.DEP_LAYOUT:
|
||
self.itemLayout = nwItemLayout.DOCUMENT
|
||
else:
|
||
logger.error("Unrecognised item layout '%s'", theLayout)
|
||
self.itemLayout = nwItemLayout.NO_LAYOUT
|
||
return
|
||
|
||
def setStatus(self, theStatus):
|
||
"""Set the item status by looking it up in the valid status
|
||
items of the current project.
|
||
"""
|
||
if self.itemClass in nwLists.CLS_NOVEL:
|
||
self.itemStatus = self.theProject.statusItems.checkEntry(theStatus)
|
||
else:
|
||
self.itemStatus = self.theProject.importItems.checkEntry(theStatus)
|
||
return
|
||
|
||
def setExpanded(self, expState):
|
||
"""Save the expanded status of an item in the project tree.
|
||
"""
|
||
if isinstance(expState, str):
|
||
self.isExpanded = (expState == str(True))
|
||
else:
|
||
self.isExpanded = (expState is True)
|
||
return
|
||
|
||
def setExported(self, expState):
|
||
"""Save the export flag.
|
||
"""
|
||
if isinstance(expState, str):
|
||
self.isExported = (expState == str(True))
|
||
else:
|
||
self.isExported = (expState is True)
|
||
return
|
||
|
||
##
|
||
# Set Document Meta Data
|
||
##
|
||
|
||
def setCharCount(self, theCount):
|
||
"""Set the character count, and ensure that it is an integer.
|
||
"""
|
||
self.charCount = checkInt(theCount, 0)
|
||
return
|
||
|
||
def setWordCount(self, theCount):
|
||
"""Set the word count, and ensure that it is an integer.
|
||
"""
|
||
self.wordCount = checkInt(theCount, 0)
|
||
return
|
||
|
||
def setParaCount(self, theCount):
|
||
"""Set the paragraph count, and ensure that it is an integer.
|
||
"""
|
||
self.paraCount = checkInt(theCount, 0)
|
||
return
|
||
|
||
def setCursorPos(self, thePosition):
|
||
"""Set the cursor position, and ensure that it is an integer.
|
||
"""
|
||
self.cursorPos = checkInt(thePosition, 0)
|
||
return
|
||
|
||
def saveInitialCount(self):
|
||
"""Set the initial word count.
|
||
"""
|
||
self.initCount = self.wordCount
|
||
return
|
||
|
||
# END Class NWItem
|