X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/2258dea3cedb63cc7b0f0bcacec6d863381d42ca..5bc29f19f308610c944d63597962fb3b0f468c54:/librarian/mobi.py diff --git a/librarian/mobi.py b/librarian/mobi.py index 1e7569b..9558452 100644 --- a/librarian/mobi.py +++ b/librarian/mobi.py @@ -8,7 +8,7 @@ import os import subprocess from tempfile import NamedTemporaryFile -from librarian import OutputFile +from librarian import IOFile from librarian.cover import WLCover from librarian import get_resource @@ -19,7 +19,7 @@ def transform(wldoc, verbose=False, 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, """ @@ -31,10 +31,10 @@ def transform(wldoc, verbose=False, if not cover: cover = WLCover cover_file = NamedTemporaryFile(suffix='.png', delete=False) - c = cover(book_info) - c.save(cover_file) + bound_cover = cover(book_info) + bound_cover.save(cover_file) - if cover.uses_dc_cover: + 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: @@ -57,4 +57,4 @@ def transform(wldoc, verbose=False, 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 + return IOFile.from_filename(output_file.name) \ No newline at end of file