Add menu at end of stack

[lightcards.git] / lightcards / display.py

diff --git a/lightcards/display.py b/lightcards/display.py
index 349d77abbb9417c9b363ea6943c59cf2166ae78e..a4e562682402dda9e3023abf38a257f009e0e50e 100755 (executable)
--- a/lightcards/display.py
+++ b/lightcards/display.py
@@ -7,56 +7,92 @@ import os
 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)) * 100)).zfill(3)
+
+    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]estart, [s]tarred only, [u]nstar all and restart, [q]uit")
+    while True:
+        key = stdscr.getkey()
+        if key == "q" or key == os.linesep:
+            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)
+
+
 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()]))
+    if obj.getIdx() == len(stack):
+        disp_menu(stdscr, stack, headers, obj)
+    else:
+        side_title = headers[obj.getSide()]
+        stdscr.addstr(side_title + "\n")
+        for i in range(len(side_title)):
+            stdscr.addch("=")
+        stdscr.addstr("\n" + str(stack[obj.getIdx()][obj.getSide()]))
+    disp_bar(stdscr, stack, headers, obj)
 
-    (mlines, mcols) = stdscr.getmaxyx()
-    try:
-        stdscr.insstr(mlines - 1, 1,
-                      "[" +
-                      stack[obj.getIdx()].getStar() +
-                      "] [" +
-                      str(round(obj.getIdx() / (len(stack) - 1) * 100)).zfill(3) +
-                      "% (" +
-                      str(obj.getIdx() + 1) +
-                      "/" +
-                      str(len(stack)) +
-                      ")]" +
-                      " [" +
-                      headers[obj.getSide()] +
-                      "]")
-    except Exception:
-        pass
-
-
-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)
-            elif key in ["i", "/"]:
-                stack[idx.getIdx()].toggleStar()
-                disp_card(stdscr, stack, headers, idx)
-        except Exception:
-            pass
+        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)
+        elif key in ["i", "/"]:
+            stack[idx.getIdx()].toggleStar()
+            disp_card(stdscr, stack, headers, idx)