# Armaan Bhojwani 2021
import curses
+from random import shuffle
def disp_bar(stdscr, stack, headers, obj):
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")
+ "[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":
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)
def disp_card(stdscr, stack, headers, obj):