Fix test.

[librarian.git] / scripts / book2txt

diff --git a/scripts/book2txt b/scripts/book2txt
index e584579..c706a07 100755 (executable)
--- a/scripts/book2txt
+++ b/scripts/book2txt
@@ -5,6 +5,7 @@
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 from librarian.book2anything import Book2Anything, Option
+from librarian.parser import WLDocument
 
 
 class Book2Txt(Book2Anything):
@@ -12,15 +13,21 @@ class Book2Txt(Book2Anything):
     ext = "txt"
     uses_cover = False
     uses_provider = False
-    parser_args = [
+    parser_options = [
         Option('-i', '--ignore-dublin-core', dest='parse_dublincore', 
                 action='store_false', default=True,
                 help='don\'t try to parse dublin core metadata')
     ]
-    transform_args = [
+    transform_options = [
         Option('-w', '--wrap', action='store', type='int', dest='wrapping', default=0,
                 help='set line wrap column')
     ]
+    transform_flags = [
+        Option('-r', '--raw', dest='raw-text', 
+                action='store_true', default=False,
+                help='Produce raw text, without any surrounding info.')
+    ]
+    transform = WLDocument.as_text
 
 
 if __name__ == '__main__':