Set index to 0 before saving if at end of set

[lightcards.git] / lightcards / lightcards.py

diff --git a/lightcards/lightcards.py b/lightcards/lightcards.py
index b4b459860ae535e95bc7a8fe72165d66b79d11a1..768ac6852a14eb9d18a9c2960125db0ced324b2f 100644 (file)
--- a/lightcards/lightcards.py
+++ b/lightcards/lightcards.py
@@ -13,7 +13,8 @@ from .deck import Status
 
 
 def parse_args():
-    parser = argparse.ArgumentParser(description="Terminal flashcards from Markdown")
+    parser = argparse.ArgumentParser(
+        description="Terminal flashcards from Markdown")
     parser.add_argument("inp",
                         metavar="input file",
                         type=str,
@@ -24,6 +25,7 @@ def parse_args():
     parser.add_argument("-p", "--purge",
                         action='store_true',
                         help="don't check cached info before starting")
+    # TODO: don't require input file when using  -P
     parser.add_argument("-P", "--purge-all",
                         action='store_true',
                         help="don't check cached info before starting")
@@ -41,37 +43,33 @@ def parse_args():
 
 def show(args, stack, headers):
     """
-    Manipulate deck according to passed arguments, and send it to the display
-    functions
+    Get objects from cache, manipulate deck according to passed arguments, and
+    send it to the display functions
     """
+    # Purge caches if asked
     if args.purge:
         progress.purge(stack)
     if args.purge_all:
         progress.purge_all()
 
-    ida = progress.dive("status", stack)
-    if ida and not args.purge:
-        idx = ida
+    # Check for caches
+    ida = progress.dive(stack)
+    if ida:
+        (idx, stack, headers) = ida
     else:
         idx = Status()
 
-    stackb = progress.dive("stack", stack)
-    if stackb and not args.purge:
-        stack = stackb
-
-    headerb = progress.dive("headers", stack)
-    if headerb and not args.purge:
-        headers = headerb
-
+    # Manipulate deck
     if args.flip:
         for x in stack:
             x[0], x[1] = x[1], x[0]
         headers[0], headers[1] = headers[1], headers[0]
     if args.shuffle:
         shuffle(stack)
-    elif args.reverse:
+    if args.reverse:
         stack.reverse()
 
+    # Send to display
     win = Display(stack, headers, idx)
     wrapper(win.run)
 
@@ -82,10 +80,12 @@ def reparse():
     os.system(f"$EDITOR {args.inp[0]}"),
     return parse.parse_html(parse.md2html(args.inp[0]))
 
+
 def get_orig():
     """Return original header and stack"""
     return((headers, stack))
 
+
 def main(args=sys.argv):
     args = parse_args()
     global headers, stack