Add shuffle functionality

[lightcards.git] / lightcards / display.py

diff --git a/lightcards/display.py b/lightcards/display.py
index 42a01f220d2e57228f097925cda50bea07bb2a14..7a6ec1b22ffbb0bd667e05fde4ef638b88daa018 100755 (executable)
--- a/lightcards/display.py
+++ b/lightcards/display.py
@@ -2,56 +2,114 @@
 # Armaan Bhojwani 2021
 
 import curses
-import os
+from random import shuffle
 
-from .deck import Status
 
+def disp_bar(stdscr, stack, headers, obj):
+    (mlines, mcols) = stdscr.getmaxyx()
+    if len(stack) <= 1:
+        percent = "100"
+    else:
+        percent = str(round(obj.getIdx() / (len(stack) - 1) * 100)).zfill(3)
 
-def disp_card(stdscr, stack, headers, obj):
-    stdscr.clear()
-    side_title = headers[obj.getSide()]
-    stdscr.addstr(side_title + "\n")
-    for i in range(len(side_title)):
-        stdscr.addstr("=")
-    stdscr.addstr("\n" + str(stack[obj.getIdx()][obj.getSide()]))
+    stdscr.insstr(mlines - 1, 0,
+                  "[" +
+                  stack[obj.getIdx()].printStar() +
+                  "] [" +
+                  percent +
+                  "% (" +
+                  str(obj.getIdx() + 1) +
+                  "/" +
+                  str(len(stack)) +
+                  ")]" +
+                  " [" +
+                  headers[obj.getSide()] +
+                  "]")
+
+
+def disp_menu(stdscr, stack, headers, idx):
+    stdscr.addstr("Good job, you've completed a round!\n\n" +
+                  "Choose one of the following options:\n" +
+                  "[r]: restart\n" +
+                  "[s]: restart with starred only\n" +
+                  "[u]: restart and unstar all\n" +
+                  "[z]: restart and shuffle cards\n" +
+                  "[q]: quit")
+    while True:
+        key = stdscr.getkey()
+        if key == "q":
+            exit(0)
+        elif key == "r":
+            idx.setIdx(0)
+            get_key(stdscr, stack, headers, idx)
+        elif key == "u":
+            idx.setIdx(0)
+            for x in stack:
+                x.unStar()
+            get_key(stdscr, stack, headers, idx)
+        elif key == "s":
+            cont = False
+            for x in stack:
+                if x.getStar():
+                    cont = True
+
+            if cont:
+                idx.setIdx(0)
+                stack = [x for x in stack if x.getStar()]
+                get_key(stdscr, stack, headers, idx)
+            else:
+                stdscr.clear()
+                stdscr.addstr("ERR: Stack empty. Choose another option\n\n")
+                disp_menu(stdscr, stack, headers, idx)
+        elif key == "z":
+            idx.setIdx(0)
+            shuffle(stack)
+            get_key(stdscr, stack, headers, idx)
 
-    (mlines, mcols) = stdscr.getmaxyx()
-    mlines -= 1
-    mcols -= 5
 
-    try:
-        stdscr.insch(mlines, mcols, str(obj.getIdx() + 1))
-        stdscr.insch(mlines, mcols+1, '-')
+def disp_card(stdscr, stack, headers, obj):
+    stdscr.clear()
+    if obj.getIdx() == len(stack):
+        disp_menu(stdscr, stack, headers, obj)
+    else:
         if obj.getSide() == 0:
-            stdscr.insch(mlines, mcols+2, '1')
+            top = headers[obj.getSide()] + " " + str(obj.getIdx() + 1) + "\n"
         else:
-            stdscr.insch(mlines, mcols+2, '2')
-        stdscr.insch(mlines, mcols+3, '/')
-        stdscr.insch(mlines, mcols+4, str(len(stack)))
-    except Exception:
-        pass
+            top = headers[obj.getSide()] + " " + str(obj.getIdx() + 1) + \
+                "; " + str(stack[obj.getIdx()][0]) + "\n"
+
+        stdscr.addstr(top)
+        for i in range(len(top)):
+            stdscr.addch("=")
+        stdscr.addstr("\n" + str(stack[obj.getIdx()][obj.getSide()]))
+    disp_bar(stdscr, stack, headers, obj)
 
 
-def get_key(stdscr, stack, headers):
-    idx = Status()
+def get_key(stdscr, stack, headers, idx):
     curses.curs_set(0)
     disp_card(stdscr, stack, headers, idx)
 
     while True:
         key = stdscr.getkey()
-        try:
-            if key == "q" or key == os.linesep:
-                exit(0)
-            elif key in ["l", "KEY_LEFT"]:
-                idx.forward(stack)
-                idx.setSide(0)
-                disp_card(stdscr, stack, headers, idx)
-            elif key in ["h", "KEY_RIGHT"]:
-                idx.back()
-                idx.setSide(0)
-                disp_card(stdscr, stack, headers,  idx)
-            elif key in ["j", "k", "KEY_UP", "KEY_DOWN"]:
-                idx.flip()
-                disp_card(stdscr, stack, headers, idx)
-        except Exception:
-            pass
+        if key == "q":
+            exit(0)
+        elif key in ["l", "KEY_LEFT"]:
+            idx.forward(stack)
+            idx.setSide(0)
+            disp_card(stdscr, stack, headers, idx)
+        elif key in ["h", "KEY_RIGHT"]:
+            idx.back()
+            idx.setSide(0)
+            disp_card(stdscr, stack, headers,  idx)
+        elif key in ["j", "k", "KEY_UP", "KEY_DOWN"]:
+            idx.flip()
+            disp_card(stdscr, stack, headers, idx)
+        elif key in ["i", "/"]:
+            stack[idx.getIdx()].toggleStar()
+            disp_card(stdscr, stack, headers, idx)
+        elif key in ["0", "^", "KEY_HOME"]:
+            idx.setIdx(0)
+            disp_card(stdscr, stack, headers, idx)
+        elif key in ["$", "KEY_END"]:
+            idx.setIdx(len(stack) - 1)
+            disp_card(stdscr, stack, headers, idx)