lightcards/lightcards.py

   1 # Markdown flashcard utility
   2 # Armaan Bhojwani 2021
   3
   4 import argparse
   5 from curses import wrapper
   6 import os
   7 from random import shuffle
   8 import sys
   9
  10 from . import parse, progress
  11 from .display import Display
  12 from .deck import Status
  13
  14
  15 def parse_args():
  16     parser = argparse.ArgumentParser(
  17         description="Terminal flashcards from Markdown"
  18     )
  19     parser.add_argument("inp", metavar="input file", type=str, nargs=1)
  20     parser.add_argument(
  21         "-a",
  22         "--alphabetize",
  23         action="store_true",
  24         help="alphabetize card order",
  25     )
  26     parser.add_argument(
  27         "-f", "--flip", action="store_true", help="show second column first"
  28     )
  29     parser.add_argument(
  30         "-p",
  31         "--purge",
  32         action="store_true",
  33         help="don't check cached info before starting",
  34     )
  35     parser.add_argument(
  36         "-r", "--reverse", action="store_true", help="reverse card order"
  37     )
  38     parser.add_argument(
  39         "-s", "--shuffle", action="store_true", help="shuffle card order"
  40     )
  41     parser.add_argument(
  42         "-v", "--version", action="version", version="lightcards 0.6.0"
  43     )
  44     return parser.parse_args()
  45
  46
  47 def show(args, stack, headers):
  48     """
  49     Get objects from cache, manipulate deck according to passed arguments, and
  50     send it to the display functions
  51     """
  52     # Purge caches if asked
  53     if args.purge:
  54         progress.purge(stack)
  55
  56     # Check for caches
  57     idx = Status()
  58     cache = progress.dive(get_orig())
  59     if cache:
  60         (stack) = cache
  61
  62     # Manipulate deck
  63     if args.shuffle:
  64         shuffle(stack)
  65     if args.alphabetize:
  66         stack.sort()
  67     if args.reverse:
  68         stack.reverse()
  69     if args.flip:
  70         for x in stack:
  71             x[0], x[1] = x[1], x[0]
  72         headers[0], headers[1] = headers[1], headers[0]
  73
  74     # Send to display
  75     win = Display(stack, headers, idx)
  76     wrapper(win.run)
  77
  78
  79 def reparse():
  80     """Parse arguments and input file again"""
  81     args = parse_args()
  82     os.system(f"$EDITOR {args.inp[0]}"),
  83     return parse.parse_html(parse.md2html(args.inp[0]))
  84
  85
  86 def get_orig():
  87     """Return original header and stack"""
  88     return (headers, stack)
  89
  90
  91 def main(args=sys.argv):
  92     args = parse_args()
  93     global headers, stack
  94     (headers, stack) = parse.parse_html(parse.md2html(args.inp[0]))
  95     show(args, stack, headers)
  96
  97
  98 if __name__ == "__main__":
  99     main()