from librarian import DirDocProvider, ParseError
from librarian.parser import WLDocument
-from librarian.cover import WLCover
+from librarian.cover import WLCover, FutureOfCopyrightCover, CoverFromFile
class Option(object):
# Add cover support, if any.
if cls.uses_cover:
if options.image_cache:
- transform_args['cover'] = lambda x: WLCover(x, image_cache = options.image_cache)
+ def cover_class(*args, **kwargs):
+ return FutureOfCopyrightCover(image_cache=options.image_cache, *args, **kwargs)
+ transform_args['cover'] = cover_class
+ elif options.cover_file:
+ def cover_class(*args, **kwargs):
+ return CoverFromFile(options.cover_file, *args, **kwargs)
+ transform_args['cover'] = cover_class
elif not cls.cover_optional or options.with_cover:
- transform_args['cover'] = WLCover
+ transform_args['cover'] = FutureOfCopyrightCover
# Do some real work