Add colors and text styling

[lightcards.git] / lightcards / display.py

diff --git a/lightcards/display.py b/lightcards/display.py
index aa331b107ad12b7a7afb2cbd40d94b1c67953a25..59a5bca194a0a8500728a477d34d00ba2190e429 100755 (executable)
--- a/lightcards/display.py
+++ b/lightcards/display.py
@@ -2,6 +2,7 @@
 # Armaan Bhojwani 2021
 
 import curses
+from random import shuffle
 
 
 def disp_bar(stdscr, stack, headers, obj):
@@ -11,6 +12,7 @@ def disp_bar(stdscr, stack, headers, obj):
     else:
         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,13 +25,20 @@ 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")
+    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":
@@ -54,8 +63,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):
@@ -69,9 +83,13 @@ def disp_card(stdscr, stack, headers, obj):
             top = headers[obj.getSide()] + " " + str(obj.getIdx() + 1) + \
                 "; " + str(stack[obj.getIdx()][0]) + "\n"
 
-        stdscr.addstr(top)
-        for i in range(len(top)):
+        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)