lightcards/lightcards.py

   1 #!/usr/bin/env python
   2 # Markdown flashcard utility
   3 # Armaan Bhojwani 2021
   4
   5 import argparse
   6 from curses import wrapper
   7 from random import shuffle
   8 import sys
   9
  10 from . import display, parse
  11 from .deck import Status
  12
  13
  14 def parse_args():
  15     parser = argparse.ArgumentParser(description="Simple terminal flashcards")
  16     parser.add_argument("inp",
  17                         metavar="input file",
  18                         type=str,
  19                         nargs=1)
  20     parser.add_argument("-f", "--flip",
  21                         action='store_true',
  22                         help="show second column first")
  23     parser.add_argument("-r", "--reverse",
  24                         action='store_true',
  25                         help="reverse card order")
  26     parser.add_argument("-s", "--shuffle",
  27                         action='store_true',
  28                         help="shuffle card order")
  29     parser.add_argument("-v", "--version",
  30                         action='version',
  31                         version="lightcards 0.2.0")
  32     return parser.parse_args()
  33
  34
  35 def show(args, stack, headers):
  36     idx = Status()
  37     if args.flip:
  38         for x in stack:
  39             x[0], x[1] = x[1], x[0]
  40         headers[0], headers[1] = headers[1], headers[0]
  41     if args.shuffle:
  42         shuffle(stack)
  43     elif args.reverse:
  44         stack.reverse()
  45     wrapper(display.get_key, stack, headers, idx)
  46
  47
  48 def main(args=sys.argv):
  49     args = parse_args()
  50     headers = parse.parse_html(parse.md2html(args.inp[0]))[0]
  51     stack = parse.parse_html(parse.md2html(args.inp[0]))[1]
  52     show(args, stack, headers)
  53
  54
  55 if __name__ == "__main__":
  56     main()