Highlight keys when wrong button is pressed

This commit is contained in:
Tanner Collin 2022-09-07 15:00:20 -06:00
parent abba72c03c
commit 3b0e1c53fa

81
main.py
View File

@ -15,6 +15,7 @@ import requests
import pytz
import re
import os
import time
from datetime import datetime
try:
@ -168,19 +169,14 @@ skip_input = False
current_screen = 'home'
prev_screen = current_screen
c = 0
# highlighting:
#wattron(menu_win, A_REVERSE)
#mvwaddstr(menu_win, y, x, choices[i])
#wattroff(menu_win, A_REVERSE)
stdscr = curses.initscr()
curses.noecho()
curses.cbreak()
stdscr.keypad(True)
curses.curs_set(0)
highlight_keys = False
highlight_debounce = time.time()
sign_to_send = ''
think_to_send = ''
think_result = ''
@ -194,6 +190,8 @@ info_line = 0
logging.info('Starting main loop...')
while True:
c = 0
if current_screen == 'home':
stdscr.addstr(0, 1, ' _______ _______ ___ _________ ___ ____ ____ _ ______ ')
stdscr.addstr(1, 1, '|_ __ \|_ __ \ .\' `. | _ _ | .\' `.|_ _| |_ _|/ \ .\' ___ |')
@ -204,22 +202,29 @@ while True:
stdscr.addstr(6, 1, '')
stdscr.addstr(7, 1, ' UNIVERSAL COMPUTER')
stdscr.addstr(8, 1, '')
stdscr.addstr(9, 4, '[I] Info')
stdscr.addstr(9, 4, '[I]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(9, 8, 'Info')
stdscr.addstr(10, 1, '')
stdscr.addstr(11, 4, '[S] Stats')
stdscr.addstr(11, 4, '[S]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(11, 8, 'Stats')
stdscr.addstr(12, 1, '')
stdscr.addstr(13, 4, '[N] Sign')
stdscr.addstr(13, 4, '[N]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(13, 8, 'Sign')
stdscr.addstr(14, 1, '')
stdscr.addstr(15, 4, '[C] Classes')
stdscr.addstr(15, 4, '[C]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(15, 8, 'Classes')
stdscr.addstr(16, 1, '')
stdscr.addstr(17, 4, '[P] Protocoin')
stdscr.addstr(17, 4, '[P]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(17, 8, 'Protocoin')
stdscr.addstr(18, 1, '')
if wa_api_key:
stdscr.addstr(19, 4, '[T] Think')
stdscr.addstr(19, 4, '[T]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(19, 8, 'Think')
else:
stdscr.addstr(19, 1, '')
stdscr.addstr(20, 1, '')
stdscr.addstr(21, 4, '[A] About')
stdscr.addstr(21, 4, '[A]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(21, 8, 'About')
stdscr.addstr(22, 1, '')
stdscr.addstr(23, 1, ' Copyright (c) 1985 Bikeshed Computer Systems Corp.')
stdscr.clrtoeol()
@ -231,7 +236,7 @@ while True:
stdscr.addstr(5, 1, str.format('Character pressed = {0}', c))
stdscr.clrtoeol()
stdscr.addstr(23, 1, '[B] Back')
stdscr.addstr(23, 1, '[B] Back', curses.A_REVERSE if highlight_keys else 0)
stdscr.clrtoeol()
stdscr.refresh()
elif current_screen == 'stats':
@ -257,7 +262,7 @@ while True:
else:
stdscr.addstr(5, 1, 'Loading...')
stdscr.addstr(23, 1, '[B] Back')
stdscr.addstr(23, 1, '[B] Back', curses.A_REVERSE if highlight_keys else 0)
stdscr.clrtoeol()
stdscr.refresh()
@ -291,7 +296,7 @@ while True:
else:
stdscr.addstr(5, 1, 'Loading...')
stdscr.addstr(23, 1, '[B] Back [J] Down [K] Up')
stdscr.addstr(23, 1, '[B] Back [J] Down [K] Up', curses.A_REVERSE if highlight_keys else 0)
stdscr.clrtoeol()
stdscr.refresh()
@ -307,7 +312,7 @@ while True:
for num, line in enumerate(lines[info_line:info_line+20]):
stdscr.addstr(num + offset, 1, line)
stdscr.addstr(23, 1, '[B] Back [J] Down [K] Up')
stdscr.addstr(23, 1, '[B] Back [J] Down [K] Up', curses.A_REVERSE if highlight_keys else 0)
stdscr.clrtoeol()
stdscr.refresh()
elif current_screen == 'protocoin':
@ -343,7 +348,7 @@ while True:
else:
stdscr.addstr(5, 1, 'Loading...')
stdscr.addstr(23, 1, '[B] Back [J] Down [K] Up')
stdscr.addstr(23, 1, '[B] Back [J] Down [K] Up', curses.A_REVERSE if highlight_keys else 0)
stdscr.clrtoeol()
stdscr.refresh()
@ -364,7 +369,7 @@ while True:
stdscr.addstr(23, 1, '[ENTER] Send [ESC] Cancel')
else:
stdscr.addstr(8, 4, '[E] Edit message')
stdscr.addstr(23, 1, '[B] Back')
stdscr.addstr(23, 1, '[B] Back', curses.A_REVERSE if highlight_keys else 0)
stdscr.clrtoeol()
stdscr.refresh()
@ -381,7 +386,7 @@ while True:
stdscr.addstr(23, 1, '[ENTER] Send [ESC] Cancel')
else:
stdscr.addstr(7, 4, '[E] Edit prompt')
stdscr.addstr(23, 1, '[B] Back')
stdscr.addstr(23, 1, '[B] Back', curses.A_REVERSE if highlight_keys else 0)
if think_result:
for _ in range(100):
@ -392,7 +397,7 @@ while True:
think_result = think_result[:-5]
stdscr.addstr(22, 1, '')
stdscr.clrtoeol()
stdscr.addstr(23, 1, '[B] Back')
stdscr.addstr(23, 1, '[B] Back', curses.A_REVERSE if highlight_keys else 0)
stdscr.clrtoeol()
if not think_result and not think_to_send:
@ -423,10 +428,16 @@ while True:
stdscr.addstr(18, 1, '----------------')
stdscr.addstr(20, 1, 'Thanks to Peter for lending the Morrow MTD-60 terminal and Jamie for the Pi.')
stdscr.addstr(23, 1, '[B] Back')
stdscr.addstr(23, 1, '[B] Back', curses.A_REVERSE if highlight_keys else 0)
stdscr.refresh()
stdscr.move(23, 79)
if highlight_keys:
time.sleep(0.5)
highlight_keys = False
continue
if skip_input:
skip_input = False
else:
@ -441,6 +452,12 @@ while True:
except:
button = None
def try_highlight():
global c, highlight_debounce, highlight_keys
if c and time.time() - highlight_debounce > 0.6:
highlight_debounce = time.time()
highlight_keys = True
if current_screen == 'home':
if button == 's':
current_screen = 'stats'
@ -458,18 +475,26 @@ while True:
current_screen = 'about'
elif button == 'p':
current_screen = 'protocoin'
else:
try_highlight()
elif current_screen == 'debug':
if button == 'b' or c == KEY_ESCAPE:
current_screen = 'home'
if button == 'x':
break
else:
try_highlight()
elif current_screen == 'stats':
if button == 'b' or c == KEY_ESCAPE:
current_screen = 'home'
stats = {}
else:
try_highlight()
elif current_screen == 'about':
if button == 'b' or c == KEY_ESCAPE:
current_screen = 'home'
else:
try_highlight()
elif current_screen == 'classes':
if button == 'b' or c == KEY_ESCAPE:
current_screen = 'home'
@ -482,6 +507,8 @@ while True:
if classes_start > 0:
classes_start -= 1
stdscr.erase()
else:
try_highlight()
elif current_screen == 'info':
if button == 'b' or c == KEY_ESCAPE:
current_screen = 'home'
@ -494,6 +521,8 @@ while True:
if info_line > 0:
info_line -= 19
stdscr.erase()
else:
try_highlight()
elif current_screen == 'protocoin':
if button == 'b' or c == KEY_ESCAPE:
current_screen = 'home'
@ -506,6 +535,8 @@ while True:
if protocoin_line > 0:
protocoin_line -= 1
stdscr.erase()
else:
try_highlight()
elif current_screen == 'sign':
if sign_to_send:
if c == curses.KEY_BACKSPACE:
@ -527,6 +558,8 @@ while True:
current_screen = 'home'
elif button == 'e':
sign_to_send = '_'
else:
try_highlight()
elif current_screen == 'think':
if think_to_send:
if c == curses.KEY_BACKSPACE:
@ -555,6 +588,8 @@ while True:
think_result = ''
elif button == 'e':
think_to_send = '_'
else:
try_highlight()
if current_screen != prev_screen:
prev_screen = current_screen