diff --git a/main.py b/main.py new file mode 100644 index 0000000..abe989c --- /dev/null +++ b/main.py @@ -0,0 +1,170 @@ +from unicurses import * +from curses.textpad import Textbox, rectangle +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' + +highlight = 1 +choice = 0 +c = 0 + +# highlighting: +#wattron(menu_win, A_REVERSE) +#mvwaddstr(menu_win, y, x, choices[i]) +#wattroff(menu_win, A_REVERSE) + +stdscr = initscr() +clear() +noecho() +cbreak() +curs_set(0) + +keypad(stdscr, True) + +sign_to_send = '' +stats = {} + +while True: + if current_screen == 'home': + mvaddstr(1, 1, 'PROTOVAC - can you smell the phosphorus?') + mvaddstr(4, 4, '[T] Stats') + mvaddstr(6, 4, '[S] Sign') + #mvaddstr(8, 4, '[C] Classes') + #mvaddstr(10, 4, '[F] Forum') + + mvaddstr(23, 1, '') + clrtoeol() + refresh() + elif current_screen == 'debug': + mvaddstr(1, 1, 'PROTOVAC') + mvaddstr(3, 1, 'Debug Mode') + mvaddstr(4, 1, '==========') + mvaddstr(6, 1, str.format('Character pressed = {0}', c)) + clrtoeol() + + mvaddstr(23, 1, '[B] Back') + clrtoeol() + refresh() + elif current_screen == 'stats': + mvaddstr(1, 1, 'PROTOVAC') + mvaddstr(3, 1, 'Protospace Stats') + mvaddstr(4, 1, '================') + if stats: + mvaddstr(6 , 1, 'Next meeting: {}'.format(stats['next_meeting'])) + mvaddstr(8 , 1, 'Next clean: {}'.format(stats['next_clean'])) + mvaddstr(10, 1, 'Next class: {}'.format(stats['next_class']['datetime'])) + mvaddstr(11, 1, ' {}'.format(stats['next_class']['name'])) + mvaddstr(13, 1, 'Last class: {}'.format(stats['prev_class']['datetime'])) + mvaddstr(14, 1, ' {}'.format(stats['prev_class']['name'])) + + mvaddstr(16, 1, 'Member count: {} Green: {} Paused / expired: {}'.format( + stats['member_count'], + stats['green_count'], + stats['paused_count'], + )) + mvaddstr(18, 1, 'Card scans: {}'.format(stats['card_scans'])) + else: + mvaddstr(6, 1, 'Loading...') + + mvaddstr(23, 1, '[B] Back') + clrtoeol() + refresh() + + if not stats: + stats = fetch_stats() + skip_input = True + elif current_screen == 'sign': + mvaddstr(1, 1, 'PROTOVAC') + mvaddstr(3, 1, 'Protospace Sign') + mvaddstr(4, 1, '===============') + mvaddstr(6, 1, 'Send a message to the sign in the welcome room and classroom.') + + if sign_to_send: + mvaddstr(8, 4, sign_to_send) + mvaddstr(23, 1, '[ENTER] Send [ESC] Cancel') + else: + mvaddstr(8, 4, '[E] Edit message') + mvaddstr(23, 1, '[B] Back') + + clrtoeol() + refresh() + + if skip_input: + skip_input = False + else: + try: + c = wgetch(stdscr) + except KeyboardInterrupt: + refresh() + endwin() + break + + if c == KEY_UP: + pass + elif c == KEY_DOWN: + pass + elif c == 10: # ENTER is pressed + pass + + button = chr(c).lower() + + if current_screen == 'home': + if button == 't': + current_screen = 'stats' + clear() + elif button == 's': + current_screen = 'sign' + clear() + elif button == 'd': + current_screen = 'debug' + clear() + elif current_screen == 'stats': + if button == 'b': + current_screen = 'home' + stats = {} + clear() + 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: + mvaddstr(10, 4, 'Sending...') + refresh() + sign_send(sign_to_send[:-1]) + sign_to_send = '' + else: + if c < 127 and c > 31: + sign_to_send = sign_to_send[:-1] + chr(c) + '_' + clear() + elif button == 'b': + current_screen = 'home' + clear() + elif button == 'e': + sign_to_send = '_' + clear() + diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..185e8a9 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,7 @@ +certifi==2022.6.15 +charset-normalizer==2.1.1 +idna==3.3 +requests==2.28.1 +Uni-Curses==2.0.8 +urllib3==1.26.12 +x256==0.0.3