protovac/main.py

#!/home/tanner/protovac/env/bin/python

import curses
import requests

ENTER_KEY = 10
BACKSPACE_KEY = 263
ESCAPE_KEY = 27


def sign_send(to_send):
    try:
        data = dict(sign=to_send, on_behalf_of='protovac')
        r = requests.post('https://api.my.protospace.ca/stats/sign/', data=data, timeout=5)
        r.raise_for_status()
        return 'Success!'
    except:
        return 'Error.'

def fetch_stats():
    try:
        r = requests.get('https://api.my.protospace.ca/stats/', timeout=5)
        r.raise_for_status()
        return r.json()
    except:
        return 'Error.'

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)

sign_to_send = ''
stats = {}

while True:
    if current_screen == 'home':
        stdscr.addstr(1, 1, ' _______  _______      ___    _________    ___   ____   ____  _        ______ ')
        stdscr.addstr(2, 1, '|_   __ \|_   __ \   .\'   `. |  _   _  | .\'   `.|_  _| |_  _|/ \     .\' ___  |')
        stdscr.addstr(3, 1, '  | |__) | | |__) | /  .-.  \|_/ | | \_|/  .-.  \ \ \   / / / _ \   / .\'   \_|')
        stdscr.addstr(4, 1, '  |  ___/  |  __ /  | |   | |    | |    | |   | |  \ \ / / / ___ \  | |       ')
        stdscr.addstr(5, 1, ' _| |_    _| |  \ \_\  `-\'  /   _| |_   \  `-\'  /   \ \' /_/ /   \ \_\ `.___.\'\\')
        stdscr.addstr(6, 1, '|_____|  |____| |___|`.___.\'   |_____|   `.___.\'     \_/|____| |____|`.____ .\'')

        stdscr.addstr(9, 4, '[T] Stats')
        stdscr.addstr(11, 4, '[S] Sign')
        #stdscr.addstr(8, 4, '[C] Classes')
        #stdscr.addstr(10, 4, '[F] Forum')

        stdscr.addstr(23, 1, '             Copyright (c) 1985 Bikeshed Computer Systems Corp.')
        stdscr.clrtoeol()
        stdscr.refresh()
    elif current_screen == 'debug':
        stdscr.addstr(1, 1, 'PROTOVAC')
        stdscr.addstr(3, 1, 'Debug Mode')
        stdscr.addstr(4, 1, '==========')
        stdscr.addstr(6, 1, str.format('Character pressed = {0}', c))
        stdscr.clrtoeol()

        stdscr.addstr(23, 1, '[B] Back')
        stdscr.clrtoeol()
        stdscr.refresh()
    elif current_screen == 'stats':
        stdscr.addstr(1, 1, 'PROTOVAC')
        stdscr.addstr(3, 1, 'Protospace Stats')
        stdscr.addstr(4, 1, '================')
        if stats:
            stdscr.addstr(6 , 1, 'Next meeting: {}'.format(stats['next_meeting']))
            stdscr.addstr(8 , 1, 'Next clean:   {}'.format(stats['next_clean']))
            stdscr.addstr(10, 1, 'Next class:   {}'.format(stats['next_class']['datetime']))
            stdscr.addstr(11, 1, '              {}'.format(stats['next_class']['name']))
            stdscr.addstr(13, 1, 'Last class:   {}'.format(stats['prev_class']['datetime']))
            stdscr.addstr(14, 1, '              {}'.format(stats['prev_class']['name']))

            stdscr.addstr(16, 1, 'Member count: {}   Green: {}   Paused / expired: {}'.format(
                stats['member_count'],
                stats['green_count'],
                stats['paused_count'],
            ))
            stdscr.addstr(18, 1, 'Card scans:   {}'.format(stats['card_scans']))
        else:
            stdscr.addstr(6, 1, 'Loading...')

        stdscr.addstr(23, 1, '[B] Back')
        stdscr.clrtoeol()
        stdscr.refresh()

        if not stats:
            stats = fetch_stats()
            skip_input = True
    elif current_screen == 'sign':
        stdscr.addstr(1, 1, 'PROTOVAC')
        stdscr.addstr(3, 1, 'Protospace Sign')
        stdscr.addstr(4, 1, '===============')
        stdscr.addstr(6, 1, 'Send a message to the sign in the welcome room and classroom.')
        stdscr.addstr(7, 1, 'After sending, turn your head right and wait.')

        if sign_to_send:
            stdscr.addstr(9, 4, sign_to_send)
            stdscr.clrtoeol()
            stdscr.addstr(23, 1, '[ENTER] Send  [ESC] Cancel')
        else:
            stdscr.addstr(9, 4, '[E] Edit message')
            stdscr.addstr(23, 1, '[B] Back')

        stdscr.clrtoeol()
        stdscr.refresh()

    stdscr.move(23, 79)

    if skip_input:
        skip_input = False
    else:
        try:
            c = stdscr.getch()
        except KeyboardInterrupt:
            break

    if c == curses.KEY_UP:
        pass
    elif c == curses.KEY_DOWN:
        pass
    elif c == 10:   # ENTER is pressed
        pass

    button = chr(c).lower()

    if current_screen == 'home':
        if button == 't':
            current_screen = 'stats'
        elif button == 's':
            current_screen = 'sign'
        elif button == 'd':
            current_screen = 'debug'
    elif current_screen == 'stats':
        if button == 'b':
            current_screen = 'home'
            stats = {}
    elif current_screen == 'sign':
        if sign_to_send:
            if c == BACKSPACE_KEY:
                sign_to_send = sign_to_send[:-2] + '_'
            elif c == ESCAPE_KEY:
                sign_to_send = ''
            elif c == ENTER_KEY:
                if len(sign_to_send) > 1:
                    stdscr.addstr(15, 4, 'Sending...')
                    stdscr.refresh()
                    sign_send(sign_to_send[:-1])
                    stdscr.erase()
                    sign_to_send = ''
            else:
                if c < 127 and c > 31:
                    sign_to_send = sign_to_send[:-1] + chr(c) + '_'
        elif button == 'b':
            current_screen = 'home'
        elif button == 'e':
            sign_to_send = '_'

    if current_screen != prev_screen:
        prev_screen = current_screen
        stdscr.erase()


curses.nocbreak()
stdscr.keypad(False)
curses.echo()
curses.endwin()