1 files changed, 12 insertions, 4 deletions
diff --git a/do.py b/do.py
index c003b4f..7015e6b 100755
--- a/do.py
+++ b/do.py
@@ -2,7 +2,6 @@
 
 import re
 import os
-import sys
 import pwd
 import subprocess
 from subprocess import PIPE
@@ -11,11 +10,19 @@ import os.path as path
 import email
 from email.policy import default
 from datetime import datetime
+import argparse
 
 path_base = '/'
 
-# mode = 'tex'
-mode = 'txt'
+parser = argparse.ArgumentParser(description='parse vimwiki files')
+parser.add_argument('--mode', choices=["tex", "txt"],
+        help="Format of output")
+parser.add_argument('infile', help="Wiki-file to use for input")
+parser.add_argument('heading', help="Which heading to output")
+args = parser.parse_args()
+
+
+mode = args.mode
 
 # ------------------------------------------------------------
 
@@ -202,7 +209,8 @@ def get_heading(name, data):
 
 # ------------------------------------------------------------
 
-[_, infile, heading, *rest] = sys.argv
+infile = args.infile
+heading = args.heading
 
 
 file = open(infile, 'r')