Fix validation.

[librarian.git] / scripts / book2mobi

diff --git a/scripts/book2mobi b/scripts/book2mobi
index f477a83..174ef57 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.
 #
 # 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):
 
 
 class Book2Mobi(Book2Anything):


@@ -14,6 +16,18 @@ class Book2Mobi(Book2Anything):
     cover_optional = False
     uses_provider = True
 
     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'
+        ),
+        Option(
+            '-b', '--base-url', dest='base_url', metavar='URL',
+            help='specifies the base URL for relative image references'
+        ),
+    ]
+

 
 if __name__ == '__main__':
     Book2Mobi.run()
 
 if __name__ == '__main__':
     Book2Mobi.run()