From b5a3c17b571d8d1b46d26ed3462e7fe80a12b6af Mon Sep 17 00:00:00 2001
From: dbroqua <dbroqua@mousur.org>
Date: Fri, 22 Jan 2021 18:39:15 +0100
Subject: [PATCH] Added exit function

---
 raspicar.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/raspicar.py b/raspicar.py
index bcb4c42..afdacbc 100644
--- a/raspicar.py
+++ b/raspicar.py
@@ -1,4 +1,5 @@
 import time
+import atexit
 import board
 import RPi.GPIO as GPIO
 
@@ -29,6 +30,14 @@ def handlePushButton(button):
     display.print("  :"+str(CHANNEL)+"-")
 
 
+# On exit
+def handleExit():
+    print("GoodBye!")
+    display.print("--:--")
+    sleep(2)
+    display.fill(0)
+
+
 # Init program
 def init():
     # Set display
@@ -114,12 +123,10 @@ def main():
         time.sleep(10)
 
 
+atexit.register(handleExit)
+
 if __name__ =='__main__':
     try:
         main()
     except KeyboardInterrupt:
         pass
-
-
-display.print("--:--")
-print("GoodBye!")