Support for ref.

[librarian.git] / scripts / book2mobi

diff --git a/scripts/book2mobi b/scripts/book2mobi
index f477a83..b0d0686 100755 (executable)
--- 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()