Veronica K. B. Olsen
375 lines
12 KiB
Python
375 lines
12 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""novelWriter GUI Export Tools
|
||
|
||
novelWriter – GUI Export Tools
|
||
================================
|
||
Tool for exporting project files to other formats
|
||
|
||
File History:
|
||
Created: 2019-10-13 [0.2.3]
|
||
|
||
"""
|
||
|
||
import logging
|
||
import json
|
||
import nw
|
||
|
||
from os import path
|
||
|
||
from PyQt5.QtCore import Qt, QSize
|
||
from PyQt5.QtSvg import QSvgWidget
|
||
from PyQt5.QtWidgets import (
|
||
QDialog, QHBoxLayout, QVBoxLayout, QWidget, QTabWidget, QDialogButtonBox, QGridLayout,
|
||
QGroupBox, QCheckBox, QLabel, QComboBox, QLineEdit, QPushButton, QFileDialog
|
||
)
|
||
|
||
from nw.tools.translate import numberToWord
|
||
from nw.common import checkString, checkBool, checkInt
|
||
from nw.constants import nwFiles
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
class GuiExport(QDialog):
|
||
|
||
def __init__(self, theParent, theProject):
|
||
QDialog.__init__(self, theParent)
|
||
|
||
logger.debug("Initialising GuiExport ...")
|
||
|
||
self.mainConf = nw.CONFIG
|
||
self.theParent = theParent
|
||
self.theProject = theProject
|
||
self.optState = ExportLastState(self.theProject)
|
||
|
||
self.outerBox = QHBoxLayout()
|
||
self.innerBox = QVBoxLayout()
|
||
self.setWindowTitle("Export Project")
|
||
self.setLayout(self.outerBox)
|
||
|
||
self.gradPath = path.abspath(path.join(self.mainConf.appPath,"graphics","gear.svg"))
|
||
self.svgGradient = QSvgWidget(self.gradPath)
|
||
self.svgGradient.setFixedSize(QSize(64,64))
|
||
|
||
self.theProject.countStatus()
|
||
self.tabMain = GuiExportMain(self.theParent, self.theProject, self.optState)
|
||
|
||
self.tabWidget = QTabWidget()
|
||
self.tabWidget.addTab(self.tabMain, "Settings")
|
||
|
||
self.outerBox.addWidget(self.svgGradient, 0, Qt.AlignTop)
|
||
self.outerBox.addLayout(self.innerBox)
|
||
|
||
self.buttonBox = QDialogButtonBox(QDialogButtonBox.Save | QDialogButtonBox.Cancel)
|
||
self.buttonBox.accepted.connect(self._doSave)
|
||
self.buttonBox.rejected.connect(self._doClose)
|
||
|
||
self.innerBox.addWidget(self.tabWidget)
|
||
self.innerBox.addWidget(self.buttonBox)
|
||
|
||
self.show()
|
||
|
||
logger.debug("GuiExport initialisation complete")
|
||
|
||
return
|
||
|
||
##
|
||
# Buttons
|
||
##
|
||
|
||
def _doSave(self):
|
||
|
||
logger.verbose("GuiExport save button clicked")
|
||
|
||
wNovel = self.tabMain.expNovel.isChecked()
|
||
wNotes = self.tabMain.expNotes.isChecked()
|
||
wTOC = self.tabMain.expTOC.isChecked()
|
||
eFormat = self.tabMain.outputFormat.currentData()
|
||
wComments = self.tabMain.outputComments.isChecked()
|
||
chFormat = self.tabMain.chapterFormat.text()
|
||
scFormat = self.tabMain.sceneFormat.text()
|
||
saveTo = self.tabMain.exportPath.text()
|
||
|
||
self.optState.setSetting("wNovel", wNovel)
|
||
self.optState.setSetting("wNotes", wNotes)
|
||
self.optState.setSetting("wTOC", wTOC)
|
||
self.optState.setSetting("eFormat", eFormat)
|
||
self.optState.setSetting("wComments",wComments)
|
||
self.optState.setSetting("chFormat", chFormat)
|
||
self.optState.setSetting("scFormat", scFormat)
|
||
self.optState.setSetting("saveTo", saveTo)
|
||
|
||
self.optState.saveSettings()
|
||
self.close()
|
||
|
||
return
|
||
|
||
def _doClose(self):
|
||
logger.verbose("GuiExport close button clicked")
|
||
self.close()
|
||
return
|
||
|
||
# END Class GuiExport
|
||
|
||
class GuiExportMain(QWidget):
|
||
|
||
FMT_MD = 1
|
||
FMT_HTML = 2
|
||
FMT_EBOOK = 3
|
||
FMT_ODT = 4
|
||
FMT_TEX = 5
|
||
FMT_HELP = {
|
||
FMT_MD : (
|
||
"Exports a standard markdown file. "
|
||
"Comments are converted to preformatted text blocks."
|
||
),
|
||
FMT_HTML : (
|
||
"Exports a plain html5 file. "
|
||
"Comments are converted to preformatted text blocks."
|
||
),
|
||
FMT_EBOOK : (
|
||
"Exports an html5 file that can be converted to eBook with Calibre. "
|
||
"Comments are not exported in this format."
|
||
),
|
||
FMT_ODT : (
|
||
"Exports an open document file that can be read by office applications. "
|
||
"Comments are exported as grey text."
|
||
),
|
||
FMT_TEX : (
|
||
"Exports a LaTeX file that can be compiled to PDF using for instance PDFLaTeX. "
|
||
"Comments are exported as LaTeX comments."
|
||
),
|
||
}
|
||
|
||
def __init__(self, theParent, theProject, optState):
|
||
QWidget.__init__(self, theParent)
|
||
|
||
self.theParent = theParent
|
||
self.theProject = theProject
|
||
self.theTheme = theParent.theTheme
|
||
self.outerBox = QGridLayout()
|
||
self.optState = optState
|
||
|
||
# Select Files
|
||
self.guiFiles = QGroupBox("Export Files", self)
|
||
self.guiFilesForm = QGridLayout(self)
|
||
self.guiFiles.setLayout(self.guiFilesForm)
|
||
|
||
self.expNovel = QCheckBox(self)
|
||
self.expNotes = QCheckBox(self)
|
||
self.expTOC = QCheckBox(self)
|
||
self.expNovel.setToolTip("Include all novel files in the exported document")
|
||
self.expNotes.setToolTip("Include all note files in the exported document")
|
||
self.expTOC.setToolTip("Generate a Table of Contents (ToC)")
|
||
self.expNovel.setChecked(self.optState.wNovel())
|
||
self.expNotes.setChecked(self.optState.wNotes())
|
||
self.expTOC.setChecked(self.optState.wTOC())
|
||
|
||
self.guiFilesForm.addWidget(QLabel("Novel files"), 0, 0)
|
||
self.guiFilesForm.addWidget(self.expNovel, 0, 1)
|
||
self.guiFilesForm.addWidget(QLabel("Note files"), 1, 0)
|
||
self.guiFilesForm.addWidget(self.expNotes, 1, 1)
|
||
self.guiFilesForm.addWidget(QLabel("Contents"), 2, 0)
|
||
self.guiFilesForm.addWidget(self.expTOC, 2, 1)
|
||
|
||
# Chapter Settings
|
||
self.guiChapters = QGroupBox("Chapters", self)
|
||
self.guiChaptersForm = QGridLayout(self)
|
||
self.guiChapters.setLayout(self.guiChaptersForm)
|
||
|
||
self.chapterFormat = QLineEdit()
|
||
self.chapterFormat.setText(self.optState.chFormat())
|
||
self.chapterFormat.setToolTip("Available formats: %num%, %numword%, %title%, %label%")
|
||
self.chapterFormat.setMinimumWidth(250)
|
||
|
||
self.guiChaptersForm.addWidget(QLabel("Format"), 0, 0)
|
||
self.guiChaptersForm.addWidget(self.chapterFormat, 0, 1)
|
||
|
||
# Output Format
|
||
self.guiOutput = QGroupBox("Output", self)
|
||
self.guiOutputForm = QGridLayout(self)
|
||
self.guiOutput.setLayout(self.guiOutputForm)
|
||
|
||
self.outputComments = QCheckBox("include comments", self)
|
||
self.outputComments.setChecked(self.optState.wComments())
|
||
|
||
self.outputHelp = QLabel("")
|
||
self.outputHelp.setWordWrap(True)
|
||
self.outputHelp.setMinimumHeight(55)
|
||
self.outputHelp.setAlignment(Qt.AlignTop)
|
||
|
||
self.outputFormat = QComboBox(self)
|
||
self.outputFormat.addItem("Markdown", self.FMT_MD)
|
||
self.outputFormat.addItem("HTML5 (Plain)", self.FMT_HTML)
|
||
self.outputFormat.addItem("HTML5 (eBook)", self.FMT_EBOOK)
|
||
self.outputFormat.addItem("Open Document", self.FMT_ODT)
|
||
self.outputFormat.addItem("LaTeX (PDF)", self.FMT_TEX)
|
||
self.outputFormat.currentIndexChanged.connect(self._updateFormatHelp)
|
||
|
||
optIdx = self.outputFormat.findData(self.optState.eFormat())
|
||
if optIdx != -1:
|
||
self.outputFormat.setCurrentIndex(optIdx)
|
||
self._updateFormatHelp(optIdx)
|
||
|
||
self.guiOutputForm.addWidget(QLabel("Export format"), 0, 0)
|
||
self.guiOutputForm.addWidget(self.outputFormat, 0, 1)
|
||
self.guiOutputForm.addWidget(self.outputComments, 0, 2)
|
||
self.guiOutputForm.addWidget(self.outputHelp, 1, 0, 1, 3)
|
||
self.guiOutputForm.setColumnStretch(2, 1)
|
||
|
||
# Scene Settings
|
||
self.guiScenes = QGroupBox("Scenes", self)
|
||
self.guiScenesForm = QGridLayout(self)
|
||
self.guiScenes.setLayout(self.guiScenesForm)
|
||
|
||
self.sceneFormat = QLineEdit()
|
||
self.sceneFormat.setText(self.optState.scFormat())
|
||
self.sceneFormat.setToolTip("Available formats: %title%")
|
||
self.sceneFormat.setMinimumWidth(100)
|
||
|
||
self.guiScenesForm.addWidget(QLabel("Format"), 0, 0)
|
||
self.guiScenesForm.addWidget(self.sceneFormat, 0, 1)
|
||
|
||
# Output Path
|
||
self.exportTo = QGroupBox("Backup", self)
|
||
self.exportToForm = QGridLayout(self)
|
||
self.exportTo.setLayout(self.exportToForm)
|
||
|
||
self.exportPath = QLineEdit(self.optState.saveTo())
|
||
|
||
self.exportGetPath = QPushButton(self.theTheme.getIcon("folder"),"")
|
||
self.exportGetPath.clicked.connect(self._exportFolder)
|
||
|
||
self.exportToForm.addWidget(QLabel("Save to"), 0, 0)
|
||
self.exportToForm.addWidget(self.exportPath, 0, 1)
|
||
self.exportToForm.addWidget(self.exportGetPath, 0, 2)
|
||
|
||
# Assemble
|
||
self.outerBox.addWidget(self.guiFiles, 0, 0)
|
||
self.outerBox.addWidget(self.guiOutput, 0, 1, 1, 2)
|
||
self.outerBox.addWidget(self.guiChapters, 1, 0, 1, 2)
|
||
self.outerBox.addWidget(self.guiScenes, 1, 2)
|
||
self.outerBox.addWidget(self.exportTo, 2, 0, 1, 3)
|
||
self.outerBox.setColumnStretch(0, 1)
|
||
self.outerBox.setColumnStretch(1, 1)
|
||
self.outerBox.setColumnStretch(2, 1)
|
||
self.setLayout(self.outerBox)
|
||
|
||
return
|
||
|
||
##
|
||
# Internal Functions
|
||
##
|
||
|
||
def _updateFormatHelp(self, currIdx):
|
||
"""Update help text under output format selection.
|
||
"""
|
||
if currIdx == -1:
|
||
self.outputHelp.setText("")
|
||
else:
|
||
fmtIdx = self.outputFormat.itemData(currIdx)
|
||
self.outputHelp.setText("<i>%s</i>" % self.FMT_HELP[fmtIdx])
|
||
|
||
return
|
||
|
||
def _exportFolder(self):
|
||
|
||
currDir = self.exportPath.text()
|
||
if not path.isdir(currDir):
|
||
currDir = ""
|
||
|
||
dlgOpt = QFileDialog.Options()
|
||
dlgOpt |= QFileDialog.ShowDirsOnly
|
||
dlgOpt |= QFileDialog.DontUseNativeDialog
|
||
newDir = QFileDialog.getExistingDirectory(
|
||
self,"Export Directory",currDir,options=dlgOpt
|
||
)
|
||
if newDir:
|
||
self.exportPath.setText(newDir)
|
||
return True
|
||
|
||
return False
|
||
|
||
# END Class GuiExportMain
|
||
|
||
class ExportLastState():
|
||
|
||
def __init__(self, theProject):
|
||
|
||
self.theProject = theProject
|
||
self.theState = {
|
||
"wNovel" : True,
|
||
"wNotes" : False,
|
||
"wTOC" : True,
|
||
"eFormat" : 1,
|
||
"wComments" : False,
|
||
"chFormat" : "Chapter %numword%",
|
||
"scFormat" : "* * *",
|
||
"saveTo" : "",
|
||
}
|
||
self.loadSettings()
|
||
|
||
return
|
||
|
||
def loadSettings(self):
|
||
|
||
stateFile = path.join(self.theProject.projMeta, nwFiles.EXPORT_OPT)
|
||
if path.isfile(stateFile):
|
||
logger.debug("Loading export options file")
|
||
try:
|
||
with open(stateFile,mode="r") as inFile:
|
||
theJson = inFile.read()
|
||
self.theState = json.loads(theJson)
|
||
except Exception as e:
|
||
logger.error("Failed to load export options file")
|
||
logger.error(str(e))
|
||
return False
|
||
|
||
return True
|
||
|
||
def saveSettings(self):
|
||
|
||
stateFile = path.join(self.theProject.projMeta, nwFiles.EXPORT_OPT)
|
||
logger.debug("Saving export options file")
|
||
try:
|
||
with open(stateFile,mode="w+") as outFile:
|
||
outFile.write(json.dumps(self.theState, indent=2))
|
||
except Exception as e:
|
||
logger.error("Failed to save export options file")
|
||
logger.error(str(e))
|
||
return False
|
||
|
||
return True
|
||
|
||
def setSetting(self, setName, setValue):
|
||
if setName in self.theState:
|
||
self.theState[setName] = setValue
|
||
else:
|
||
return False
|
||
return True
|
||
|
||
def wNovel(self):
|
||
return checkBool(self.theState["wNovel"],False,False)
|
||
|
||
def wNotes(self):
|
||
return checkBool(self.theState["wNotes"],False,False)
|
||
|
||
def wTOC(self):
|
||
return checkBool(self.theState["wTOC"],False,False)
|
||
|
||
def eFormat(self):
|
||
return checkInt(self.theState["eFormat"],1,False)
|
||
|
||
def wComments(self):
|
||
return checkBool(self.theState["wComments"],False,False)
|
||
|
||
def chFormat(self):
|
||
return checkString(self.theState["chFormat"],"Chapter %numword%",False)
|
||
|
||
def scFormat(self):
|
||
return checkString(self.theState["scFormat"],"* * *",False)
|
||
|
||
def saveTo(self):
|
||
return checkString(self.theState["saveTo"],"",False)
|
||
|
||
# END Class ExportLastState
|