1 # -*- coding: utf-8 -*-
3 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
6 from copy import deepcopy
9 from tempfile import NamedTemporaryFile
11 from librarian import OutputFile
12 from librarian.cover import ImageCover as WLCover
13 from librarian import get_resource
16 def transform(wldoc, verbose=False,
17 sample=None, cover=None, flags=None):
18 """ produces a MOBI file
21 sample=n: generate sample e-book (with at least n paragraphs)
22 cover: a cover.Cover object
23 flags: less-advertising,
26 document = deepcopy(wldoc)
28 book_info = document.book_info
30 # provide a cover by default
35 c.im = Image.open('cover.jpg')
37 cover_file = NamedTemporaryFile(suffix='.' + c.ext(), delete=False)
40 if cover.uses_dc_cover:
41 if document.book_info.cover_by:
42 document.edoc.getroot().set('data-cover-by', document.book_info.cover_by)
43 if document.book_info.cover_source:
44 document.edoc.getroot().set('data-cover-source', document.book_info.cover_source)
48 flags = list(flags) + ['without-fonts']
49 epub = document.as_epub(verbose=verbose, sample=sample, html_toc=True,
50 flags=flags, style=get_resource('mobi/style.css'))
55 devnull = open("/dev/null", 'w')
56 kwargs = {"stdout": devnull, "stderr": devnull}
58 output_file = NamedTemporaryFile(prefix='librarian', suffix='.mobi', delete=False)
60 subprocess.check_call(['ebook-convert', epub.get_filename(), output_file.name,
61 '--no-inline-toc', '--cover=%s' % cover_file.name], **kwargs)
62 os.unlink(cover_file.name)
63 return OutputFile.from_filename(output_file.name)