Veronica K. B. Olsen
1326 lines
46 KiB
Python
1326 lines
46 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
novelWriter – GUI Build Novel Project
|
||
=====================================
|
||
GUI classes for the build novel project dialog
|
||
|
||
File History:
|
||
Created: 2020-05-09 [0.5]
|
||
|
||
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 nw
|
||
import logging
|
||
import json
|
||
import os
|
||
|
||
from time import time
|
||
from datetime import datetime
|
||
|
||
from PyQt5.QtCore import Qt, QTimer
|
||
from PyQt5.QtPrintSupport import QPrinter, QPrintPreviewDialog
|
||
from PyQt5.QtGui import (
|
||
QPalette, QColor, QFont, QCursor, QFontInfo
|
||
)
|
||
from PyQt5.QtWidgets import (
|
||
qApp, QDialog, QVBoxLayout, QHBoxLayout, QTextBrowser, QPushButton, QLabel,
|
||
QLineEdit, QGroupBox, QGridLayout, QProgressBar, QMenu, QAction,
|
||
QFileDialog, QFontDialog, QSpinBox, QScrollArea, QSplitter, QWidget,
|
||
QSizePolicy
|
||
)
|
||
|
||
from nw.common import fuzzyTime, makeFileNameSafe
|
||
from nw.gui.custom import QSwitch
|
||
from nw.core import ToHtml, ToOdt, ToMarkdown
|
||
from nw.constants import (
|
||
nwConst, nwAlert, nwFiles, nwItemType, nwItemLayout, nwItemClass
|
||
)
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
class GuiBuildNovel(QDialog):
|
||
|
||
FMT_ODT = 1
|
||
FMT_FODT = 2
|
||
FMT_PDF = 3
|
||
FMT_HTM = 4
|
||
FMT_MD = 5
|
||
FMT_GH = 6
|
||
FMT_NWD = 7
|
||
FMT_JSON_H = 8
|
||
FMT_JSON_M = 9
|
||
|
||
def __init__(self, theParent, theProject):
|
||
QDialog.__init__(self, theParent)
|
||
|
||
logger.debug("Initialising GuiBuildNovel ...")
|
||
self.setObjectName("GuiBuildNovel")
|
||
|
||
self.mainConf = nw.CONFIG
|
||
self.theProject = theProject
|
||
self.theParent = theParent
|
||
self.theTheme = theParent.theTheme
|
||
self.optState = self.theProject.optState
|
||
|
||
self.htmlText = [] # List of html documents
|
||
self.htmlStyle = [] # List of html styles
|
||
self.htmlSize = 0 # Size of the html document
|
||
self.buildTime = 0 # The timestamp of the last build
|
||
|
||
self.setWindowTitle("Build Novel Project")
|
||
self.setMinimumWidth(self.mainConf.pxInt(700))
|
||
self.setMinimumHeight(self.mainConf.pxInt(600))
|
||
|
||
self.resize(
|
||
self.mainConf.pxInt(self.optState.getInt("GuiBuildNovel", "winWidth", 900)),
|
||
self.mainConf.pxInt(self.optState.getInt("GuiBuildNovel", "winHeight", 800))
|
||
)
|
||
|
||
self.docView = GuiBuildNovelDocView(self, self.theProject)
|
||
|
||
# Title Formats
|
||
# =============
|
||
|
||
self.titleGroup = QGroupBox("Title Formats for Novel Files", self)
|
||
self.titleForm = QGridLayout(self)
|
||
self.titleGroup.setLayout(self.titleForm)
|
||
|
||
fmtHelp = (
|
||
r"<b>Formatting Codes:</b><br>"
|
||
r"%title% for the title as set in the document<br>"
|
||
r"%ch% for chapter number (1, 2, 3)<br>"
|
||
r"%chw% for chapter number as a word (one, two)<br>"
|
||
r"%chI% for chapter number in upper case Roman<br>"
|
||
r"%chi% for chapter number in lower case Roman<br>"
|
||
r"%sc% for scene number within chapter<br>"
|
||
r"%sca% for scene number within novel"
|
||
)
|
||
fmtScHelp = (
|
||
r"<br><br>"
|
||
r"Leave blank to skip this heading, or set to a static text, like "
|
||
r"for instance '* * *', to make a separator. The separator will "
|
||
r"be centred automatically and only appear between sections of "
|
||
r"the same type."
|
||
)
|
||
xFmt = self.mainConf.pxInt(100)
|
||
|
||
self.fmtTitle = QLineEdit()
|
||
self.fmtTitle.setMaxLength(200)
|
||
self.fmtTitle.setMinimumWidth(xFmt)
|
||
self.fmtTitle.setToolTip(fmtHelp)
|
||
self.fmtTitle.setText(
|
||
self._reFmtCodes(self.theProject.titleFormat["title"])
|
||
)
|
||
|
||
self.fmtChapter = QLineEdit()
|
||
self.fmtChapter.setMaxLength(200)
|
||
self.fmtChapter.setMinimumWidth(xFmt)
|
||
self.fmtChapter.setToolTip(fmtHelp)
|
||
self.fmtChapter.setText(
|
||
self._reFmtCodes(self.theProject.titleFormat["chapter"])
|
||
)
|
||
|
||
self.fmtUnnumbered = QLineEdit()
|
||
self.fmtUnnumbered.setMaxLength(200)
|
||
self.fmtUnnumbered.setMinimumWidth(xFmt)
|
||
self.fmtUnnumbered.setToolTip(fmtHelp)
|
||
self.fmtUnnumbered.setText(
|
||
self._reFmtCodes(self.theProject.titleFormat["unnumbered"])
|
||
)
|
||
|
||
self.fmtScene = QLineEdit()
|
||
self.fmtScene.setMaxLength(200)
|
||
self.fmtScene.setMinimumWidth(xFmt)
|
||
self.fmtScene.setToolTip(fmtHelp + fmtScHelp)
|
||
self.fmtScene.setText(
|
||
self._reFmtCodes(self.theProject.titleFormat["scene"])
|
||
)
|
||
|
||
self.fmtSection = QLineEdit()
|
||
self.fmtSection.setMaxLength(200)
|
||
self.fmtSection.setMinimumWidth(xFmt)
|
||
self.fmtSection.setToolTip(fmtHelp + fmtScHelp)
|
||
self.fmtSection.setText(
|
||
self._reFmtCodes(self.theProject.titleFormat["section"])
|
||
)
|
||
|
||
# Dummy boxes due to QGridView and QLineEdit expand bug
|
||
self.boxTitle = QHBoxLayout()
|
||
self.boxTitle.addWidget(self.fmtTitle)
|
||
self.boxChapter = QHBoxLayout()
|
||
self.boxChapter.addWidget(self.fmtChapter)
|
||
self.boxUnnumbered = QHBoxLayout()
|
||
self.boxUnnumbered.addWidget(self.fmtUnnumbered)
|
||
self.boxScene = QHBoxLayout()
|
||
self.boxScene.addWidget(self.fmtScene)
|
||
self.boxSection = QHBoxLayout()
|
||
self.boxSection.addWidget(self.fmtSection)
|
||
|
||
self.titleForm.addWidget(QLabel("Title"), 0, 0, 1, 1, Qt.AlignLeft)
|
||
self.titleForm.addLayout(self.boxTitle, 0, 1, 1, 1, Qt.AlignRight)
|
||
self.titleForm.addWidget(QLabel("Chapter"), 1, 0, 1, 1, Qt.AlignLeft)
|
||
self.titleForm.addLayout(self.boxChapter, 1, 1, 1, 1, Qt.AlignRight)
|
||
self.titleForm.addWidget(QLabel("Unnumbered"), 2, 0, 1, 1, Qt.AlignLeft)
|
||
self.titleForm.addLayout(self.boxUnnumbered, 2, 1, 1, 1, Qt.AlignRight)
|
||
self.titleForm.addWidget(QLabel("Scene"), 3, 0, 1, 1, Qt.AlignLeft)
|
||
self.titleForm.addLayout(self.boxScene, 3, 1, 1, 1, Qt.AlignRight)
|
||
self.titleForm.addWidget(QLabel("Section"), 4, 0, 1, 1, Qt.AlignLeft)
|
||
self.titleForm.addLayout(self.boxSection, 4, 1, 1, 1, Qt.AlignRight)
|
||
|
||
self.titleForm.setColumnStretch(0, 0)
|
||
self.titleForm.setColumnStretch(1, 1)
|
||
|
||
# Text Options
|
||
# =============
|
||
|
||
self.formatGroup = QGroupBox("Formatting Options", self)
|
||
self.formatForm = QGridLayout(self)
|
||
self.formatGroup.setLayout(self.formatForm)
|
||
|
||
## Font Family
|
||
self.textFont = QLineEdit()
|
||
self.textFont.setReadOnly(True)
|
||
self.textFont.setMinimumWidth(xFmt)
|
||
self.textFont.setText(
|
||
self.optState.getString("GuiBuildNovel", "textFont", self.mainConf.textFont)
|
||
)
|
||
self.fontButton = QPushButton("...")
|
||
self.fontButton.setMaximumWidth(int(2.5*self.theTheme.getTextWidth("...")))
|
||
self.fontButton.clicked.connect(self._selectFont)
|
||
|
||
self.textSize = QSpinBox(self)
|
||
self.textSize.setFixedWidth(5*self.theTheme.textNWidth)
|
||
self.textSize.setMinimum(6)
|
||
self.textSize.setMaximum(72)
|
||
self.textSize.setSingleStep(1)
|
||
self.textSize.setToolTip(
|
||
"The size is used for PDF and printing. Other formats have no size set."
|
||
)
|
||
self.textSize.setValue(
|
||
self.optState.getInt("GuiBuildNovel", "textSize", self.mainConf.textSize)
|
||
)
|
||
|
||
self.justifyText = QSwitch()
|
||
self.justifyText.setToolTip(
|
||
"Applies to PDF, printing, HTML, and Open Document exports."
|
||
)
|
||
self.justifyText.setChecked(
|
||
self.optState.getBool("GuiBuildNovel", "justifyText", False)
|
||
)
|
||
|
||
self.noStyling = QSwitch()
|
||
self.noStyling.setToolTip(
|
||
"Disable all styling of the text."
|
||
)
|
||
self.noStyling.setChecked(
|
||
self.optState.getBool("GuiBuildNovel", "noStyling", False)
|
||
)
|
||
|
||
# Dummy box due to QGridView and QLineEdit expand bug
|
||
self.boxFont = QHBoxLayout()
|
||
self.boxFont.addWidget(self.textFont)
|
||
|
||
self.formatForm.addWidget(QLabel("Font family"), 0, 0, 1, 1, Qt.AlignLeft)
|
||
self.formatForm.addLayout(self.boxFont, 0, 1, 1, 1, Qt.AlignRight)
|
||
self.formatForm.addWidget(self.fontButton, 0, 2, 1, 1, Qt.AlignRight)
|
||
self.formatForm.addWidget(QLabel("Font size"), 1, 0, 1, 1, Qt.AlignLeft)
|
||
self.formatForm.addWidget(self.textSize, 1, 1, 1, 2, Qt.AlignRight)
|
||
self.formatForm.addWidget(QLabel("Justify text"), 2, 0, 1, 1, Qt.AlignLeft)
|
||
self.formatForm.addWidget(self.justifyText, 2, 1, 1, 2, Qt.AlignRight)
|
||
self.formatForm.addWidget(QLabel("Disable styling"), 3, 0, 1, 1, Qt.AlignLeft)
|
||
self.formatForm.addWidget(self.noStyling, 3, 1, 1, 2, Qt.AlignRight)
|
||
|
||
self.formatForm.setColumnStretch(0, 0)
|
||
self.formatForm.setColumnStretch(1, 1)
|
||
self.formatForm.setColumnStretch(2, 0)
|
||
|
||
# Include Switches
|
||
# ================
|
||
|
||
self.textGroup = QGroupBox("Text Options", self)
|
||
self.textForm = QGridLayout(self)
|
||
self.textGroup.setLayout(self.textForm)
|
||
|
||
self.includeSynopsis = QSwitch()
|
||
self.includeSynopsis.setToolTip(
|
||
"Include synopsis comments in the output."
|
||
)
|
||
self.includeSynopsis.setChecked(
|
||
self.optState.getBool("GuiBuildNovel", "incSynopsis", False)
|
||
)
|
||
|
||
self.includeComments = QSwitch()
|
||
self.includeComments.setToolTip(
|
||
"Include plain comments in the output."
|
||
)
|
||
self.includeComments.setChecked(
|
||
self.optState.getBool("GuiBuildNovel", "incComments", False)
|
||
)
|
||
|
||
self.includeKeywords = QSwitch()
|
||
self.includeKeywords.setToolTip(
|
||
"Include meta keywords (tags, references) in the output."
|
||
)
|
||
self.includeKeywords.setChecked(
|
||
self.optState.getBool("GuiBuildNovel", "incKeywords", False)
|
||
)
|
||
|
||
self.includeBody = QSwitch()
|
||
self.includeBody.setToolTip(
|
||
"Include body text in the output."
|
||
)
|
||
self.includeBody.setChecked(
|
||
self.optState.getBool("GuiBuildNovel", "incBodyText", True)
|
||
)
|
||
|
||
self.textForm.addWidget(QLabel("Include synopsis"), 0, 0, 1, 1, Qt.AlignLeft)
|
||
self.textForm.addWidget(self.includeSynopsis, 0, 1, 1, 1, Qt.AlignRight)
|
||
self.textForm.addWidget(QLabel("Include comments"), 1, 0, 1, 1, Qt.AlignLeft)
|
||
self.textForm.addWidget(self.includeComments, 1, 1, 1, 1, Qt.AlignRight)
|
||
self.textForm.addWidget(QLabel("Include keywords"), 2, 0, 1, 1, Qt.AlignLeft)
|
||
self.textForm.addWidget(self.includeKeywords, 2, 1, 1, 1, Qt.AlignRight)
|
||
self.textForm.addWidget(QLabel("Include body text"), 3, 0, 1, 1, Qt.AlignLeft)
|
||
self.textForm.addWidget(self.includeBody, 3, 1, 1, 1, Qt.AlignRight)
|
||
|
||
self.textForm.setColumnStretch(0, 1)
|
||
self.textForm.setColumnStretch(1, 0)
|
||
|
||
# File Filter Options
|
||
# ===================
|
||
|
||
self.fileGroup = QGroupBox("File Filter Options", self)
|
||
self.fileForm = QGridLayout(self)
|
||
self.fileGroup.setLayout(self.fileForm)
|
||
|
||
self.novelFiles = QSwitch()
|
||
self.novelFiles.setToolTip(
|
||
"Include files with layouts 'Book', 'Page', 'Partition', "
|
||
"'Chapter', 'Unnumbered', and 'Scene'."
|
||
)
|
||
self.novelFiles.setChecked(
|
||
self.optState.getBool("GuiBuildNovel", "addNovel", True)
|
||
)
|
||
|
||
self.noteFiles = QSwitch()
|
||
self.noteFiles.setToolTip("Include files with layout 'Note'.")
|
||
self.noteFiles.setChecked(
|
||
self.optState.getBool("GuiBuildNovel", "addNotes", False)
|
||
)
|
||
|
||
self.ignoreFlag = QSwitch()
|
||
self.ignoreFlag.setToolTip(
|
||
"Ignore the 'Include when building project' setting and include "
|
||
"all files in the output."
|
||
)
|
||
self.ignoreFlag.setChecked(
|
||
self.optState.getBool("GuiBuildNovel", "ignoreFlag", False)
|
||
)
|
||
|
||
self.fileForm.addWidget(QLabel("Include novel files"), 0, 0, 1, 1, Qt.AlignLeft)
|
||
self.fileForm.addWidget(self.novelFiles, 0, 1, 1, 1, Qt.AlignRight)
|
||
self.fileForm.addWidget(QLabel("Include note files"), 1, 0, 1, 1, Qt.AlignLeft)
|
||
self.fileForm.addWidget(self.noteFiles, 1, 1, 1, 1, Qt.AlignRight)
|
||
self.fileForm.addWidget(QLabel("Ignore export flag"), 2, 0, 1, 1, Qt.AlignLeft)
|
||
self.fileForm.addWidget(self.ignoreFlag, 2, 1, 1, 1, Qt.AlignRight)
|
||
|
||
self.fileForm.setColumnStretch(0, 1)
|
||
self.fileForm.setColumnStretch(1, 0)
|
||
|
||
# Export Options
|
||
# ==============
|
||
|
||
self.exportGroup = QGroupBox("Export Options", self)
|
||
self.exportForm = QGridLayout(self)
|
||
self.exportGroup.setLayout(self.exportForm)
|
||
|
||
self.replaceTabs = QSwitch()
|
||
self.replaceTabs.setToolTip(
|
||
"Replace all tabs with eight spaces."
|
||
)
|
||
self.replaceTabs.setChecked(
|
||
self.optState.getBool("GuiBuildNovel", "replaceTabs", False)
|
||
)
|
||
|
||
self.exportForm.addWidget(QLabel("Replace tabs with spaces"), 0, 0, 1, 1, Qt.AlignLeft)
|
||
self.exportForm.addWidget(self.replaceTabs, 0, 1, 1, 1, Qt.AlignRight)
|
||
|
||
self.exportForm.setColumnStretch(0, 1)
|
||
self.exportForm.setColumnStretch(1, 0)
|
||
|
||
# Build Button
|
||
# ============
|
||
|
||
self.buildProgress = QProgressBar()
|
||
|
||
self.buildNovel = QPushButton("Build Preview")
|
||
self.buildNovel.clicked.connect(self._buildPreview)
|
||
|
||
# Action Buttons
|
||
# ==============
|
||
|
||
self.buttonBox = QHBoxLayout()
|
||
|
||
# Printing
|
||
|
||
self.printMenu = QMenu(self)
|
||
self.btnPrint = QPushButton("Print")
|
||
self.btnPrint.setMenu(self.printMenu)
|
||
|
||
self.printSend = QAction("Print Preview", self)
|
||
self.printSend.triggered.connect(self._printDocument)
|
||
self.printMenu.addAction(self.printSend)
|
||
|
||
self.printFile = QAction("Print to PDF", self)
|
||
self.printFile.triggered.connect(lambda: self._saveDocument(self.FMT_PDF))
|
||
self.printMenu.addAction(self.printFile)
|
||
|
||
# Saving to File
|
||
|
||
self.saveMenu = QMenu(self)
|
||
self.btnSave = QPushButton("Save As")
|
||
self.btnSave.setMenu(self.saveMenu)
|
||
|
||
self.saveODT = QAction("Open Document (.odt)", self)
|
||
self.saveODT.triggered.connect(lambda: self._saveDocument(self.FMT_ODT))
|
||
self.saveMenu.addAction(self.saveODT)
|
||
|
||
self.saveFODT = QAction("Flat Open Document (.fodt)", self)
|
||
self.saveFODT.triggered.connect(lambda: self._saveDocument(self.FMT_FODT))
|
||
self.saveMenu.addAction(self.saveFODT)
|
||
|
||
self.saveHTM = QAction("novelWriter HTML (.htm)", self)
|
||
self.saveHTM.triggered.connect(lambda: self._saveDocument(self.FMT_HTM))
|
||
self.saveMenu.addAction(self.saveHTM)
|
||
|
||
self.saveNWD = QAction("novelWriter Markdown (.nwd)", self)
|
||
self.saveNWD.triggered.connect(lambda: self._saveDocument(self.FMT_NWD))
|
||
self.saveMenu.addAction(self.saveNWD)
|
||
|
||
self.saveMD = QAction("Standard Markdown (.md)", self)
|
||
self.saveMD.triggered.connect(lambda: self._saveDocument(self.FMT_MD))
|
||
self.saveMenu.addAction(self.saveMD)
|
||
|
||
self.saveGH = QAction("GitHub Markdown (.md)", self)
|
||
self.saveGH.triggered.connect(lambda: self._saveDocument(self.FMT_GH))
|
||
self.saveMenu.addAction(self.saveGH)
|
||
|
||
self.saveJsonH = QAction("JSON + novelWriter HTML (.json)", self)
|
||
self.saveJsonH.triggered.connect(lambda: self._saveDocument(self.FMT_JSON_H))
|
||
self.saveMenu.addAction(self.saveJsonH)
|
||
|
||
self.saveJsonM = QAction("JSON + novelWriters Markdown (.json)", self)
|
||
self.saveJsonM.triggered.connect(lambda: self._saveDocument(self.FMT_JSON_M))
|
||
self.saveMenu.addAction(self.saveJsonM)
|
||
|
||
self.btnClose = QPushButton("Close")
|
||
self.btnClose.clicked.connect(self._doClose)
|
||
|
||
self.buttonBox.addWidget(self.btnSave)
|
||
self.buttonBox.addWidget(self.btnPrint)
|
||
self.buttonBox.addWidget(self.btnClose)
|
||
self.buttonBox.setSpacing(self.mainConf.pxInt(4))
|
||
|
||
# Assemble GUI
|
||
# ============
|
||
|
||
# Splitter Position
|
||
boxWidth = self.mainConf.pxInt(350)
|
||
boxWidth = self.optState.getInt("GuiBuildNovel", "boxWidth", boxWidth)
|
||
docWidth = max(self.width() - boxWidth, 100)
|
||
docWidth = self.optState.getInt("GuiBuildNovel", "docWidth", docWidth)
|
||
|
||
# The Tool Box
|
||
self.toolsBox = QVBoxLayout()
|
||
self.toolsBox.addWidget(self.titleGroup)
|
||
self.toolsBox.addWidget(self.formatGroup)
|
||
self.toolsBox.addWidget(self.textGroup)
|
||
self.toolsBox.addWidget(self.fileGroup)
|
||
self.toolsBox.addWidget(self.exportGroup)
|
||
self.toolsBox.addStretch(1)
|
||
|
||
# Tool Box Wrapper Widget
|
||
self.toolsWidget = QWidget()
|
||
self.toolsWidget.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum)
|
||
self.toolsWidget.setLayout(self.toolsBox)
|
||
|
||
# Tool Box Scroll Area
|
||
self.toolsArea = QScrollArea()
|
||
self.toolsArea.setMinimumWidth(self.mainConf.pxInt(250))
|
||
self.toolsArea.setWidgetResizable(True)
|
||
self.toolsArea.setWidget(self.toolsWidget)
|
||
|
||
if self.mainConf.hideVScroll:
|
||
self.toolsArea.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
|
||
else:
|
||
self.toolsArea.setVerticalScrollBarPolicy(Qt.ScrollBarAsNeeded)
|
||
|
||
if self.mainConf.hideHScroll:
|
||
self.toolsArea.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
|
||
else:
|
||
self.toolsArea.setHorizontalScrollBarPolicy(Qt.ScrollBarAsNeeded)
|
||
|
||
# Tools and Buttons Layout
|
||
tSp = self.mainConf.pxInt(8)
|
||
self.innerBox = QVBoxLayout()
|
||
self.innerBox.addWidget(self.toolsArea)
|
||
self.innerBox.addSpacing(tSp)
|
||
self.innerBox.addWidget(self.buildProgress)
|
||
self.innerBox.addWidget(self.buildNovel)
|
||
self.innerBox.addSpacing(tSp)
|
||
self.innerBox.addLayout(self.buttonBox)
|
||
|
||
# Tools and Buttons Wrapper Widget
|
||
self.innerWidget = QWidget()
|
||
self.innerWidget.setLayout(self.innerBox)
|
||
|
||
# Main Dialog Splitter
|
||
self.mainSplit = QSplitter(Qt.Horizontal)
|
||
self.mainSplit.addWidget(self.innerWidget)
|
||
self.mainSplit.addWidget(self.docView)
|
||
self.mainSplit.setSizes([boxWidth, docWidth])
|
||
|
||
self.idxSettings = self.mainSplit.indexOf(self.innerWidget)
|
||
self.idxDocument = self.mainSplit.indexOf(self.docView)
|
||
|
||
self.mainSplit.setCollapsible(self.idxSettings, False)
|
||
self.mainSplit.setCollapsible(self.idxDocument, False)
|
||
|
||
# Outer Layout
|
||
self.outerBox = QHBoxLayout()
|
||
self.outerBox.addWidget(self.mainSplit)
|
||
|
||
self.setLayout(self.outerBox)
|
||
self.buildNovel.setFocus()
|
||
|
||
logger.debug("GuiBuildNovel initialisation complete")
|
||
|
||
return
|
||
|
||
def viewCachedDoc(self):
|
||
"""Load the previously generated document from cache.
|
||
"""
|
||
if self._loadCache():
|
||
textFont = self.textFont.text()
|
||
textSize = self.textSize.value()
|
||
justifyText = self.justifyText.isChecked()
|
||
self.docView.setTextFont(textFont, textSize)
|
||
self.docView.setJustify(justifyText)
|
||
if self.noStyling.isChecked():
|
||
self.docView.clearStyleSheet()
|
||
else:
|
||
self.docView.setStyleSheet(self.htmlStyle)
|
||
|
||
htmlSize = sum([len(x) for x in self.htmlText])
|
||
if htmlSize < nwConst.MAX_BUILDSIZE:
|
||
qApp.processEvents()
|
||
self.docView.setContent(self.htmlText, self.buildTime)
|
||
else:
|
||
self.docView.setText(
|
||
"Failed to generate preview. The result is too big."
|
||
)
|
||
|
||
else:
|
||
self.htmlText = []
|
||
self.htmlStyle = []
|
||
self.buildTime = 0
|
||
return False
|
||
|
||
return True
|
||
|
||
##
|
||
# Slots and Related
|
||
##
|
||
|
||
def _buildPreview(self):
|
||
"""Build a preview of the project in the document viewer.
|
||
"""
|
||
# Get Settings
|
||
justifyText = self.justifyText.isChecked()
|
||
noStyling = self.noStyling.isChecked()
|
||
textFont = self.textFont.text()
|
||
textSize = self.textSize.value()
|
||
replaceTabs = self.replaceTabs.isChecked()
|
||
|
||
self.htmlText = []
|
||
self.htmlStyle = []
|
||
self.htmlSize = 0
|
||
|
||
# Build Preview
|
||
# =============
|
||
|
||
makeHtml = ToHtml(self.theProject, self.theParent)
|
||
self._doBuild(makeHtml, isPreview=True)
|
||
if replaceTabs:
|
||
makeHtml.replaceTabs()
|
||
|
||
self.htmlText = makeHtml.fullHTML
|
||
self.htmlStyle = makeHtml.getStyleSheet()
|
||
self.htmlSize = makeHtml.getFullResultSize()
|
||
self.buildTime = int(time())
|
||
|
||
# Load Preview
|
||
# ============
|
||
|
||
self.docView.setTextFont(textFont, textSize)
|
||
self.docView.setJustify(justifyText)
|
||
if noStyling:
|
||
self.docView.clearStyleSheet()
|
||
else:
|
||
self.docView.setStyleSheet(self.htmlStyle)
|
||
|
||
if self.htmlSize < nwConst.MAX_BUILDSIZE:
|
||
self.docView.setContent(self.htmlText, self.buildTime)
|
||
else:
|
||
self.docView.setText(
|
||
"Failed to generate preview. The result is too big."
|
||
)
|
||
|
||
self._saveCache()
|
||
|
||
return
|
||
|
||
def _doBuild(self, bldObj, isPreview=False, doConvert=True):
|
||
"""Rund the build with a specific build object.
|
||
"""
|
||
tStart = int(time())
|
||
|
||
# Get Settings
|
||
fmtTitle = self.fmtTitle.text().strip()
|
||
fmtChapter = self.fmtChapter.text().strip()
|
||
fmtUnnumbered = self.fmtUnnumbered.text().strip()
|
||
fmtScene = self.fmtScene.text().strip()
|
||
fmtSection = self.fmtSection.text().strip()
|
||
textFont = self.textFont.text()
|
||
textSize = self.textSize.value()
|
||
justifyText = self.justifyText.isChecked()
|
||
noStyling = self.noStyling.isChecked()
|
||
incSynopsis = self.includeSynopsis.isChecked()
|
||
incComments = self.includeComments.isChecked()
|
||
incKeywords = self.includeKeywords.isChecked()
|
||
novelFiles = self.novelFiles.isChecked()
|
||
noteFiles = self.noteFiles.isChecked()
|
||
ignoreFlag = self.ignoreFlag.isChecked()
|
||
includeBody = self.includeBody.isChecked()
|
||
|
||
# Get font information
|
||
fontInfo = QFontInfo(QFont(textFont, textSize))
|
||
textFixed = fontInfo.fixedPitch()
|
||
|
||
isHtml = isinstance(bldObj, ToHtml)
|
||
isOdt = isinstance(bldObj, ToOdt)
|
||
|
||
bldObj.setTitleFormat(fmtTitle)
|
||
bldObj.setChapterFormat(fmtChapter)
|
||
bldObj.setUnNumberedFormat(fmtUnnumbered)
|
||
bldObj.setSceneFormat(fmtScene, fmtScene == "")
|
||
bldObj.setSectionFormat(fmtSection, fmtSection == "")
|
||
|
||
bldObj.setFont(textFont, textSize, textFixed)
|
||
bldObj.setJustify(justifyText)
|
||
|
||
bldObj.setSynopsis(incSynopsis)
|
||
bldObj.setComments(incComments)
|
||
bldObj.setKeywords(incKeywords)
|
||
bldObj.setBodyText(includeBody)
|
||
|
||
if isHtml:
|
||
bldObj.setStyles(not noStyling)
|
||
|
||
if isOdt:
|
||
bldObj.setColourHeaders(not noStyling)
|
||
bldObj.initDocument()
|
||
|
||
# Make sure the project and document is up to date
|
||
self.theParent.treeView.flushTreeOrder()
|
||
self.theParent.saveDocument()
|
||
|
||
self.buildProgress.setMaximum(len(self.theProject.projTree))
|
||
self.buildProgress.setValue(0)
|
||
|
||
for nItt, tItem in enumerate(self.theProject.projTree):
|
||
|
||
noteRoot = noteFiles
|
||
noteRoot &= tItem.itemType == nwItemType.ROOT
|
||
noteRoot &= tItem.itemClass != nwItemClass.NOVEL
|
||
noteRoot &= tItem.itemClass != nwItemClass.ARCHIVE
|
||
|
||
try:
|
||
if noteRoot:
|
||
# Add headers for root folders of notes
|
||
bldObj.addRootHeading(tItem.itemHandle)
|
||
if doConvert:
|
||
bldObj.doConvert()
|
||
|
||
elif self._checkInclude(tItem, noteFiles, novelFiles, ignoreFlag):
|
||
bldObj.setText(tItem.itemHandle)
|
||
bldObj.doAutoReplace()
|
||
bldObj.tokenizeText()
|
||
bldObj.doHeaders()
|
||
if doConvert:
|
||
bldObj.doConvert()
|
||
bldObj.doPostProcessing()
|
||
|
||
except Exception:
|
||
logger.error("Failed to build document '%s'" % tItem.itemHandle)
|
||
nw.logException()
|
||
if isPreview:
|
||
self.docView.setText((
|
||
"Failed to generate preview. "
|
||
"Document with title '%s' could not be parsed."
|
||
) % tItem.itemName)
|
||
|
||
return False
|
||
|
||
# Update progress bar, also for skipped items
|
||
self.buildProgress.setValue(nItt+1)
|
||
|
||
if isOdt:
|
||
bldObj.closeDocument()
|
||
|
||
tEnd = int(time())
|
||
logger.debug("Built project in %.3f ms" % (1000*(tEnd - tStart)))
|
||
|
||
if bldObj.errData:
|
||
self.theParent.makeAlert((
|
||
"There were problems when building the project:"
|
||
"<br>- %s"
|
||
) % "<br>- ".join(bldObj.errData), nwAlert.ERROR)
|
||
|
||
return
|
||
|
||
def _checkInclude(self, theItem, noteFiles, novelFiles, ignoreFlag):
|
||
"""This function checks whether a file should be included in the
|
||
export or not. For standard note and novel files, this is
|
||
controlled by the options selected by the user. For other files
|
||
classified as non-exportable, a few checks must be made, and the
|
||
following are not:
|
||
* Items that are not actual files.
|
||
* Items that have been orphaned which are tagged as NO_LAYOUT
|
||
and NO_CLASS.
|
||
* Items that appear in the TRASH folder or have parent set to
|
||
None (orphaned files).
|
||
"""
|
||
if theItem is None:
|
||
return False
|
||
|
||
if not theItem.isExported and not ignoreFlag:
|
||
return False
|
||
|
||
isNone = theItem.itemType != nwItemType.FILE
|
||
isNone |= theItem.itemLayout == nwItemLayout.NO_LAYOUT
|
||
isNone |= theItem.itemClass == nwItemClass.NO_CLASS
|
||
isNone |= theItem.itemClass == nwItemClass.TRASH
|
||
isNone |= theItem.itemParent == self.theProject.projTree.trashRoot()
|
||
isNone |= theItem.itemParent is None
|
||
isNote = theItem.itemLayout == nwItemLayout.NOTE
|
||
isNovel = not isNone and not isNote
|
||
|
||
if isNone:
|
||
return False
|
||
if isNote and not noteFiles:
|
||
return False
|
||
if isNovel and not novelFiles:
|
||
return False
|
||
|
||
rootItem = self.theProject.projTree.getRootItem(theItem.itemHandle)
|
||
if rootItem.itemClass == nwItemClass.ARCHIVE:
|
||
return False
|
||
|
||
return True
|
||
|
||
def _saveDocument(self, theFmt):
|
||
"""Save the document to various formats.
|
||
"""
|
||
replaceTabs = self.replaceTabs.isChecked()
|
||
|
||
fileExt = ""
|
||
textFmt = ""
|
||
|
||
# Settings
|
||
# ========
|
||
|
||
if theFmt == self.FMT_ODT:
|
||
fileExt = "odt"
|
||
textFmt = "Open Document"
|
||
|
||
elif theFmt == self.FMT_FODT:
|
||
fileExt = "fodt"
|
||
textFmt = "Flat Open Document"
|
||
|
||
elif theFmt == self.FMT_HTM:
|
||
fileExt = "htm"
|
||
textFmt = "Plain HTML"
|
||
|
||
elif theFmt == self.FMT_NWD:
|
||
fileExt = "nwd"
|
||
textFmt = "novelWriter Markdown"
|
||
|
||
elif theFmt == self.FMT_MD:
|
||
fileExt = "md"
|
||
textFmt = "Standard Markdown"
|
||
|
||
elif theFmt == self.FMT_GH:
|
||
fileExt = "md"
|
||
textFmt = "GitHub Markdown"
|
||
|
||
elif theFmt == self.FMT_JSON_H:
|
||
fileExt = "json"
|
||
textFmt = "JSON + novelWriter HTML"
|
||
|
||
elif theFmt == self.FMT_JSON_M:
|
||
fileExt = "json"
|
||
textFmt = "JSON + novelWriter Markdown"
|
||
|
||
elif theFmt == self.FMT_PDF:
|
||
fileExt = "pdf"
|
||
textFmt = "PDF"
|
||
|
||
else:
|
||
return False
|
||
|
||
# Generate File Name
|
||
# ==================
|
||
|
||
if fileExt:
|
||
|
||
cleanName = makeFileNameSafe(self.theProject.projName)
|
||
fileName = "%s.%s" % (cleanName, fileExt)
|
||
saveDir = self.mainConf.lastPath
|
||
savePath = os.path.join(saveDir, fileName)
|
||
if not os.path.isdir(saveDir):
|
||
saveDir = self.mainConf.homePath
|
||
|
||
dlgOpt = QFileDialog.Options()
|
||
dlgOpt |= QFileDialog.DontUseNativeDialog
|
||
savePath, _ = QFileDialog.getSaveFileName(
|
||
self, "Save Document As", savePath, options=dlgOpt
|
||
)
|
||
if not savePath:
|
||
return False
|
||
|
||
self.mainConf.setLastPath(savePath)
|
||
|
||
else:
|
||
return False
|
||
|
||
# Build and Write
|
||
# ===============
|
||
|
||
errMsg = ""
|
||
wSuccess = False
|
||
|
||
if theFmt == self.FMT_ODT:
|
||
makeOdt = ToOdt(self.theProject, self.theParent, isFlat=False)
|
||
self._doBuild(makeOdt)
|
||
try:
|
||
makeOdt.saveOpenDocText(savePath)
|
||
wSuccess = True
|
||
except Exception as e:
|
||
errMsg = str(e)
|
||
|
||
elif theFmt == self.FMT_FODT:
|
||
makeOdt = ToOdt(self.theProject, self.theParent, isFlat=True)
|
||
self._doBuild(makeOdt)
|
||
try:
|
||
makeOdt.saveFlatXML(savePath)
|
||
wSuccess = True
|
||
except Exception as e:
|
||
errMsg = str(e)
|
||
|
||
elif theFmt == self.FMT_HTM:
|
||
makeHtml = ToHtml(self.theProject, self.theParent)
|
||
self._doBuild(makeHtml)
|
||
if replaceTabs:
|
||
makeHtml.replaceTabs()
|
||
|
||
try:
|
||
makeHtml.saveHTML5(savePath)
|
||
wSuccess = True
|
||
except Exception as e:
|
||
errMsg = str(e)
|
||
|
||
elif theFmt == self.FMT_NWD:
|
||
makeNwd = ToMarkdown(self.theProject, self.theParent)
|
||
makeNwd.setKeepMarkdown(True)
|
||
self._doBuild(makeNwd, doConvert=False)
|
||
if replaceTabs:
|
||
makeNwd.replaceTabs(spaceChar=" ")
|
||
|
||
try:
|
||
makeNwd.saveRawMarkdown(savePath)
|
||
wSuccess = True
|
||
except Exception as e:
|
||
errMsg = str(e)
|
||
|
||
elif theFmt in (self.FMT_MD, self.FMT_GH):
|
||
makeMd = ToMarkdown(self.theProject, self.theParent)
|
||
if theFmt == self.FMT_GH:
|
||
makeMd.setGitHubMarkdown()
|
||
else:
|
||
makeMd.setStandardMarkdown()
|
||
|
||
self._doBuild(makeMd)
|
||
if replaceTabs:
|
||
makeMd.replaceTabs(nSpaces=4, spaceChar=" ")
|
||
|
||
try:
|
||
makeMd.saveMarkdown(savePath)
|
||
wSuccess = True
|
||
except Exception as e:
|
||
errMsg = str(e)
|
||
|
||
elif theFmt == self.FMT_JSON_H or theFmt == self.FMT_JSON_M:
|
||
jsonData = {
|
||
"meta" : {
|
||
"workingTitle" : self.theProject.projName,
|
||
"novelTitle" : self.theProject.bookTitle,
|
||
"authors" : self.theProject.bookAuthors,
|
||
"buildTime" : self.buildTime,
|
||
}
|
||
}
|
||
|
||
if theFmt == self.FMT_JSON_H:
|
||
makeHtml = ToHtml(self.theProject, self.theParent)
|
||
self._doBuild(makeHtml)
|
||
if replaceTabs:
|
||
makeHtml.replaceTabs()
|
||
|
||
theBody = []
|
||
for htmlPage in makeHtml.fullHTML:
|
||
theBody.append(htmlPage.rstrip("\n").split("\n"))
|
||
jsonData["text"] = {
|
||
"css" : self.htmlStyle,
|
||
"html" : theBody,
|
||
}
|
||
|
||
elif theFmt == self.FMT_JSON_M:
|
||
makeMd = ToHtml(self.theProject, self.theParent)
|
||
makeMd.setKeepMarkdown(True)
|
||
self._doBuild(makeMd, doConvert=False)
|
||
if replaceTabs:
|
||
makeMd.replaceTabs(spaceChar=" ")
|
||
|
||
theBody = []
|
||
for nwdPage in makeMd.theMarkdown:
|
||
theBody.append(nwdPage.split("\n"))
|
||
jsonData["text"] = {
|
||
"nwd" : theBody,
|
||
}
|
||
|
||
try:
|
||
with open(savePath, mode="w", encoding="utf8") as outFile:
|
||
outFile.write(json.dumps(jsonData, indent=2))
|
||
wSuccess = True
|
||
except Exception as e:
|
||
errMsg = str(e)
|
||
|
||
elif theFmt == self.FMT_PDF:
|
||
try:
|
||
thePrinter = QPrinter()
|
||
thePrinter.setOutputFormat(QPrinter.PdfFormat)
|
||
thePrinter.setOrientation(QPrinter.Portrait)
|
||
thePrinter.setDuplex(QPrinter.DuplexLongSide)
|
||
thePrinter.setFontEmbeddingEnabled(True)
|
||
thePrinter.setColorMode(QPrinter.Color)
|
||
thePrinter.setOutputFileName(savePath)
|
||
self.docView.qDocument.print(thePrinter)
|
||
wSuccess = True
|
||
|
||
except Exception as e:
|
||
errMsg - str(e)
|
||
|
||
else:
|
||
errMsg = "Unknown format"
|
||
|
||
# Report to user
|
||
if wSuccess:
|
||
self.theParent.makeAlert(
|
||
"%s file successfully written to:<br> %s" % (
|
||
textFmt, savePath
|
||
), nwAlert.INFO
|
||
)
|
||
else:
|
||
self.theParent.makeAlert(
|
||
"Failed to write %s file. %s" % (
|
||
textFmt, errMsg
|
||
), nwAlert.ERROR
|
||
)
|
||
|
||
return wSuccess
|
||
|
||
def _printDocument(self):
|
||
"""Open the print preview dialog.
|
||
"""
|
||
thePreview = QPrintPreviewDialog(self)
|
||
thePreview.paintRequested.connect(self._doPrintPreview)
|
||
thePreview.exec_()
|
||
return
|
||
|
||
def _doPrintPreview(self, thePrinter):
|
||
"""Connect the print preview painter to the document viewer.
|
||
"""
|
||
qApp.setOverrideCursor(QCursor(Qt.WaitCursor))
|
||
thePrinter.setOrientation(QPrinter.Portrait)
|
||
self.docView.qDocument.print(thePrinter)
|
||
qApp.restoreOverrideCursor()
|
||
return
|
||
|
||
def _selectFont(self):
|
||
"""Open the QFontDialog and set a font for the font style.
|
||
"""
|
||
currFont = QFont()
|
||
currFont.setFamily(self.textFont.text())
|
||
currFont.setPointSize(self.textSize.value())
|
||
theFont, theStatus = QFontDialog.getFont(currFont, self)
|
||
if theStatus:
|
||
self.textFont.setText(theFont.family())
|
||
self.textSize.setValue(theFont.pointSize())
|
||
|
||
self.raise_() # Move the dialog to front (fixes a bug on macOS)
|
||
|
||
return
|
||
|
||
def _loadCache(self):
|
||
"""Save the current data to cache.
|
||
"""
|
||
buildCache = os.path.join(self.theProject.projCache, nwFiles.BUILD_CACHE)
|
||
dataCount = 0
|
||
if os.path.isfile(buildCache):
|
||
|
||
logger.debug("Loading build cache")
|
||
try:
|
||
with open(buildCache, mode="r", encoding="utf8") as inFile:
|
||
theJson = inFile.read()
|
||
theData = json.loads(theJson)
|
||
except Exception:
|
||
logger.error("Failed to load build cache")
|
||
nw.logException()
|
||
return False
|
||
|
||
if "htmlText" in theData.keys():
|
||
self.htmlText = theData["htmlText"]
|
||
dataCount += 1
|
||
if "htmlStyle" in theData.keys():
|
||
self.htmlStyle = theData["htmlStyle"]
|
||
dataCount += 1
|
||
if "buildTime" in theData.keys():
|
||
self.buildTime = theData["buildTime"]
|
||
|
||
return dataCount == 2
|
||
|
||
def _saveCache(self):
|
||
"""Save the current data to cache.
|
||
"""
|
||
buildCache = os.path.join(self.theProject.projCache, nwFiles.BUILD_CACHE)
|
||
|
||
logger.debug("Saving build cache")
|
||
try:
|
||
with open(buildCache, mode="w+", encoding="utf8") as outFile:
|
||
outFile.write(json.dumps({
|
||
"htmlText" : self.htmlText,
|
||
"htmlStyle" : self.htmlStyle,
|
||
"buildTime" : self.buildTime,
|
||
}, indent=2))
|
||
except Exception:
|
||
logger.error("Failed to save build cache")
|
||
nw.logException()
|
||
return False
|
||
|
||
return True
|
||
|
||
def _doClose(self):
|
||
"""Close button was clicked.
|
||
"""
|
||
self.close()
|
||
return
|
||
|
||
##
|
||
# Events
|
||
##
|
||
|
||
def closeEvent(self, theEvent):
|
||
"""Capture the user closing the window so we can save settings.
|
||
"""
|
||
self._saveSettings()
|
||
self.docView.clear()
|
||
theEvent.accept()
|
||
return
|
||
|
||
##
|
||
# Internal Functions
|
||
##
|
||
|
||
def _saveSettings(self):
|
||
"""Save the various user settings.
|
||
"""
|
||
logger.debug("Saving GuiBuildNovel settings")
|
||
|
||
# Formatting
|
||
self.theProject.setTitleFormat({
|
||
"title" : self.fmtTitle.text().strip(),
|
||
"chapter" : self.fmtChapter.text().strip(),
|
||
"unnumbered" : self.fmtUnnumbered.text().strip(),
|
||
"scene" : self.fmtScene.text().strip(),
|
||
"section" : self.fmtSection.text().strip(),
|
||
})
|
||
|
||
winWidth = self.mainConf.rpxInt(self.width())
|
||
winHeight = self.mainConf.rpxInt(self.height())
|
||
justifyText = self.justifyText.isChecked()
|
||
noStyling = self.noStyling.isChecked()
|
||
textFont = self.textFont.text()
|
||
textSize = self.textSize.value()
|
||
novelFiles = self.novelFiles.isChecked()
|
||
noteFiles = self.noteFiles.isChecked()
|
||
ignoreFlag = self.ignoreFlag.isChecked()
|
||
incSynopsis = self.includeSynopsis.isChecked()
|
||
incComments = self.includeComments.isChecked()
|
||
incKeywords = self.includeKeywords.isChecked()
|
||
incBodyText = self.includeBody.isChecked()
|
||
replaceTabs = self.replaceTabs.isChecked()
|
||
|
||
mainSplit = self.mainSplit.sizes()
|
||
if len(mainSplit) == 2:
|
||
boxWidth = self.mainConf.rpxInt(mainSplit[0])
|
||
docWidth = self.mainConf.rpxInt(mainSplit[1])
|
||
else:
|
||
boxWidth = 100
|
||
docWidth = 100
|
||
|
||
# GUI Settings
|
||
self.optState.setValue("GuiBuildNovel", "winWidth", winWidth)
|
||
self.optState.setValue("GuiBuildNovel", "winHeight", winHeight)
|
||
self.optState.setValue("GuiBuildNovel", "boxWidth", boxWidth)
|
||
self.optState.setValue("GuiBuildNovel", "docWidth", docWidth)
|
||
self.optState.setValue("GuiBuildNovel", "justifyText", justifyText)
|
||
self.optState.setValue("GuiBuildNovel", "noStyling", noStyling)
|
||
self.optState.setValue("GuiBuildNovel", "textFont", textFont)
|
||
self.optState.setValue("GuiBuildNovel", "textSize", textSize)
|
||
self.optState.setValue("GuiBuildNovel", "addNovel", novelFiles)
|
||
self.optState.setValue("GuiBuildNovel", "addNotes", noteFiles)
|
||
self.optState.setValue("GuiBuildNovel", "ignoreFlag", ignoreFlag)
|
||
self.optState.setValue("GuiBuildNovel", "incSynopsis", incSynopsis)
|
||
self.optState.setValue("GuiBuildNovel", "incComments", incComments)
|
||
self.optState.setValue("GuiBuildNovel", "incKeywords", incKeywords)
|
||
self.optState.setValue("GuiBuildNovel", "incBodyText", incBodyText)
|
||
self.optState.setValue("GuiBuildNovel", "replaceTabs", replaceTabs)
|
||
self.optState.saveSettings()
|
||
|
||
return
|
||
|
||
def _reFmtCodes(self, theFormat):
|
||
"""Translates old formatting codes to new ones.
|
||
"""
|
||
theFormat = theFormat.replace(r"%chnum%", r"%ch%")
|
||
theFormat = theFormat.replace(r"%scnum%", r"%sc%")
|
||
theFormat = theFormat.replace(r"%scabsnum%", r"%sca%")
|
||
theFormat = theFormat.replace(r"%chnumword%", r"%chw%")
|
||
return theFormat
|
||
|
||
# END Class GuiBuildNovel
|
||
|
||
class GuiBuildNovelDocView(QTextBrowser):
|
||
|
||
def __init__(self, theParent, theProject):
|
||
QTextBrowser.__init__(self, theParent)
|
||
|
||
logger.debug("Initialising GuiBuildNovelDocView ...")
|
||
|
||
self.mainConf = nw.CONFIG
|
||
self.theProject = theProject
|
||
self.theParent = theParent
|
||
self.theTheme = theParent.theTheme
|
||
self.buildTime = 0
|
||
|
||
self.setMinimumWidth(40*self.theParent.theTheme.textNWidth)
|
||
self.setOpenExternalLinks(False)
|
||
|
||
self.qDocument = self.document()
|
||
self.qDocument.setDocumentMargin(self.mainConf.getTextMargin())
|
||
self.setPlaceholderText(
|
||
"This area will show the content of the document to be "
|
||
"exported or printed. Press the \"Build Preview\" button "
|
||
"to generate content."
|
||
)
|
||
|
||
theFont = QFont()
|
||
if self.mainConf.textFont is None:
|
||
# If none is defined, set the default back to config
|
||
self.mainConf.textFont = self.qDocument.defaultFont().family()
|
||
theFont.setFamily(self.mainConf.textFont)
|
||
theFont.setPointSize(self.mainConf.textSize)
|
||
self.setFont(theFont)
|
||
|
||
# Set the tab stops
|
||
if self.mainConf.verQtValue >= 51000:
|
||
self.setTabStopDistance(self.mainConf.getTabWidth())
|
||
else:
|
||
self.setTabStopWidth(self.mainConf.getTabWidth())
|
||
|
||
docPalette = self.palette()
|
||
docPalette.setColor(QPalette.Base, QColor(255, 255, 255))
|
||
docPalette.setColor(QPalette.Text, QColor(0, 0, 0))
|
||
self.setPalette(docPalette)
|
||
|
||
lblPalette = self.palette()
|
||
lblPalette.setColor(QPalette.Background, lblPalette.toolTipBase().color())
|
||
lblPalette.setColor(QPalette.Foreground, lblPalette.toolTipText().color())
|
||
|
||
lblFont = self.font()
|
||
lblFont.setPointSizeF(0.9*self.theTheme.fontPointSize)
|
||
|
||
fPx = int(1.1*self.theTheme.fontPixelSize)
|
||
|
||
self.theTitle = QLabel("<b>Build Time:</b> Unknown", self)
|
||
self.theTitle.setIndent(0)
|
||
self.theTitle.setAutoFillBackground(True)
|
||
self.theTitle.setAlignment(Qt.AlignCenter)
|
||
self.theTitle.setFixedHeight(fPx)
|
||
self.theTitle.setPalette(lblPalette)
|
||
self.theTitle.setFont(lblFont)
|
||
|
||
self._updateDocMargins()
|
||
self.setStyleSheet()
|
||
|
||
# Age Timer
|
||
self.ageTimer = QTimer()
|
||
self.ageTimer.setInterval(10000)
|
||
self.ageTimer.timeout.connect(self._updateBuildAge)
|
||
self.ageTimer.start()
|
||
|
||
logger.debug("GuiBuildNovelDocView initialisation complete")
|
||
|
||
return
|
||
|
||
def setJustify(self, doJustify):
|
||
"""Set the justify text option.
|
||
"""
|
||
theOpt = self.qDocument.defaultTextOption()
|
||
if doJustify:
|
||
theOpt.setAlignment(Qt.AlignJustify)
|
||
else:
|
||
theOpt.setAlignment(Qt.AlignAbsolute)
|
||
self.qDocument.setDefaultTextOption(theOpt)
|
||
return
|
||
|
||
def setTextFont(self, textFont, textSize):
|
||
"""Set the text font properties.
|
||
"""
|
||
theFont = QFont()
|
||
theFont.setFamily(textFont)
|
||
theFont.setPointSize(textSize)
|
||
self.setFont(theFont)
|
||
return
|
||
|
||
def setContent(self, theText, timeStamp):
|
||
"""Set the content, either from text or list of text.
|
||
"""
|
||
if isinstance(theText, list):
|
||
theText = "".join(theText)
|
||
|
||
self.buildTime = timeStamp
|
||
sPos = self.verticalScrollBar().value()
|
||
qApp.setOverrideCursor(QCursor(Qt.WaitCursor))
|
||
|
||
theText = theText.replace("\t", "!!tab!!")
|
||
theText = theText.replace("<del>", "<span style='text-decoration: line-through;'>")
|
||
theText = theText.replace("</del>", "</span>")
|
||
self.setHtml(theText)
|
||
qApp.processEvents()
|
||
|
||
while self.find("!!tab!!"):
|
||
theCursor = self.textCursor()
|
||
theCursor.insertText("\t")
|
||
|
||
self.verticalScrollBar().setValue(sPos)
|
||
self._updateBuildAge()
|
||
|
||
# Since we change the content while it may still be rendering, we mark
|
||
# the document dirty again to make sure it's re-rendered properly.
|
||
self.qDocument.markContentsDirty(0, self.qDocument.characterCount())
|
||
qApp.restoreOverrideCursor()
|
||
|
||
return
|
||
|
||
def setStyleSheet(self, theStyles=[]):
|
||
"""Set the stylesheet for the preview document.
|
||
"""
|
||
if not theStyles:
|
||
theStyles.append(r"h1, h2 {color: rgb(66, 113, 174);}")
|
||
theStyles.append(r"h3, h4 {color: rgb(50, 50, 50);}")
|
||
theStyles.append(r"a {color: rgb(66, 113, 174);}")
|
||
theStyles.append(r".tags {color: rgb(245, 135, 31); font-weight: bold;}")
|
||
|
||
self.qDocument.setDefaultStyleSheet("\n".join(theStyles))
|
||
|
||
return
|
||
|
||
def clearStyleSheet(self):
|
||
"""Clears the document stylesheet.
|
||
"""
|
||
self.qDocument.setDefaultStyleSheet("")
|
||
return
|
||
|
||
##
|
||
# Events
|
||
##
|
||
|
||
def resizeEvent(self, theEvent):
|
||
"""Make sure the document title is the same width as the window.
|
||
"""
|
||
QTextBrowser.resizeEvent(self, theEvent)
|
||
self._updateDocMargins()
|
||
return
|
||
|
||
##
|
||
# Internal Functions
|
||
##
|
||
|
||
def _updateBuildAge(self):
|
||
"""Update the build time and the fuzzy age.
|
||
"""
|
||
if self.buildTime > 0:
|
||
strBuildTime = "%s (%s)" % (
|
||
datetime.fromtimestamp(self.buildTime).strftime("%x %X"),
|
||
fuzzyTime(time() - self.buildTime)
|
||
)
|
||
else:
|
||
strBuildTime = "Unknown"
|
||
self.theTitle.setText("<b>Build Time:</b> %s" % strBuildTime)
|
||
|
||
def _updateDocMargins(self):
|
||
"""Automatically adjust the header to fill the top of the
|
||
document within the viewport.
|
||
"""
|
||
vBar = self.verticalScrollBar()
|
||
if vBar.isVisible():
|
||
sW = vBar.width()
|
||
else:
|
||
sW = 0
|
||
|
||
tB = self.frameWidth()
|
||
tW = self.width() - 2*tB - sW
|
||
tH = self.theTitle.height()
|
||
|
||
self.theTitle.setGeometry(tB, tB, tW, tH)
|
||
self.setViewportMargins(0, tH, 0, 0)
|
||
|
||
return
|
||
|
||
# END Class GuiBuildNovelDocView
|