Add menu at end of stack

[lightcards.git] / lightcards / display.py

diff --git a/lightcards/display.py b/lightcards/display.py
index b276680adc3331689928bf7f32835d981f58e982..a4e562682402dda9e3023abf38a257f009e0e50e 100755 (executable)
--- a/lightcards/display.py
+++ b/lightcards/display.py
@@ -4,79 +4,95 @@
 import curses
 import os
 
-class Status():
-    index = 0
-    side = 0
+from .deck import Status
 
-    def forward(self, stack):
-        if not self.index == len(stack) - 1:
-            self.index += 1
 
-    def back(self):
-        if not self.index < 1:
-            self.index -= 1
+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)
 
-    def flip(self):
-        if self.side == 0:
-            self.side = 1
-        else:
-            self.side = 0
+    stdscr.insstr(mlines - 1, 0,
+                  "[" +
+                  stack[obj.getIdx()].printStar() +
+                  "] [" +
+                  percent +
+                  "% (" +
+                  str(obj.getIdx() + 1) +
+                  "/" +
+                  str(len(stack)) +
+                  ")]" +
+                  " [" +
+                  headers[obj.getSide()] +
+                  "]")
 
-    def setSide(self, inp):
-        self.side = inp
 
-    def getSide(self):
-        return self.side
+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
 
-    def getIdx(self):
-        return self.index
+            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()]))
-
-
-    (mlines, mcols) = stdscr.getmaxyx()
-    mlines -= 1
-    mcols -= 5
+    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)
 
-    try:
-        stdscr.insch(mlines, mcols, str(obj.getIdx() + 1))
-        stdscr.insch(mlines, mcols+1, '-')
-        if obj.getSide() == 0:
-            stdscr.insch(mlines, mcols+2, '1')
-        else:
-            stdscr.insch(mlines, mcols+2, '2')
-        stdscr.insch(mlines, mcols+3, '/')
-        stdscr.insch(mlines, mcols+4, str(len(stack)))
-    except:
-        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)
-        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)