mirror of
https://github.com/gryf/ebook-converter.git
synced 2026-01-02 16:54:12 +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.
83 lines
3.1 KiB
Python
83 lines
3.1 KiB
Python
#!/usr/bin/env python2
|
|
# vim:fileencoding=UTF-8:ts=4:sw=4:sta:et:sts=4:ai
|
|
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
|
|
__license__ = 'GPL v3'
|
|
__copyright__ = '2009, Kovid Goyal <kovid@kovidgoyal.net>'
|
|
__docformat__ = 'restructuredtext en'
|
|
|
|
import os, sys
|
|
from ebook_converter.customize.conversion import InputFormatPlugin
|
|
|
|
|
|
class LRFInput(InputFormatPlugin):
|
|
|
|
name = 'LRF Input'
|
|
author = 'Kovid Goyal'
|
|
description = 'Convert LRF files to HTML'
|
|
file_types = {'lrf'}
|
|
commit_name = 'lrf_input'
|
|
|
|
def convert(self, stream, options, file_ext, log,
|
|
accelerators):
|
|
from ebook_converter.ebooks.lrf.input import (MediaType, Styles, TextBlock,
|
|
Canvas, ImageBlock, RuledLine)
|
|
self.log = log
|
|
self.log('Generating XML')
|
|
from ebook_converter.ebooks.lrf.lrfparser import LRFDocument
|
|
from ebook_converter.utils.xml_parse import safe_xml_fromstring
|
|
from lxml import etree
|
|
d = LRFDocument(stream)
|
|
d.parse()
|
|
xml = d.to_xml(write_files=True)
|
|
if options.verbose > 2:
|
|
open(u'lrs.xml', 'wb').write(xml.encode('utf-8'))
|
|
doc = safe_xml_fromstring(xml)
|
|
|
|
char_button_map = {}
|
|
for x in doc.xpath('//CharButton[@refobj]'):
|
|
ro = x.get('refobj')
|
|
jump_button = doc.xpath('//*[@objid="%s"]'%ro)
|
|
if jump_button:
|
|
jump_to = jump_button[0].xpath('descendant::JumpTo[@refpage and @refobj]')
|
|
if jump_to:
|
|
char_button_map[ro] = '%s.xhtml#%s'%(jump_to[0].get('refpage'),
|
|
jump_to[0].get('refobj'))
|
|
plot_map = {}
|
|
for x in doc.xpath('//Plot[@refobj]'):
|
|
ro = x.get('refobj')
|
|
image = doc.xpath('//Image[@objid="%s" and @refstream]'%ro)
|
|
if image:
|
|
imgstr = doc.xpath('//ImageStream[@objid="%s" and @file]'%
|
|
image[0].get('refstream'))
|
|
if imgstr:
|
|
plot_map[ro] = imgstr[0].get('file')
|
|
|
|
self.log('Converting XML to HTML...')
|
|
styledoc = safe_xml_fromstring(P('templates/lrf.xsl', data=True))
|
|
media_type = MediaType()
|
|
styles = Styles()
|
|
text_block = TextBlock(styles, char_button_map, plot_map, log)
|
|
canvas = Canvas(doc, styles, text_block, log)
|
|
image_block = ImageBlock(canvas)
|
|
ruled_line = RuledLine()
|
|
extensions = {
|
|
('calibre', 'media-type') : media_type,
|
|
('calibre', 'text-block') : text_block,
|
|
('calibre', 'ruled-line') : ruled_line,
|
|
('calibre', 'styles') : styles,
|
|
('calibre', 'canvas') : canvas,
|
|
('calibre', 'image-block'): image_block,
|
|
}
|
|
transform = etree.XSLT(styledoc, extensions=extensions)
|
|
try:
|
|
result = transform(doc)
|
|
except RuntimeError:
|
|
sys.setrecursionlimit(5000)
|
|
result = transform(doc)
|
|
|
|
with open('content.opf', 'wb') as f:
|
|
f.write(result)
|
|
styles.write()
|
|
return os.path.abspath('content.opf')
|