X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/f2d38866c775f83eb7d941345a1f4ac8a9133290..f981e2fe9c976709edb03f8617079319be6605e8:/librarian/mobi.py diff --git a/librarian/mobi.py b/librarian/mobi.py index 3c4c949..65f2598 100644 --- a/librarian/mobi.py +++ b/librarian/mobi.py @@ -3,18 +3,17 @@ # This file is part of Librarian, licensed under GNU Affero GPLv3 or later. # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. # + from copy import deepcopy import os import subprocess from tempfile import NamedTemporaryFile from librarian import OutputFile -from librarian.cover import DefaultEbookCover -from librarian import get_resource -def transform(wldoc, verbose=False, - sample=None, cover=None, flags=None): +def transform(wldoc, verbose=False, sample=None, cover=None, + use_kindlegen=False, flags=None, hyphenate=True, ilustr_path=''): """ produces a MOBI file wldoc: a WLDocument @@ -25,36 +24,27 @@ def transform(wldoc, verbose=False, document = deepcopy(wldoc) del wldoc - book_info = document.book_info - - # provide a cover by default - if not cover: - cover = DefaultEbookCover - cover_file = NamedTemporaryFile(suffix='.png', delete=False) - bound_cover = cover(book_info) - bound_cover.save(cover_file) - - if bound_cover.uses_dc_cover: - if document.book_info.cover_by: - document.edoc.getroot().set('data-cover-by', document.book_info.cover_by) - if document.book_info.cover_source: - document.edoc.getroot().set('data-cover-source', document.book_info.cover_source) - - if not flags: - flags = [] - flags = list(flags) + ['with-full-fonts'] - epub = document.as_epub(verbose=verbose, sample=sample, html_toc=True, - flags=flags, style=get_resource('epub/style.css')) + epub = document.as_epub(verbose=verbose, sample=sample, + html_toc=True, cover=cover or True, flags=flags, + hyphenate=hyphenate, ilustr_path=ilustr_path) if verbose: kwargs = {} else: devnull = open("/dev/null", 'w') kwargs = {"stdout": devnull, "stderr": devnull} - output_file = NamedTemporaryFile(prefix='librarian', suffix='.mobi', delete=False) + output_file = NamedTemporaryFile(prefix='librarian', suffix='.mobi', + delete=False) output_file.close() - subprocess.check_call(['ebook-convert', epub.get_filename(), output_file.name, - '--no-inline-toc', '--mobi-file-type=both', '--subset-embedded-fonts', '--mobi-ignore-margins', '--cover=%s' % cover_file.name], **kwargs) - os.unlink(cover_file.name) + + if use_kindlegen: + output_file_basename = os.path.basename(output_file.name) + subprocess.check_call(['kindlegen', '-c2', epub.get_filename(), + '-o', output_file_basename], **kwargs) + else: + subprocess.check_call(['ebook-convert', epub.get_filename(), + output_file.name, '--no-inline-toc', + '--mobi-file-type=both', + '--mobi-ignore-margins'], **kwargs) return OutputFile.from_filename(output_file.name)