Veronica K. B. Olsen
98 lines
2.4 KiB
Python
98 lines
2.4 KiB
Python
# -*- coding: utf-8 -*
|
||
"""novelWriter Project Wrapper
|
||
|
||
novelWriter – Project Wrapper
|
||
===============================
|
||
Class holding a project
|
||
|
||
File History:
|
||
Created: 2018-09-29 [0.1.0]
|
||
|
||
"""
|
||
|
||
import logging
|
||
import nw
|
||
|
||
from os import path, mkdir
|
||
from lxml import etree
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
class NWProject():
|
||
|
||
def __init__(self):
|
||
|
||
self.projFolder = None
|
||
self.projTree = []
|
||
|
||
# Project Settings
|
||
self.projPath = ""
|
||
self.projData = ""
|
||
self.projName = ""
|
||
self.bookTitle = ""
|
||
self.bookAuthors = []
|
||
|
||
return
|
||
|
||
def buildProjectTree(self):
|
||
|
||
return True
|
||
|
||
def saveProject(self):
|
||
|
||
projDir = path.dirname(self.projPath)
|
||
projFile = path.basename(self.projPath)
|
||
logger.vverbose("Project folder is %s" % projDir)
|
||
logger.vverbose("Project file is %s" % projFile)
|
||
|
||
if bookFile[-4:] == ".nwx":
|
||
self.projData = path.join(projDir,projFile[:-4]+".nwd")
|
||
if not path.isdir(self.projData):
|
||
logger.info("Created folder %s" % self.projData)
|
||
mkdir(self.projData)
|
||
|
||
# Root element and book details
|
||
nwXML = etree.Element("novelWriterXML",attrib={
|
||
"fileVersion" : "1.0",
|
||
"appVersion" : str(nw.__version__),
|
||
"timeStamp" : getTimeStamp("-"),
|
||
})
|
||
xBook = etree.SubElement(nwXML,"book")
|
||
xBookTitle = etree.SubElement(xBook,"title")
|
||
xBookTitle.text = self.bookTitle
|
||
for bookAuthor in self.bookAuthors:
|
||
if bookAuthor == "": continue
|
||
xBookAuthor = etree.SubElement(xBook,"author")
|
||
xBookAuthor.text = bookAuthor
|
||
|
||
# Write the xml tree to file
|
||
with open(self.bookPath,"wb") as outFile:
|
||
outFile.write(etree.tostring(
|
||
nwXML,
|
||
pretty_print = True,
|
||
encoding = "utf-8",
|
||
xml_declaration = True
|
||
))
|
||
|
||
return True
|
||
|
||
#
|
||
# Set Functions
|
||
#
|
||
|
||
def setProjectName(self, projName):
|
||
self.projName = projName.strip()
|
||
return True
|
||
|
||
def setBookTitle(self, bookTitle):
|
||
self.bookTitle = bookTitle.strip()
|
||
return True
|
||
|
||
def setBookAuthors(self, bookAuthors):
|
||
self.bookAuthors = []
|
||
for bookAuthor in bookAuthors.split(","):
|
||
self.bookAuthors.append(bookAuthor.strip())
|
||
return True
|
||
|
||
# END Class NWProject
|