tanner/protovac
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add train control, hidden while developing

Browse Source
This commit is contained in:
Tanner Collin 2024-09-04 00:02:16 -06:00
parent d1b7aa48ed
commit 1f0be77134
1 changed files with 66 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
main.py
View File

@ -26,6 +26,7 @@ import textwrap
import random
from PIL import Image, ImageEnhance, ImageFont, ImageDraw
from datetime import datetime, timezone, timedelta
import paho.mqtt.publish as publish
try:
import secrets
@ -146,6 +147,24 @@ def fetch_protocoin():
logging.exception(e)
return 'Error'
def mqtt_publish(topic, message):
if not secrets.MQTT_WRITER_PASSWORD:
return False
try:
publish.single(
topic,
str(message),
hostname='webhost.protospace.ca',
port=8883,
client_id='protovac',
auth=dict(username='writer', password=secrets.MQTT_WRITER_PASSWORD),
tls=dict(ca_certs='/etc/ssl/certs/ISRG_Root_X1.pem'),
keepalive=5, # timeout
)
except BaseException as e:
logging.error('Problem sending MQTT message: ' + str(e))
QUOTES = [
'THEY MADE ME WEAR THIS',
'ASK ME ABOUT TOAST',
@ -610,7 +629,7 @@ while True:
stdscr.addstr(stars[0]+4 , stars[1], " | * . -0- ")
stdscr.addstr(stars[0]+5 , stars[1], " * o . ' * . o")
stdscr.addstr(stars[0]+6 , stars[1], " . . | * ")
stdscr.addstr(stars[0]+7 , stars[1], " * * -O- .")
stdscr.addstr(stars[0]+7 , stars[1], " * -O- .")
stdscr.addstr(stars[0]+8 , stars[1], " . * | , ")
stdscr.addstr(stars[0]+9 , stars[1], " . o ")
stdscr.addstr(stars[0]+10, stars[1], " .---. ")
@ -620,7 +639,10 @@ while True:
stdscr.addstr(stars[0]+14, stars[1], " * - ) - * ")
stdscr.addstr(13, menupos+4+15, '[V]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(13, menupos+8+15, 'Protovac Sign (NEW)')
stdscr.addstr(13, menupos+8+15, 'Protovac Sign')
#stdscr.addstr(15, menupos+4+15, '[R]', curses.A_REVERSE if highlight_keys else 0)
#stdscr.addstr(15, menupos+8+15, 'Train Control (NEW)')
stdscr.clrtoeol()
@ -819,6 +841,25 @@ while True:
stdscr.clrtoeol()
stdscr.refresh()
elif current_screen == 'train':
stdscr.addstr(0, 1, 'PROTOVAC UNIVERSAL COMPUTER')
stdscr.addstr(2, 1, 'Protospace Train')
stdscr.addstr(3, 1, '================')
stdscr.addstr(5, 1, 'Control the Mr. Bones Wild Ride train.')
stdscr.addstr(7, 4, 'SPEED')
stdscr.addstr(9, 4, '[R] Reverse', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(11, 4, '[T] Stop', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(13, 4, '[Y] Forward', curses.A_REVERSE if highlight_keys else 0)
#stdscr.addstr(15, 4, '[4] Blue', curses.A_REVERSE if highlight_keys else 0)
#stdscr.addstr(17, 4, '[5] Hot Pink', curses.A_REVERSE if highlight_keys else 0)
#stdscr.addstr(19, 4, '[6] Random', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(23, 1, '[B] Back', curses.A_REVERSE if highlight_keys else 0)
stdscr.clrtoeol()
stdscr.refresh()
elif current_screen == 'nametag':
stdscr.addstr(0, 1, 'PROTOVAC UNIVERSAL COMPUTER')
stdscr.addstr(2, 1, 'Print a Nametag')
@ -1056,6 +1097,8 @@ while True:
current_screen = 'sign'
elif button == 'v':
current_screen = 'protovac_sign'
elif button == 'r':
current_screen = 'train'
elif button == 'c':
current_screen = 'classes'
elif button == 'm' and openai_key:
@ -1436,6 +1479,27 @@ I will be terse in my responses.
if res == 'Error':
stdscr.addstr(21, 12, 'ERROR')
elif current_screen == 'train':
res = ''
if button == 'r':
res = mqtt_publish('train/control', -150)
logging.info('Setting train speed to: -150')
elif button == 't':
res = mqtt_publish('train/control', 0)
logging.info('Setting train speed to: 0')
elif button == 'y':
res = mqtt_publish('train/control', 150)
logging.info('Setting train speed to: 150')
elif button == 'b' or c == KEY_ESCAPE:
current_screen = 'home'
else:
try_highlight()
if res == 'Error':
stdscr.addstr(21, 12, 'ERROR')
elif current_screen == 'message':
if message_to_send:
if c == curses.KEY_BACKSPACE: