Veronica Berglyd Olsen
296 lines
9.7 KiB
Python
296 lines
9.7 KiB
Python
"""
|
||
novelWriter – Project Document
|
||
==============================
|
||
Data class for a single novelWriter document
|
||
|
||
File History:
|
||
Created: 2018-09-29 [0.0.1]
|
||
|
||
This file is a part of novelWriter
|
||
Copyright 2018–2023, 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/>.
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
|
||
from typing import TYPE_CHECKING
|
||
from pathlib import Path
|
||
from novelwriter.core.item import NWItem
|
||
|
||
from novelwriter.enum import nwItemLayout, nwItemClass
|
||
from novelwriter.error import formatException
|
||
from novelwriter.common import isHandle, sha256sum
|
||
|
||
if TYPE_CHECKING: # pragma: no cover
|
||
from novelwriter.core.project import NWProject
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class NWDocument:
|
||
|
||
def __init__(self, project: NWProject, tHandle: str) -> None:
|
||
|
||
self._project = project
|
||
|
||
# Internal Variables
|
||
self._theItem = None # The currently open item
|
||
self._docHandle = None # The handle of the currently open item
|
||
self._fileLoc = None # The file location of the currently open item
|
||
self._docMeta = {} # The meta data of the currently open item
|
||
self._docError = "" # The latest encountered IO error
|
||
self._prevHash = None # Previous sha256sum of the document file
|
||
self._currHash = None # Latest sha256sum of the document file
|
||
|
||
if isHandle(tHandle):
|
||
self._docHandle = tHandle
|
||
|
||
if self._docHandle is not None:
|
||
self._theItem = self._project.tree[tHandle]
|
||
|
||
return
|
||
|
||
def __repr__(self) -> str:
|
||
return f"<NWDocument handle={self._docHandle}>"
|
||
|
||
def __bool__(self) -> bool:
|
||
return self._docHandle is not None and bool(self._theItem)
|
||
|
||
##
|
||
# Class Methods
|
||
##
|
||
|
||
def fileExists(self) -> bool:
|
||
"""Check if the document file exists."""
|
||
if self._docHandle is None:
|
||
return False
|
||
|
||
contentPath = self._project.storage.contentPath
|
||
if not isinstance(contentPath, Path):
|
||
logger.error("No content path set")
|
||
return False
|
||
|
||
return (contentPath / f"{self._docHandle}.nwd").is_file()
|
||
|
||
def readDocument(self, isOrphan: bool = False) -> str | None:
|
||
"""Read the document specified by the handle set in the
|
||
contructor, capturing potential file system errors and parse
|
||
meta data. If the document doesn't exist on disk, return an
|
||
empty string. If something went wrong, return None.
|
||
"""
|
||
self._docError = ""
|
||
if not isinstance(self._docHandle, str):
|
||
logger.error("No document handle set")
|
||
return None
|
||
|
||
if self._theItem is None and not isOrphan:
|
||
logger.error("Unknown novelWriter document")
|
||
return None
|
||
|
||
contentPath = self._project.storage.contentPath
|
||
if not isinstance(contentPath, Path):
|
||
logger.error("No content path set")
|
||
return None
|
||
|
||
docFile = f"{self._docHandle}.nwd"
|
||
logger.debug("Opening document: %s", docFile)
|
||
|
||
docPath = contentPath / docFile
|
||
self._fileLoc = docPath
|
||
|
||
theText = ""
|
||
self._docMeta = {}
|
||
self._prevHash = None
|
||
|
||
if docPath.exists():
|
||
self._prevHash = sha256sum(docPath)
|
||
try:
|
||
with open(docPath, mode="r", encoding="utf-8") as inFile:
|
||
# Check the first <= 10 lines for metadata
|
||
for i in range(10):
|
||
inLine = inFile.readline()
|
||
if inLine.startswith(r"%%~"):
|
||
self._parseMeta(inLine)
|
||
else:
|
||
theText = inLine
|
||
break
|
||
|
||
# Load the rest of the file
|
||
theText += inFile.read()
|
||
|
||
except Exception as exc:
|
||
self._docError = formatException(exc)
|
||
return None
|
||
|
||
else:
|
||
# The document file does not exist, so we assume it's a new
|
||
# document and initialise an empty text string.
|
||
logger.debug("The requested document does not exist")
|
||
return ""
|
||
|
||
return theText
|
||
|
||
def writeDocument(self, docText: str, forceWrite: bool = False) -> bool:
|
||
"""Write the document specified by the handle attribute. Handle
|
||
any IO errors in the process Returns True if successful, False
|
||
if not.
|
||
"""
|
||
self._docError = ""
|
||
if not isinstance(self._docHandle, str):
|
||
logger.error("No document handle set")
|
||
return False
|
||
|
||
contentPath = self._project.storage.contentPath
|
||
if not isinstance(contentPath, Path):
|
||
logger.error("No content path set")
|
||
return False
|
||
|
||
docFile = f"{self._docHandle}.nwd"
|
||
logger.debug("Saving document: %s", docFile)
|
||
|
||
docPath = contentPath / docFile
|
||
docTemp = docPath.with_suffix(".tmp")
|
||
|
||
if self._prevHash is not None and not forceWrite:
|
||
self._currHash = sha256sum(docPath)
|
||
if self._currHash is not None and self._currHash != self._prevHash:
|
||
logger.error("File has been altered on disk since opened")
|
||
return False
|
||
|
||
# DocMeta Line
|
||
if self._theItem is None:
|
||
docMeta = ""
|
||
else:
|
||
docMeta = (
|
||
f"%%~name: {self._theItem.itemName}\n"
|
||
f"%%~path: {self._theItem.itemParent}/{self._theItem.itemHandle}\n"
|
||
f"%%~kind: {self._theItem.itemClass.name}/{self._theItem.itemLayout.name}\n"
|
||
)
|
||
|
||
try:
|
||
with open(docTemp, mode="w", encoding="utf-8") as outFile:
|
||
outFile.write(docMeta)
|
||
outFile.write(docText)
|
||
except Exception as exc:
|
||
self._docError = formatException(exc)
|
||
return False
|
||
|
||
# If we're here, the file was successfully saved, so we can
|
||
# replace the temp file with the actual file
|
||
try:
|
||
docTemp.replace(docPath)
|
||
except OSError as exc:
|
||
self._docError = formatException(exc)
|
||
return False
|
||
|
||
self._prevHash = sha256sum(docPath)
|
||
self._currHash = self._prevHash
|
||
|
||
return True
|
||
|
||
def deleteDocument(self) -> bool:
|
||
"""Permanently delete a document source file and related files
|
||
from the project data folder.
|
||
"""
|
||
self._docError = ""
|
||
if not isinstance(self._docHandle, str):
|
||
logger.error("No document handle set")
|
||
return False
|
||
|
||
contentPath = self._project.storage.contentPath
|
||
if not isinstance(contentPath, Path):
|
||
logger.error("No content path set")
|
||
return False
|
||
|
||
docPath = contentPath / f"{self._docHandle}.nwd"
|
||
docTemp = docPath.with_suffix(".tmp")
|
||
|
||
try:
|
||
# ToDo: When Python 3.7 is dropped, these can be changed to
|
||
# path.unlink(missing_ok=True)
|
||
if docPath.exists():
|
||
docPath.unlink()
|
||
if docTemp.exists():
|
||
docTemp.unlink()
|
||
except Exception as exc:
|
||
self._docError = formatException(exc)
|
||
return False
|
||
|
||
return True
|
||
|
||
##
|
||
# Getters
|
||
##
|
||
|
||
def getFileLocation(self) -> str:
|
||
"""Return the file location of the current document."""
|
||
return str(self._fileLoc)
|
||
|
||
def getCurrentItem(self) -> NWItem | None:
|
||
"""Return a pointer to the currently open NWItem."""
|
||
return self._theItem
|
||
|
||
def getMeta(self) -> tuple[str, str | None, str | None, str | None]:
|
||
"""Parse the document meta tag and return the name, parent,
|
||
class and layout meta values.
|
||
"""
|
||
theName = self._docMeta.get("name", "")
|
||
theParent = self._docMeta.get("parent", None)
|
||
theClass = self._docMeta.get("class", None)
|
||
theLayout = self._docMeta.get("layout", None)
|
||
|
||
return theName, theParent, theClass, theLayout
|
||
|
||
def getError(self) -> str:
|
||
"""Return the last recorded exception."""
|
||
return self._docError
|
||
|
||
##
|
||
# Internal Functions
|
||
##
|
||
|
||
def _parseMeta(self, metaLine: str) -> None:
|
||
"""Parse a line from the document starting with the characters
|
||
%%~ that may contain meta data.
|
||
"""
|
||
if metaLine.startswith("%%~name:"):
|
||
self._docMeta["name"] = metaLine[8:].strip()
|
||
|
||
elif metaLine.startswith("%%~path:"):
|
||
metaVal = metaLine[8:].strip()
|
||
metaBits = metaVal.split("/")
|
||
if len(metaBits) == 2:
|
||
if isHandle(metaBits[0]):
|
||
self._docMeta["parent"] = metaBits[0]
|
||
if isHandle(metaBits[1]):
|
||
self._docMeta["handle"] = metaBits[1]
|
||
|
||
elif metaLine.startswith("%%~kind:"):
|
||
metaVal = metaLine[8:].strip()
|
||
metaBits = metaVal.split("/")
|
||
if len(metaBits) == 2:
|
||
if metaBits[0] in nwItemClass.__members__:
|
||
self._docMeta["class"] = nwItemClass[metaBits[0]]
|
||
if metaBits[1] in nwItemLayout.__members__:
|
||
self._docMeta["layout"] = nwItemLayout[metaBits[1]]
|
||
|
||
else:
|
||
logger.debug("Ignoring meta data: '%s'", metaLine.strip())
|
||
|
||
return
|
||
|
||
# END Class NWDocument
|