Initial import

ebook_converter/ebooks/mobi/tweak.py

@@ -0,0 +1,109 @@
+#!/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 calibre import CurrentDir
+from calibre.ebooks.mobi import MobiError
+from calibre.ebooks.mobi.reader.mobi6 import MobiReader
+from calibre.ebooks.mobi.reader.headers import MetadataHeader
+from calibre.utils.logging import default_log
+from calibre.ebooks import DRMError
+from calibre.ebooks.mobi.reader.mobi8 import Mobi8Reader
+from calibre.ebooks.conversion.plumber import Plumber, create_oebbook
+from calibre.customize.ui import (plugin_for_input_format,
+        plugin_for_output_format)
+from calibre.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('calibre.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('calibre.ebooks.mobi.tweak', 'do_rebuild', args=(opf, dest_path),
+            no_output=True)
+