fixes for edge cases

[librarian.git] / scripts / book2mobi

diff --git a/scripts/book2mobi b/scripts/book2mobi
index b283309..174ef57 100755 (executable)
--- a/scripts/book2mobi
+++ b/scripts/book2mobi
@@ -4,6 +4,8 @@
 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
+from __future__ import unicode_literals
+
 from librarian.book2anything import Book2Anything, Option
 
 
@@ -15,10 +17,16 @@ class Book2Mobi(Book2Anything):
     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(
+            '-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__':