101 lines
3.0 KiB
Python
Executable File
101 lines
3.0 KiB
Python
Executable File
#!/home/pi/protovac/env/bin/python
|
|
|
|
import os
|
|
import logging
|
|
import curses
|
|
import time
|
|
import tui
|
|
import utils
|
|
|
|
DEBUG = os.environ.get('DEBUG')
|
|
logging.basicConfig(
|
|
filename='protovax.log',
|
|
format='[%(asctime)s] %(levelname)s %(module)s/%(funcName)s - %(message)s',
|
|
level=logging.DEBUG if DEBUG else logging.INFO)
|
|
|
|
logging.info('')
|
|
logging.info('Boot up')
|
|
|
|
os.system('stty -ixon')
|
|
|
|
def main_loop(stdscr):
|
|
state = tui.AppState()
|
|
screens = {
|
|
'home': tui.HomeScreen(state, stdscr),
|
|
'debug': tui.DebugScreen(state, stdscr),
|
|
'stats': tui.StatsScreen(state, stdscr),
|
|
'classes': tui.ClassesScreen(state, stdscr),
|
|
'asimov': tui.TextScreen(state, stdscr, "The Last Question", utils.LAST_QUESTION),
|
|
'info': tui.TextScreen(state, stdscr, "Info", utils.PROTO_INFO),
|
|
'protocoin': tui.ProtocoinScreen(state, stdscr),
|
|
'sign': tui.SignScreen(state, stdscr),
|
|
'protovac_sign': tui.ProtovacSignScreen(state, stdscr),
|
|
'train': tui.TrainScreen(state, stdscr),
|
|
'nametag': tui.NametagScreen(state, stdscr),
|
|
'label': tui.LabelScreen(state, stdscr),
|
|
'games': tui.GamesScreen(state, stdscr),
|
|
'message': tui.MessageScreen(state, stdscr),
|
|
'think': tui.ThinkScreen(state, stdscr),
|
|
'about': tui.AboutScreen(state, stdscr),
|
|
'help': tui.HelpScreen(state, stdscr),
|
|
}
|
|
|
|
logging.info('Starting main loop...')
|
|
|
|
while True:
|
|
# Rate limit key presses unless in a text input field
|
|
if not screens[state.current_screen].is_typing and time.time() < state.last_key_time + 1:
|
|
time.sleep(0.05)
|
|
else:
|
|
c = stdscr.getch()
|
|
if c != curses.ERR:
|
|
state.c = c
|
|
state.last_key_time = time.time()
|
|
screens[state.current_screen].handle_input(c)
|
|
|
|
if state.current_screen != state.prev_screen:
|
|
logging.info('Switching to screen: %s', state.current_screen)
|
|
stdscr.erase()
|
|
|
|
if hasattr(screens[state.current_screen], 'on_enter'):
|
|
screens[state.current_screen].on_enter()
|
|
|
|
if state.current_screen != 'help':
|
|
state.highlight_count = 0
|
|
|
|
state.prev_screen = state.current_screen
|
|
|
|
screens[state.current_screen].draw()
|
|
stdscr.move(23, 79)
|
|
stdscr.refresh()
|
|
|
|
if state.highlight_keys:
|
|
time.sleep(0.5)
|
|
state.highlight_keys = False
|
|
|
|
time.sleep(0.01)
|
|
|
|
def main():
|
|
stdscr = curses.initscr()
|
|
curses.noecho()
|
|
curses.cbreak()
|
|
stdscr.keypad(True)
|
|
curses.curs_set(0)
|
|
stdscr.nodelay(True)
|
|
|
|
try:
|
|
main_loop(stdscr)
|
|
except KeyboardInterrupt:
|
|
pass
|
|
except Exception as e:
|
|
logging.exception("Caught exception in main loop")
|
|
finally:
|
|
curses.nocbreak()
|
|
stdscr.keypad(False)
|
|
curses.echo()
|
|
curses.endwin()
|
|
logging.info('Exiting.')
|
|
|
|
if __name__ == '__main__':
|
|
main()
|