Departure board for Deptford

trains/main.py

@@ -20,17 +20,23 @@ import app
 import sleep
 from tilda import Buttons, LED
 
-
-# initialize screen
-ugfx.init()
-ugfx.clear()
-
-# initial screen
-ugfx.text(5, 5, "Will monitor station:", ugfx.BLACK)
-ugfx.text(200, 5, STATION_CODE, ugfx.BLUE)
-
+def init():
+    # initialize screen
+    ugfx.init()
+    ugfx.clear()
+    ugfx.orientation(90)
+    ugfx.backlight(50)
+    
+    # ensure wifi connection
+    if not wifi.is_connected():
+        wifi.connect(show_wait_message=True)
+    
+    # show initial screen
+    ugfx.text(5, 5, "Will monitor station:", ugfx.BLACK)
+    ugfx.text(200, 5, STATION_CODE, ugfx.BLUE)
+    
 def get_trains():
-    global station_data
+    station_data = None
 
     LED(LED.RED).on() # Red for total get_trains
     try:
@@ -38,17 +44,67 @@ def get_trains():
         LED(LED.GREEN).on() # Green for parsing
         station_data = ujson.loads(station_json)
     except:
-        print('oh poop')
+        print('Fuck')
 
     LED(LED.RED).off()
     LED(LED.GREEN).off()
+    return station_data
 
-get_trains()
-ugfx.text(5, 20, station_data['locationName'], ugfx.RED)
+def get_time(station_data):
+    return ':'.join(station_data['generatedAt'].split('T')[1].split(':')[0:2])
 
-# waiting until a button has been pressed
+def is_red(service):
+    return service['isCancelled'] or service['etd'] != 'On time'
+
+def get_arrival(service):
+    if service['isCancelled']:
+        return 'CANX'
+
+    if service['eta'] == 'On time':
+        return service['sta']
+
+    return service['eta']
+
+def get_title(name, has_error):
+    if has_error:
+        return 'ERR ' + name
+    
+    return name
+
+def show_trains(station_data, has_error):
+    ugfx.clear()
+    ugfx.area(0, 0, 240, 25,
+              ugfx.RED if has_error else ugfx.GRAY)
+    title = get_title(station_data['locationName'], has_error)
+    ugfx.text(5, 5, title,
+              ugfx.WHITE if has_error else ugfx.BLACK)
+    ugfx.text(195, 5, get_time(station_data), ugfx.BLUE)
+    names = ugfx.Container(0, 25, 190, 295)
+    names.show()
+    for idx, service in enumerate(station_data['trainServices']):
+        names.text(5, 15 * idx, service['destination'][0]['locationName'], ugfx.BLACK)
+        ugfx.text(195, 25 + (15 * idx), get_arrival(service), ugfx.RED if is_red(service) else ugfx.BLUE)
+
+def show_error():
+    ugfx.clear()
+    ugfx.text(5, 5, 'Error :(', ugfx.RED)
+
+init()
+station_data = None
+has_error = False
 while (not Buttons.is_pressed(Buttons.BTN_A)) and (not Buttons.is_pressed(Buttons.BTN_B)) and (not Buttons.is_pressed(Buttons.BTN_Menu)):
-    sleep.wfi()
+    new_station_data = get_trains()
+    if new_station_data == None:
+        has_error = True
+    else:
+        station_data = new_station_data
+        has_error = False
+    
+    if station_data == None:
+        show_error()
+    else:
+        show_trains(station_data, has_error)
+    sleep.sleep_ms(30 * 1000)
 
 
 # closing