X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/5c55be51e46b296d368b86375f2f4b90ea46c00b..d26d6a7fdcb2c6e26fd097532765b62fd17b429c:/scripts/book2mobi

diff --git a/scripts/book2mobi b/scripts/book2mobi
index f477a83..b0d0686 100755
--- a/scripts/book2mobi
+++ b/scripts/book2mobi
@@ -4,7 +4,9 @@
 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
-from librarian.book2anything import Book2Anything
+from __future__ import unicode_literals
+
+from librarian.book2anything import Book2Anything, Option
 
 
 class Book2Mobi(Book2Anything):
@@ -14,6 +16,12 @@ class Book2Mobi(Book2Anything):
     cover_optional = False
     uses_provider = True
 
+    transform_options = [
+        Option('-k', '--use-kindlegen',
+                action='store_true', dest='use_kindlegen', default=False,
+                help='use kindlegen tool instead of Calibre')
+        ]
+
 
 if __name__ == '__main__':
     Book2Mobi.run()