mirror of
https://github.com/gryf/ebook-converter.git
synced 2026-03-05 16:45:48 +01:00
gryf
0f9792df36
Here is the first batch of modules, which are needed for converting several formats to LRF. Some of the logic has been change, more cleanups will follow.
79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
|
|
""" elements.py -- replacements and helpers for ElementTree """
|
|
|
|
from ebook_converter.polyglot.builtins import unicode_type, string_or_bytes
|
|
|
|
|
|
class ElementWriter(object):
|
|
|
|
def __init__(self, e, header=False, sourceEncoding="ascii",
|
|
spaceBeforeClose=True, outputEncodingName="UTF-16"):
|
|
self.header = header
|
|
self.e = e
|
|
self.sourceEncoding=sourceEncoding
|
|
self.spaceBeforeClose = spaceBeforeClose
|
|
self.outputEncodingName = outputEncodingName
|
|
|
|
def _encodeCdata(self, rawText):
|
|
if isinstance(rawText, bytes):
|
|
rawText = rawText.decode(self.sourceEncoding)
|
|
|
|
text = rawText.replace("&", "&")
|
|
text = text.replace("<", "<")
|
|
text = text.replace(">", ">")
|
|
return text
|
|
|
|
def _writeAttribute(self, f, name, value):
|
|
f.write(' %s="' % unicode_type(name))
|
|
if not isinstance(value, string_or_bytes):
|
|
value = unicode_type(value)
|
|
value = self._encodeCdata(value)
|
|
value = value.replace('"', '"')
|
|
f.write(value)
|
|
f.write('"')
|
|
|
|
def _writeText(self, f, rawText):
|
|
text = self._encodeCdata(rawText)
|
|
f.write(text)
|
|
|
|
def _write(self, f, e):
|
|
f.write('<' + unicode_type(e.tag))
|
|
|
|
attributes = e.items()
|
|
attributes.sort()
|
|
for name, value in attributes:
|
|
self._writeAttribute(f, name, value)
|
|
|
|
if e.text is not None or len(e) > 0:
|
|
f.write('>')
|
|
|
|
if e.text:
|
|
self._writeText(f, e.text)
|
|
|
|
for e2 in e:
|
|
self._write(f, e2)
|
|
|
|
f.write('</%s>' % e.tag)
|
|
else:
|
|
if self.spaceBeforeClose:
|
|
f.write(' ')
|
|
f.write('/>')
|
|
|
|
if e.tail is not None:
|
|
self._writeText(f, e.tail)
|
|
|
|
def toString(self):
|
|
class x:
|
|
pass
|
|
buffer = []
|
|
x.write = buffer.append
|
|
self.write(x)
|
|
return ''.join(buffer)
|
|
|
|
def write(self, f):
|
|
if self.header:
|
|
f.write('<?xml version="1.0" encoding="%s"?>\n' % self.outputEncodingName)
|
|
|
|
self._write(f, self.e)
|