Fix calculation of progress

[lightcards.git] / lightcards / lightcards.py

diff --git a/lightcards/lightcards.py b/lightcards/lightcards.py
index 23f94df29f34f04eb2251c7b47c31f7dcec5cd25..33da9ee19b9dede6731e6f01102fc9ed547ce67f 100755 (executable)
--- a/lightcards/lightcards.py
+++ b/lightcards/lightcards.py
@@ -4,8 +4,10 @@
 
 import argparse
 from curses import wrapper
+import sys
 
 from . import display, parse
+from .deck import Status
 
 
 def parse_args():
@@ -13,23 +15,20 @@ def parse_args():
     parser.add_argument("inp",
                         metavar="input file",
                         type=str,
-                        nargs="*")
+                        nargs=1)
     parser.add_argument("-v", "--version",
-                        action='store_true',
-                        help="print version")
+                        action='version',
+                        version="lightcards 0.0.0")
     return parser.parse_args()
 
 
 def show(stack, headers):
-    wrapper(display.get_key, stack, headers)
+    idx = Status()
+    wrapper(display.get_key, stack, headers, idx)
 
 
-def main():
+def main(args=sys.argv):
     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, headers)