mirror of
https://github.com/gryf/ebook-converter.git
synced 2026-01-22 21:05:45 +01:00
gryf
ce89f5c9d1
This is progressing refactor of the calibre code to make it more readable, and transform it to something more coherent. In this patch, there are changes regarding imports for some modules, instead of polluting namespace of each module with some other modules symbols, which often were imported from other modules. Yuck.
130 lines
5.1 KiB
Python
130 lines
5.1 KiB
Python
import os
|
|
|
|
from lxml import etree
|
|
|
|
from ebook_converter.customize.conversion import InputFormatPlugin
|
|
from ebook_converter.ptempfile import TemporaryDirectory
|
|
from ebook_converter.utils.filenames import ascii_filename
|
|
|
|
|
|
HTML_TEMPLATE = ('<html><head><meta http-equiv="Content-Type" '
|
|
'content="text/html; charset=utf-8"/><title>%s</title>'
|
|
'</head><body>\n%s\n</body></html>')
|
|
|
|
|
|
def html_encode(s):
|
|
return (s.replace('&', '&')
|
|
.replace('<', '<')
|
|
.replace('>', '>')
|
|
.replace('"', '"')
|
|
.replace("'", ''')
|
|
.replace('\n', '<br/>')
|
|
.replace(' ', ' '))
|
|
|
|
|
|
class SNBInput(InputFormatPlugin):
|
|
|
|
name = 'SNB Input'
|
|
author = 'Li Fanxi'
|
|
description = 'Convert SNB files to OEB'
|
|
file_types = {'snb'}
|
|
commit_name = 'snb_input'
|
|
|
|
options = set()
|
|
|
|
def convert(self, stream, options, file_ext, log,
|
|
accelerators):
|
|
import uuid
|
|
|
|
from ebook_converter.ebooks.oeb.base import DirContainer
|
|
from ebook_converter.ebooks.snb.snbfile import SNBFile
|
|
|
|
log.debug("Parsing SNB file...")
|
|
snbFile = SNBFile()
|
|
try:
|
|
snbFile.Parse(stream)
|
|
except Exception:
|
|
raise ValueError("Invalid SNB file")
|
|
if not snbFile.IsValid():
|
|
log.debug("Invalid SNB file")
|
|
raise ValueError("Invalid SNB file")
|
|
log.debug("Handle meta data ...")
|
|
from ebook_converter.ebooks.conversion.plumber import create_oebbook
|
|
oeb = create_oebbook(log, None, options,
|
|
encoding=options.input_encoding, populate=False)
|
|
meta = snbFile.GetFileStream('snbf/book.snbf')
|
|
if meta is not None:
|
|
meta = etree.fromstring(meta)
|
|
item_map = {'title': './/head/name',
|
|
'creator': './/head/author',
|
|
'language': './/head/language',
|
|
'generator': './/head/generator',
|
|
'publisher': './/head/publisher',
|
|
'cover': './/head/cover'}
|
|
d = {}
|
|
for key, item in item_map.items():
|
|
node = meta.find(item)
|
|
if node is not None:
|
|
d[key] = node.text if node.text is not None else ''
|
|
else:
|
|
d[key] = ''
|
|
|
|
oeb.metadata.add('title', d['title'])
|
|
oeb.metadata.add('creator', d['creator'], attrib={'role': 'aut'})
|
|
oeb.metadata.add('language',
|
|
d['language'].lower().replace('_', '-'))
|
|
oeb.metadata.add('generator', d['generator'])
|
|
oeb.metadata.add('publisher', d['publisher'])
|
|
if d['cover'] != '':
|
|
oeb.guide.add('cover', 'Cover', d['cover'])
|
|
|
|
bookid = str(uuid.uuid4())
|
|
oeb.metadata.add('identifier', bookid, id='uuid_id', scheme='uuid')
|
|
for ident in oeb.metadata.identifier:
|
|
if 'id' in ident.attrib:
|
|
oeb.uid = oeb.metadata.identifier[0]
|
|
break
|
|
|
|
with TemporaryDirectory('_snb2oeb', keep=True) as tdir:
|
|
log.debug('Process TOC ...')
|
|
toc = snbFile.GetFileStream('snbf/toc.snbf')
|
|
oeb.container = DirContainer(tdir, log)
|
|
if toc is not None:
|
|
toc = etree.fromstring(toc)
|
|
i = 1
|
|
for ch in toc.find('.//body'):
|
|
chapterName = ch.text
|
|
chapterSrc = ch.get('src')
|
|
fname = 'ch_%d.htm' % i
|
|
data = snbFile.GetFileStream('snbc/' + chapterSrc)
|
|
if data is None:
|
|
continue
|
|
snbc = etree.fromstring(data)
|
|
lines = []
|
|
for line in snbc.find('.//body'):
|
|
if line.tag == 'text':
|
|
lines.append('<p>%s</p>' % html_encode(line.text))
|
|
elif line.tag == 'img':
|
|
lines.append('<p><img src="%s" /></p>' %
|
|
html_encode(line.text))
|
|
with open(os.path.join(tdir, fname), 'wb') as f:
|
|
f.write((HTML_TEMPLATE %
|
|
(chapterName,
|
|
'\n'.join(lines))).encode('utf-8',
|
|
'replace'))
|
|
oeb.toc.add(ch.text, fname)
|
|
id, href = oeb.manifest.generate(
|
|
id='html', href=ascii_filename(fname))
|
|
item = oeb.manifest.add(id, href, 'text/html')
|
|
item.html_input_href = fname
|
|
oeb.spine.add(item, True)
|
|
i = i + 1
|
|
imageFiles = snbFile.OutputImageFiles(tdir)
|
|
for f, m in imageFiles:
|
|
id, href = oeb.manifest.generate(id='image',
|
|
href=ascii_filename(f))
|
|
item = oeb.manifest.add(id, href, m)
|
|
item.html_input_href = f
|
|
|
|
return oeb
|