Pico-Oled-13 Key1 toggles display ON/OFF

Author: samveen

main.py

@@ -30,36 +30,50 @@
 # of lcd.clear() or lcd.fill().)
 loc=(lcd.width,lcd.height-4)
 
+screenawake=1
+
 while True:
     # Is a button pressed
     if lcd.is_pressed(lcd.KEY0):
         selected_idx = (selected_idx + 1) % len(codes)
 
-    # Did the selection change from last time
-    if code != codes[selected_idx]:
-        code = codes[selected_idx]
-        (passwd, expiry) = totp(synchronised_time(),
-                                  code['key'],
-                                  step_secs=code['step'],
-                                  digits=code['digits'])
-
-    # is it time to recalculate things yet?
-    s=time()%30
-    if s == 0:
-        (passwd, expiry) = totp(synchronised_time(),
-                                  code['key'],
-                                  step_secs=code['step'],
-                                  digits=code['digits'])
-
-    # Wipe previous text (first run wipes full screen except timer bar)
-    lcd.fill_rect(0,0,128,loc[1],0x0000)
-    loc=lcd.text(code['name'],0,0,0xffff)
-    loc=lcd.text(passwd,0,loc[1],0xffff)
-
-    # Time left bar
-    if s == 0:
-        lcd.fill_rect(4,60,120,4,0x0000)
-    lcd.fill_rect(4,60,(s+1)<<2,4,0xffff)
-    lcd.show()
-    sleep_ms(500)
+    if lcd.is_pressed(lcd.KEY1):
+        if screenawake:
+            screenawake=0
+        else:
+            # Empty the timer bar in the framebuffer
+            lcd.fill_rect(4,60,120,4,0x0000)
+            screenawake=1
+
+    if screenawake:
+        # Did the selection change from last time
+        if code != codes[selected_idx]:
+            code = codes[selected_idx]
+            (passwd, expiry) = totp(synchronised_time(),
+                                      code['key'],
+                                      step_secs=code['step'],
+                                      digits=code['digits'])
+
+        # is it time to recalculate things yet?
+        s=time()%30
+        if s == 0:
+            (passwd, expiry) = totp(synchronised_time(),
+                                      code['key'],
+                                      step_secs=code['step'],
+                                      digits=code['digits'])
+
+        # Wipe previous text (first run wipes full screen except timer bar)
+        lcd.fill_rect(0,0,128,loc[1],0x0000)
+        loc=lcd.text(code['name'],0,0,0xffff)
+        loc=lcd.text(passwd,0,loc[1],0xffff)
+
+        # Time left bar
+        if s == 0:
+            lcd.fill_rect(4,60,120,4,0x0000)
+        lcd.fill_rect(4,60,(s+1)<<2,4,0xffff)
+        lcd.on()
+        lcd.show()
+    else:
+        lcd.off()
 
+    sleep_ms(250)