improve arguments handling in reduce.py

2 files changed, 7 insertions, 4 deletions

diff --git a/reduce.py b/reduce.py
index 62119e5..6ad9824 100755
--- a/reduce.py
+++ b/reduce.py
@@ -47,13 +47,15 @@ def iterateSubDirectory(oldroot, newroot, level):
 
 if __name__ == '__main__':
     parser = ArgumentParser(description='Reduce the levels of categories.')
-    parser.add_argument('--level', action='store', nargs=1, default=2, \
+    parser.add_argument('--level', action='store', default=2, \
                             help='reduce to n levels of index')
     parser.add_argument('origdir', action='store', \
                             help='original index directory')
-    parser.add_argument('destdir', action='store', \
-                            help='reduced index directory')
+    parser.add_argument('--destdir', action='store', \
+                            help='reduced index directory', \
+                            default=os.path.join(config.getTextDir(), 'index'))
 
     args = parser.parse_args()
-    iterateSubDirectory(args.origdir, args.destdir, int(args.level[0]))
+    print(args)
+    iterateSubDirectory(args.origdir, args.destdir, args.level)
     print('done')
diff --git a/segment.py b/segment.py
index 45d4f93..fbd7f65 100755
--- a/segment.py
+++ b/segment.py
@@ -87,6 +87,7 @@ if __name__ == '__main__':
                             default=os.path.join(config.getTextDir(), 'index'))
 
     args = parser.parse_args()
+    print(args)
     walkThroughIndex(args.indexdir)
     print('done')