2016-04-05 06:00:19 +12:00
|
|
|
#!/usr/bin/env python
|
|
|
|
# --!-- coding: utf8 --!--
|
2016-04-07 22:18:20 +12:00
|
|
|
import re
|
|
|
|
from PyQt5.QtGui import QFont, QTextCharFormat
|
2016-04-08 22:02:50 +12:00
|
|
|
from PyQt5.QtWidgets import QPlainTextEdit, qApp, QFrame, QFileDialog
|
2016-04-05 06:00:19 +12:00
|
|
|
|
2016-04-07 22:18:20 +12:00
|
|
|
from manuskript.exporter.basic import basicFormat
|
2016-04-06 03:21:07 +12:00
|
|
|
from manuskript.functions import mainWindow
|
|
|
|
from manuskript.models.outlineModel import outlineItem
|
2016-04-07 22:18:20 +12:00
|
|
|
from manuskript.ui.exporters.manuskript.plainTextSettings import exporterSettings
|
2016-04-06 03:21:07 +12:00
|
|
|
|
2016-04-05 06:00:19 +12:00
|
|
|
|
|
|
|
class plainText(basicFormat):
|
2016-04-08 00:47:29 +12:00
|
|
|
name = qApp.translate("Export", "Plain text")
|
|
|
|
description = qApp.translate("Export", """Simplest export to plain text. Allows you to use your own markup not understood
|
2016-04-08 00:27:51 +12:00
|
|
|
by manuskript, for example <a href='www.fountain.io'>Fountain</a>.""")
|
2016-04-06 03:21:07 +12:00
|
|
|
implemented = True
|
2016-04-05 06:00:19 +12:00
|
|
|
requires = {
|
|
|
|
"Settings": True,
|
2016-04-06 03:21:07 +12:00
|
|
|
"Preview": True,
|
2016-04-05 06:00:19 +12:00
|
|
|
}
|
2016-04-08 22:48:19 +12:00
|
|
|
icon = "text-plain"
|
2016-04-05 06:00:19 +12:00
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
# Default settings used in self.getExportFilename. For easy subclassing when exporting plaintext.
|
2016-04-08 22:02:50 +12:00
|
|
|
exportVarName = "lastPlainText"
|
|
|
|
exportFilter = "Text files (*.txt);; Any files (*)"
|
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
def __init__(self):
|
|
|
|
pass
|
2016-04-08 22:02:50 +12:00
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
def settingsWidget(self):
|
|
|
|
w = exporterSettings(self)
|
2016-04-07 23:17:14 +12:00
|
|
|
w.loadSettings()
|
2016-04-06 03:21:07 +12:00
|
|
|
return w
|
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
def previewWidget(self):
|
2016-04-06 03:21:07 +12:00
|
|
|
w = QPlainTextEdit()
|
2016-04-08 00:27:51 +12:00
|
|
|
w.setFrameShape(QFrame.NoFrame)
|
2016-04-06 03:21:07 +12:00
|
|
|
w.setReadOnly(True)
|
|
|
|
return w
|
|
|
|
|
2016-04-15 21:58:09 +12:00
|
|
|
def output(self, settingsWidget):
|
|
|
|
settings = settingsWidget.getSettings()
|
2016-04-15 00:30:42 +12:00
|
|
|
return self.concatenate(mainWindow().mdlOutline.rootItem, settings)
|
2016-04-08 00:27:51 +12:00
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
def getExportFilename(self, settingsWidget, varName=None, filter=None):
|
2016-04-08 22:02:50 +12:00
|
|
|
|
|
|
|
if varName is None:
|
2016-04-15 00:30:42 +12:00
|
|
|
varName = self.exportVarName
|
2016-04-08 22:02:50 +12:00
|
|
|
|
|
|
|
if filter is None:
|
2016-04-15 00:30:42 +12:00
|
|
|
filter = self.exportFilter
|
2016-04-08 22:02:50 +12:00
|
|
|
|
|
|
|
settings = settingsWidget.getSettings()
|
|
|
|
|
|
|
|
s = settings.get("Output", {})
|
|
|
|
if varName in s:
|
|
|
|
filename = s[varName]
|
|
|
|
else:
|
|
|
|
filename = ""
|
|
|
|
|
|
|
|
filename, filter = QFileDialog.getSaveFileName(settingsWidget.parent(),
|
|
|
|
caption=qApp.translate("Export", "Chose output file..."),
|
|
|
|
filter=filter,
|
|
|
|
directory=filename)
|
|
|
|
|
|
|
|
if filename:
|
|
|
|
s[varName] = filename
|
|
|
|
settingsWidget.settings["Output"] = s
|
|
|
|
|
|
|
|
# Auto adds extension if necessary
|
|
|
|
try:
|
|
|
|
# Extract the extension from "Some name (*.ext)"
|
|
|
|
ext = filter.split("(")[1].split(")")[0]
|
|
|
|
ext = ext.split(".")[1]
|
|
|
|
if " " in ext: # In case there are multiple extensions: "Images (*.png *.jpg)"
|
|
|
|
ext = ext.split(" ")[0]
|
|
|
|
except:
|
|
|
|
ext = ""
|
|
|
|
|
|
|
|
if ext and filename[-len(ext)-1:] != ".{}".format(ext):
|
|
|
|
filename += "." + ext
|
|
|
|
|
|
|
|
# Save settings
|
|
|
|
settingsWidget.writeSettings()
|
|
|
|
|
|
|
|
return filename
|
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
def export(self, settingsWidget):
|
2016-04-08 22:02:50 +12:00
|
|
|
settings = settingsWidget.getSettings()
|
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
filename = self.getExportFilename(settingsWidget)
|
|
|
|
settingsWidget.writeSettings()
|
2016-04-15 21:58:09 +12:00
|
|
|
content = self.output(settingsWidget)
|
|
|
|
|
|
|
|
if not content:
|
|
|
|
print("Error: content is empty. Nothing saved.")
|
|
|
|
return
|
2016-04-08 22:02:50 +12:00
|
|
|
|
|
|
|
if filename:
|
|
|
|
with open(filename, "w") as f:
|
2016-04-15 21:58:09 +12:00
|
|
|
f.write(content)
|
2016-04-08 22:02:50 +12:00
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
def preview(self, settingsWidget, previewWidget):
|
2016-04-07 22:18:20 +12:00
|
|
|
settings = settingsWidget.getSettings()
|
2016-04-07 23:17:14 +12:00
|
|
|
|
|
|
|
# Save settings
|
|
|
|
settingsWidget.writeSettings()
|
2016-04-06 03:21:07 +12:00
|
|
|
|
2016-04-15 21:58:09 +12:00
|
|
|
r = self.output(settingsWidget)
|
2016-04-07 22:18:20 +12:00
|
|
|
|
|
|
|
# Set preview font
|
2016-04-15 00:30:42 +12:00
|
|
|
self.preparesTextEditView(previewWidget, settings["Preview"]["PreviewFont"])
|
2016-04-08 00:27:51 +12:00
|
|
|
|
|
|
|
previewWidget.setPlainText(r)
|
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
def preparesTextEditView(self, view, textFont):
|
2016-04-07 22:18:20 +12:00
|
|
|
cf = QTextCharFormat()
|
|
|
|
f = QFont()
|
2016-04-08 00:27:51 +12:00
|
|
|
f.fromString(textFont)
|
2016-04-07 22:18:20 +12:00
|
|
|
cf.setFont(f)
|
2016-04-08 00:27:51 +12:00
|
|
|
view.setCurrentCharFormat(cf)
|
2016-04-06 03:21:07 +12:00
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
def concatenate(self, item: outlineItem, settings) -> str:
|
2016-04-07 22:18:20 +12:00
|
|
|
s = settings
|
2016-04-06 03:21:07 +12:00
|
|
|
r = ""
|
|
|
|
|
2016-04-07 22:18:20 +12:00
|
|
|
# Do we include item
|
|
|
|
if not item.compile() or s["Content"]["IgnoreCompile"]:
|
2016-04-06 03:21:07 +12:00
|
|
|
return ""
|
|
|
|
|
2016-04-07 22:18:20 +12:00
|
|
|
# What do we include
|
|
|
|
l = item.level()
|
|
|
|
if l >= 0: # item is not root
|
|
|
|
|
|
|
|
if item.isFolder():
|
|
|
|
if not s["Content"]["More"] and s["Content"]["FolderTitle"] or\
|
|
|
|
s["Content"]["More"] and s["Content"]["FolderTitle"][l]:
|
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
r += self.processTitle(item.title(), l, settings)
|
2016-04-06 03:21:07 +12:00
|
|
|
|
2016-04-07 22:18:20 +12:00
|
|
|
elif item.isText():
|
|
|
|
if not s["Content"]["More"] and s["Content"]["TextTitle"] or \
|
|
|
|
s["Content"]["More"] and s["Content"]["TextTitle"][l]:
|
2016-04-06 03:21:07 +12:00
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
r += self.processTitle(item.title(), l, settings)
|
2016-04-07 22:18:20 +12:00
|
|
|
|
|
|
|
if not s["Content"]["More"] and s["Content"]["TextText"] or \
|
|
|
|
s["Content"]["More"] and s["Content"]["TextText"][l]:
|
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
r += self.processText(item.text(), settings)
|
2016-04-06 03:21:07 +12:00
|
|
|
|
|
|
|
content = ""
|
|
|
|
|
|
|
|
# Add item children
|
|
|
|
last = None
|
|
|
|
for c in item.children():
|
|
|
|
|
|
|
|
# Separator
|
|
|
|
if last:
|
|
|
|
# Between folder
|
|
|
|
if last == c.type() == "folder":
|
2016-04-07 22:18:20 +12:00
|
|
|
content += s["Separator"]["FF"]
|
|
|
|
|
2016-04-06 03:21:07 +12:00
|
|
|
elif last == c.type() == "md":
|
2016-04-07 22:18:20 +12:00
|
|
|
content += s["Separator"]["TT"]
|
|
|
|
|
2016-04-06 03:21:07 +12:00
|
|
|
elif last == "folder" and c.type() == "md":
|
2016-04-07 22:18:20 +12:00
|
|
|
content += s["Separator"]["FT"]
|
|
|
|
|
2016-04-06 03:21:07 +12:00
|
|
|
elif last == "md" and c.type() == "folder":
|
2016-04-07 22:18:20 +12:00
|
|
|
content += s["Separator"]["TF"]
|
2016-04-06 03:21:07 +12:00
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
content += self.concatenate(c, settings)
|
2016-04-06 03:21:07 +12:00
|
|
|
|
|
|
|
last = c.type()
|
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
# r += self.processContent(content, settings)
|
2016-04-07 22:18:20 +12:00
|
|
|
r += content
|
|
|
|
|
|
|
|
return r
|
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
def processTitle(self, text, level, settings):
|
2016-04-07 23:17:14 +12:00
|
|
|
return text + "\n"
|
|
|
|
|
2016-04-15 00:30:42 +12:00
|
|
|
def processText(self, content, settings):
|
2016-04-07 22:18:20 +12:00
|
|
|
s = settings["Transform"]
|
|
|
|
|
|
|
|
if s["Dash"]:
|
|
|
|
content = content.replace("---", "—")
|
|
|
|
|
|
|
|
if s["Ellipse"]:
|
|
|
|
content = content.replace("...", "…")
|
|
|
|
|
|
|
|
if s["Spaces"]:
|
|
|
|
o = ""
|
|
|
|
while o != content:
|
|
|
|
o = content
|
|
|
|
content = content.replace(" ", " ")
|
|
|
|
|
|
|
|
if s["DoubleQuotes"]:
|
|
|
|
q = s["DoubleQuotes"].split("___")
|
|
|
|
s["Custom"].append([True, '"(.*?)"', "{}\\1{}".format(q[0], q[1]), True])
|
|
|
|
|
|
|
|
if s["SingleQuote"]:
|
|
|
|
q = s["SingleQuote"].split("___")
|
|
|
|
s["Custom"].append([True, "'(.*?)'", "{}\\1{}".format(q[0], q[1]), True])
|
|
|
|
|
|
|
|
for enabled, A, B, reg in s["Custom"]:
|
|
|
|
if not enabled:
|
|
|
|
continue
|
|
|
|
|
|
|
|
if not reg:
|
|
|
|
content = content.replace(A, B)
|
|
|
|
|
|
|
|
else:
|
|
|
|
content = re.sub(A, B, content)
|
|
|
|
|
|
|
|
content += "\n"
|
|
|
|
|
|
|
|
return content
|
2016-04-06 03:21:07 +12:00
|
|
|
|