From 0985b624b6c94f1b04b3bf71a56983c4a0b4da60 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Hugo=20H=C3=B6rnquist?= Date: Mon, 9 Nov 2020 21:38:22 +0100 Subject: Change to real arg parser. --- do.py | 16 ++++++++++++---- 1 file 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') -- cgit v1.2.3