Add -v flag

[lightcards.git] / lightcards / lightcards.py

diff --git a/lightcards/lightcards.py b/lightcards/lightcards.py
index c744f3a55a139d63b5d70361a34efe7095059c99..968f86c51c3eb887b155d15ef78beb4931e06b97 100755 (executable)
--- a/lightcards/lightcards.py
+++ b/lightcards/lightcards.py
@@ -10,7 +10,13 @@ from . import display, parse
 
 def parse_args():
     parser = argparse.ArgumentParser(description="Simple terminal flashcards")
-    parser.add_argument("inp", metavar="input file", type=str, nargs=1)
+    parser.add_argument("inp",
+                        metavar="input file",
+                        type=str,
+                        nargs="*")
+    parser.add_argument("-v", "--version",
+                        action='store_true',
+                        help="print version")
     return parser.parse_args()
 
 
@@ -20,6 +26,10 @@ def show(stack):
 
 def main():
     args = parse_args()
+    if args.version:
+        print("0.0.0")
+        exit(0)
+
     headers = parse.parse_html(parse.md2html(args.inp[0]))[0]
     stack = parse.parse_html(parse.md2html(args.inp[0]))[1]
     show(stack)