X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/b6ec0976a383cc1823f4a199bc3e6dc40880b049..bddf82ce5ec2a7c34b6b14d3bdb6c4cf24af7388:/librarian/mobi.py diff --git a/librarian/mobi.py b/librarian/mobi.py old mode 100755 new mode 100644 index a93315e..82ff343 --- a/librarian/mobi.py +++ b/librarian/mobi.py @@ -3,40 +3,36 @@ # 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 WLCover +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): """ produces a MOBI file wldoc: a WLDocument sample=n: generate sample e-book (with at least n paragraphs) - cover: a cover.Cover object + cover: a cover.Cover factory overriding default flags: less-advertising, """ - book_info = wldoc.book_info - - # provide a cover by default - if not cover: - cover = WLCover - cover_file = NamedTemporaryFile(suffix='.png', delete=False) - c = cover(book_info.author.readable(), book_info.title) - c.save(cover_file) + document = deepcopy(wldoc) + del wldoc + book_info = document.book_info if not flags: flags = [] - flags = list(flags) + ['without-fonts'] - epub = wldoc.as_epub(verbose=verbose, sample=sample, html_toc=True, - flags=flags, style=get_resource('mobi/style.css')) - + flags = list(flags) + + epub = document.as_epub(verbose=verbose, sample=sample, + html_toc=True, cover=True, flags=flags) if verbose: kwargs = {} else: @@ -45,7 +41,18 @@ def transform(wldoc, verbose=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', '--cover=%s' % cover_file.name], **kwargs) - os.unlink(cover_file.name) - return OutputFile.from_filename(output_file.name) \ No newline at end of file + + 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)