Clean up code

[lightcards.git] / lightcards / display.py

diff --git a/lightcards/display.py b/lightcards/display.py
index 4dab9e1ad47a1e533a79c300c0ce1a674fe88b95..0ce321fcb1727105bd865b5d5f9ac8a1517b7f6b 100755 (executable)
--- a/lightcards/display.py
+++ b/lightcards/display.py
@@ -2,6 +2,8 @@
 # Armaan Bhojwani 2021
 
 import curses
+from random import shuffle
+import sys
 
 
 def disp_bar(stdscr, stack, headers, obj):
@@ -9,8 +11,9 @@ def disp_bar(stdscr, stack, headers, obj):
     if len(stack) <= 1:
         percent = "100"
     else:
-        percent = str(round(obj.getIdx() / (len(stack)) * 100)).zfill(3)
+        percent = str(round(obj.getIdx() / (len(stack) - 1) * 100)).zfill(3)
 
+    curses.init_pair(1, curses.COLOR_CYAN, 0)
     stdscr.insstr(mlines - 1, 0,
                   "[" +
                   stack[obj.getIdx()].printStar() +
@@ -23,17 +26,25 @@ def disp_bar(stdscr, stack, headers, obj):
                   ")]" +
                   " [" +
                   headers[obj.getSide()] +
-                  "]")
+                  "]", curses.color_pair(1))
 
 
 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")
+    stdscr.clear()
+    curses.init_pair(1, curses.COLOR_CYAN, 0)
+    curses.init_pair(2, curses.COLOR_RED, 0)
+    stdscr.addstr("Good job, you've completed a round!\n\n",
+                  curses.color_pair(1))
+    stdscr.addstr("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)
+            sys.exit(0)
         elif key == "r":
             idx.setIdx(0)
             get_key(stdscr, stack, headers, idx)
@@ -54,8 +65,13 @@ def disp_menu(stdscr, stack, headers, idx):
                 get_key(stdscr, stack, headers, idx)
             else:
                 stdscr.clear()
-                stdscr.addstr("ERR: Stack empty. Choose another option\n\n")
+                stdscr.addstr("ERR: Stack empty. Choose another option\n\n",
+                              curses.color_pair(2))
                 disp_menu(stdscr, stack, headers, idx)
+        elif key == "z":
+            idx.setIdx(0)
+            shuffle(stack)
+            get_key(stdscr, stack, headers, idx)
 
 
 def disp_card(stdscr, stack, headers, obj):
@@ -63,10 +79,19 @@ def disp_card(stdscr, stack, headers, obj):
     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)):
+        if obj.getSide() == 0:
+            top = headers[obj.getSide()] + "; " + str(obj.getIdx() + 1) + "\n"
+        else:
+            top = headers[obj.getSide()] + "; " + str(obj.getIdx() + 1) + \
+                "; " + str(stack[obj.getIdx()][0]) + "\n"
+
+        stdscr.addstr(top, curses.A_BOLD)
+        (mlines, mcols) = stdscr.getmaxyx()
+        if len(top) < mcols:
+            mcols = len(top)
+        for i in range(mcols):
             stdscr.addch("=")
+
         stdscr.addstr("\n" + str(stack[obj.getIdx()][obj.getSide()]))
     disp_bar(stdscr, stack, headers, obj)
 
@@ -78,7 +103,7 @@ def get_key(stdscr, stack, headers, idx):
     while True:
         key = stdscr.getkey()
         if key == "q":
-            exit(0)
+            sys.exit(0)
         elif key in ["l", "KEY_LEFT"]:
             idx.forward(stack)
             idx.setSide(0)
@@ -93,3 +118,9 @@ def get_key(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)