Add shuffle functionality

[lightcards.git] / lightcards / lightcards.py

diff --git a/lightcards/lightcards.py b/lightcards/lightcards.py
index 33da9ee19b9dede6731e6f01102fc9ed547ce67f..15f8d67a9370ee486e9db338293b47cad512b313 100755 (executable)
--- a/lightcards/lightcards.py
+++ b/lightcards/lightcards.py
@@ -4,6 +4,7 @@
 
 import argparse
 from curses import wrapper
+from random import shuffle
 import sys
 
 from . import display, parse
@@ -19,11 +20,16 @@ def parse_args():
     parser.add_argument("-v", "--version",
                         action='version',
                         version="lightcards 0.0.0")
+    parser.add_argument("-s", "--shuffle",
+                        action='store_true',
+                        help="shuffle cards before starting")
     return parser.parse_args()
 
 
-def show(stack, headers):
+def show(args, stack, headers):
     idx = Status()
+    if args:
+        shuffle(stack)
     wrapper(display.get_key, stack, headers, idx)
 
 
@@ -31,7 +37,7 @@ 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]
-    show(stack, headers)
+    show(args, stack, headers)
 
 
 if __name__ == "__main__":