Cache the stack and headers

[lightcards.git] / lightcards / lightcards.py

diff --git a/lightcards/lightcards.py b/lightcards/lightcards.py
old mode 100755 (executable)
new mode 100644 (file)
index ca3eac8..b4b4598
--- a/lightcards/lightcards.py
+++ b/lightcards/lightcards.py
@@ -1,18 +1,19 @@
-#!/usr/bin/env python
 # Markdown flashcard utility
 # Armaan Bhojwani 2021
 
 import argparse
 from curses import wrapper
+import os
 from random import shuffle
 import sys
 
-from . import display, parse
+from . import parse, progress
+from .display import Display
 from .deck import Status
 
 
 def parse_args():
-    parser = argparse.ArgumentParser(description="Simple terminal flashcards")
+    parser = argparse.ArgumentParser(description="Terminal flashcards from Markdown")
     parser.add_argument("inp",
                         metavar="input file",
                         type=str,
@@ -20,6 +21,12 @@ def parse_args():
     parser.add_argument("-f", "--flip",
                         action='store_true',
                         help="show second column first")
+    parser.add_argument("-p", "--purge",
+                        action='store_true',
+                        help="don't check cached info before starting")
+    parser.add_argument("-P", "--purge-all",
+                        action='store_true',
+                        help="don't check cached info before starting")
     parser.add_argument("-r", "--reverse",
                         action='store_true',
                         help="reverse card order")
@@ -28,12 +35,34 @@ def parse_args():
                         help="shuffle card order")
     parser.add_argument("-v", "--version",
                         action='version',
-                        version="lightcards 0.2.0")
+                        version="lightcards 0.4.0")
     return parser.parse_args()
 
 
 def show(args, stack, headers):
-    idx = Status()
+    """
+    Manipulate deck according to passed arguments, and send it to the display
+    functions
+    """
+    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
+    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
+
     if args.flip:
         for x in stack:
             x[0], x[1] = x[1], x[0]
@@ -42,13 +71,25 @@ def show(args, stack, headers):
         shuffle(stack)
     elif args.reverse:
         stack.reverse()
-    wrapper(display.get_key, stack, headers, idx)
 
+    win = Display(stack, headers, idx)
+    wrapper(win.run)
+
+
+def reparse():
+    """Parse arguments and input file again"""
+    args = parse_args()
+    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()
-    headers = parse.parse_html(parse.md2html(args.inp[0]))[0]
-    stack = parse.parse_html(parse.md2html(args.inp[0]))[1]
+    global headers, stack
+    (headers, stack) = parse.parse_html(parse.md2html(args.inp[0]))
     show(args, stack, headers)