Cleaned up the nw/convert folder a bit

2019-10-26 22:24:59 +02:00
parent b9608dcdbd
commit 753b248b9f
11 changed files with 143 additions and 115 deletions
@@ -0,0 +1,104 @@
+# -*- coding: utf-8 -*-
+"""novelWriter HTML Text Converter
+
+ novelWriter – HTML Text Converter
+===================================
+ Extends the Tokenizer class to write HTML
+
+ File History:
+ Created: 2019-05-07 [0.0.1]
+
+"""
+
+import logging
+import re
+import nw
+
+from nw.convert.tokenizer import Tokenizer
+
+logger = logging.getLogger(__name__)
+
+class ToHtml(Tokenizer):
+
+    def __init__(self, theProject, theParent):
+        Tokenizer.__init__(self, theProject, theParent)
+        return
+
+    def doAutoReplace(self):
+        Tokenizer.doAutoReplace(self)
+
+        repDict = {
+            "<"      : "&lt;",
+            ">"      : "&gt;",
+            "&"      : "&amp;",
+            "\u2013" : "&endash;",
+            "\u2014" : "$emdash;",
+            "\u2500" : "$emdash;",
+            "\u2026" : "&hellip;",
+        }
+        xRep = re.compile("|".join([re.escape(k) for k in repDict.keys()]), flags=re.DOTALL)
+        self.theText = xRep.sub(lambda x: repDict[x.group(0)], self.theText)
+
+        return
+
+    def doConvert(self):
+
+        htmlTags = {
+            self.FMT_B_B : "<strong>",
+            self.FMT_B_E : "</strong>",
+            self.FMT_I_B : "<em>",
+            self.FMT_I_E : "</em>",
+            self.FMT_U_B : "<u>",
+            self.FMT_U_E : "</u>",
+        }
+
+        self.theResult = ""
+        thisPar = []
+        for tType, tText, tFormat, tAlign in self.theTokens:
+
+            aStyle = []
+            if tAlign == self.A_CENTRE:
+                aStyle.append("text-align: center;")
+
+            if len(aStyle) > 0:
+                hStyle = " style='%s'" % (" ".join(aStyle))
+            else:
+                hStyle = ""
+
+            if tType == self.T_EMPTY:
+                if len(thisPar) > 0:
+                    self.theResult += "<p%s>%s</p>\n" % (hStyle," ".join(thisPar))
+                thisPar = []
+
+            elif tType == self.T_HEAD1:
+                self.theResult += "<h1%s>%s</h1>\n" % (hStyle,tText)
+
+            elif tType == self.T_HEAD2:
+                self.theResult += "<h2%s>%s</h2>\n" % (hStyle,tText)
+
+            elif tType == self.T_HEAD3:
+                self.theResult += "<h3%s>%s</h3>\n" % (hStyle,tText)
+
+            elif tType == self.T_HEAD4:
+                self.theResult += "<h4%s>%s</h4>\n" % (hStyle,tText)
+
+            elif tType == self.T_SEP:
+                self.theResult += "<div%s>%s</div>\n" % (hStyle,tText)
+
+            elif tType == self.T_TEXT:
+                tTemp = tText
+                for xPos, xLen, xFmt in reversed(tFormat):
+                    tTemp = tTemp[:xPos]+htmlTags[xFmt]+tTemp[xPos+xLen:]
+                thisPar.append(tTemp)
+
+            elif tType == self.T_COMMENT and self.doComments:
+                self.theResult += "<pre>%s</pre>\n" % tText
+
+            elif tType == self.T_COMMAND and self.doCommands:
+                self.theResult += "<pre>%s</pre>\n" % tText
+
+        # print(self.theResult)
+
+        return
+
+# END Class ToHtml