mirror of
https://github.com/gryf/ebook-converter.git
synced 2026-02-26 12:15:57 +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.
110 lines
3.7 KiB
Python
110 lines
3.7 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__ = '2012, Kovid Goyal <kovid@kovidgoyal.net>'
|
|
__docformat__ = 'restructuredtext en'
|
|
|
|
import os, glob
|
|
|
|
from ebook_converter import CurrentDir
|
|
from ebook_converter.ebooks.mobi import MobiError
|
|
from ebook_converter.ebooks.mobi.reader.mobi6 import MobiReader
|
|
from ebook_converter.ebooks.mobi.reader.headers import MetadataHeader
|
|
from ebook_converter.utils.logging import default_log
|
|
from ebook_converter.ebooks import DRMError
|
|
from ebook_converter.ebooks.mobi.reader.mobi8 import Mobi8Reader
|
|
from ebook_converter.ebooks.conversion.plumber import Plumber, create_oebbook
|
|
from ebook_converter.customize.ui import (plugin_for_input_format,
|
|
plugin_for_output_format)
|
|
from ebook_converter.utils.ipc.simple_worker import fork_job
|
|
|
|
|
|
class BadFormat(ValueError):
|
|
pass
|
|
|
|
|
|
def do_explode(path, dest):
|
|
with open(path, 'rb') as stream:
|
|
mr = MobiReader(stream, default_log, None, None)
|
|
|
|
with CurrentDir(dest):
|
|
mr = Mobi8Reader(mr, default_log)
|
|
opf = os.path.abspath(mr())
|
|
try:
|
|
os.remove('debug-raw.html')
|
|
except:
|
|
pass
|
|
|
|
return opf
|
|
|
|
|
|
def explode(path, dest, question=lambda x:True):
|
|
with open(path, 'rb') as stream:
|
|
raw = stream.read(3)
|
|
stream.seek(0)
|
|
if raw == b'TPZ':
|
|
raise BadFormat(_('This is not a MOBI file. It is a Topaz file.'))
|
|
|
|
try:
|
|
header = MetadataHeader(stream, default_log)
|
|
except MobiError:
|
|
raise BadFormat(_('This is not a MOBI file.'))
|
|
|
|
if header.encryption_type != 0:
|
|
raise DRMError(_('This file is locked with DRM. It cannot be tweaked.'))
|
|
|
|
kf8_type = header.kf8_type
|
|
|
|
if kf8_type is None:
|
|
raise BadFormat(_('This MOBI file does not contain a KF8 format '
|
|
'book. KF8 is the new format from Amazon. calibre can '
|
|
'only tweak MOBI files that contain KF8 books. Older '
|
|
'MOBI files without KF8 are not tweakable.'))
|
|
|
|
if kf8_type == 'joint':
|
|
if not question(_('This MOBI file contains both KF8 and '
|
|
'older Mobi6 data. Tweaking it will remove the Mobi6 data, which '
|
|
'means the file will not be usable on older Kindles. Are you '
|
|
'sure?')):
|
|
return None
|
|
|
|
return fork_job('ebook_converter.ebooks.mobi.tweak', 'do_explode', args=(path,
|
|
dest), no_output=True)['result']
|
|
|
|
|
|
def set_cover(oeb):
|
|
if 'cover' not in oeb.guide or oeb.metadata['cover']:
|
|
return
|
|
cover = oeb.guide['cover']
|
|
if cover.href in oeb.manifest.hrefs:
|
|
item = oeb.manifest.hrefs[cover.href]
|
|
oeb.metadata.clear('cover')
|
|
oeb.metadata.add('cover', item.id)
|
|
|
|
|
|
def do_rebuild(opf, dest_path):
|
|
plumber = Plumber(opf, dest_path, default_log)
|
|
plumber.setup_options()
|
|
inp = plugin_for_input_format('azw3')
|
|
outp = plugin_for_output_format('azw3')
|
|
|
|
plumber.opts.mobi_passthrough = True
|
|
oeb = create_oebbook(default_log, opf, plumber.opts)
|
|
set_cover(oeb)
|
|
outp.convert(oeb, dest_path, inp, plumber.opts, default_log)
|
|
|
|
|
|
def rebuild(src_dir, dest_path):
|
|
opf = glob.glob(os.path.join(src_dir, '*.opf'))
|
|
if not opf:
|
|
raise ValueError('No OPF file found in %s'%src_dir)
|
|
opf = opf[0]
|
|
# For debugging, uncomment the following two lines
|
|
# def fork_job(a, b, args=None, no_output=True):
|
|
# do_rebuild(*args)
|
|
fork_job('ebook_converter.ebooks.mobi.tweak', 'do_rebuild', args=(opf, dest_path),
|
|
no_output=True)
|
|
|