X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/e57b146bf49e38b3bb57615110b27de5b4d1ae69..deca3e31b94bc7ec95d47c309ff560a097337e97:/scripts/book2mobi?ds=inline

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()