# This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
#
-import os
-import optparse
+from __future__ import unicode_literals
-from librarian import ParseError
-from librarian.parser import WLDocument
-from librarian.cover import WLCover
+from librarian.book2anything import Book2Anything, Option
-if __name__ == '__main__':
- # Parse commandline arguments
- usage = """Usage: %prog [options] SOURCE [SOURCE...]
- Create cover images for SOURCE files."""
-
- parser = optparse.OptionParser(usage=usage)
-
- parser.add_option('-v', '--verbose', action='store_true', dest='verbose', default=False,
- help='print status messages to stdout')
+class Book2Cover(Book2Anything):
+ format_name = "JPEG"
+ ext = "jpg"
+ uses_cover = True
+ cover_optional = False
- options, input_filenames = parser.parse_args()
+ transform_options = [
+ Option('-W', '--width', action='store', type='int', dest='width', default=None,
+ help='Set width.'),
+ Option('-H', '--height', action='store', type='int', dest='height', default=None,
+ help='Set height.'),
+ ]
- if len(input_filenames) < 1:
- parser.print_help()
- exit(1)
+ @staticmethod
+ def transform(wldoc, cover, *args, **kwargs):
+ return wldoc.as_cover(cover_class=cover, *args, **kwargs)
- # Do some real work
- for input_filename in input_filenames:
- if options.verbose:
- print input_filename
- output_filename = os.path.splitext(input_filename)[0] + '.png'
-
- doc = WLDocument.from_file(input_filename)
- WLCover(doc.book_info).save(output_filename)
+if __name__ == '__main__':
+ Book2Cover.run()