Update installation

[lightcards.git] / lightcards / lightcards.py

diff --git a/lightcards/lightcards.py b/lightcards/lightcards.py
index 968f86c51c3eb887b155d15ef78beb4931e06b97..62019a52b6078f07d3999e27cfda033f906c70e0 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():
@@ -20,11 +22,12 @@ def parse_args():
     return parser.parse_args()
 
 
-def show(stack):
-    wrapper(display.get_key, stack)
+def show(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")
@@ -32,7 +35,7 @@ def main():
 
     headers = parse.parse_html(parse.md2html(args.inp[0]))[0]
     stack = parse.parse_html(parse.md2html(args.inp[0]))[1]
-    show(stack)
+    show(stack, headers)
 
 
 if __name__ == "__main__":