X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/ad7b5c75f9161ac14f1f26032f4b4d54077b65a4..f4fdb08e0b0ff0da9faed1ac5d89a1523cad7271:/librarian/packagers.py?ds=sidebyside diff --git a/librarian/packagers.py b/librarian/packagers.py index a32a2a0..f57a983 100644 --- a/librarian/packagers.py +++ b/librarian/packagers.py @@ -4,9 +4,11 @@ # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. # import os -from librarian import pdf, epub, mobi, DirDocProvider, ParseError, cover +from librarian import pdf, epub, mobi, DirDocProvider, ParseError from librarian.parser import WLDocument +from util import makedirs + class Packager(object): cover = None @@ -23,20 +25,15 @@ class Packager(object): slug, ext = os.path.splitext(fname) if output_dir != '': - try: - os.makedirs(output_dir) - except: - pass + makedirs(output_dir) outfile = os.path.join(output_dir, slug + '.' + cls.ext) if os.path.exists(outfile) and not overwrite: return doc = WLDocument.from_file(main_input, provider=provider) - output_file = cls.transform(doc, - cover=cls.cover, flags=cls.flags) + output_file = cls.transform(doc, cover=cls.cover, flags=cls.flags) doc.save_output_file(output_file, output_path=outfile) - @classmethod def prepare(cls, input_filenames, output_dir='', verbose=False, overwrite=False): try: @@ -56,10 +53,12 @@ class EpubPackager(Packager): converter = epub ext = 'epub' + class MobiPackager(Packager): converter = mobi ext = 'mobi' + class PdfPackager(Packager): converter = pdf ext = 'pdf'