fix mouse or enter

pandora-box.py

@@ -242,19 +242,6 @@ def display_image(status):
             os.system(f"fim -qa {image} </dev/null 2>/dev/null >/dev/null &")
 
 
-# -----------------------------------------------------------
-
-def wait_mouse_click():
-    """ Wait for mouse click event """
-    with open("/dev/input/mice", "rb") as mouse:
-        down = False
-        while True:
-            buf = mouse.read(3)
-            if (buf[0] & 0x1) == 1:
-                down = True
-            if ((buf[0] & 0x1) == 0) and down:
-                break
-
 # -----------------------------------------------------------
 # has_curses Screen
 # -----------------------------------------------------------
@@ -695,6 +682,59 @@ def error():
         display_image("ERROR")
     return "WAIT"
 
+# -----------------------------------------------------------
+# Wait for mouse click or enter
+# -----------------------------------------------------------
+
+
+mouseEvent = threading.Event()
+enterEvent = threading.Event()
+mouseOrEnterCondition = threading.Condition()
+
+
+def mouseClickThread():
+    mouse = open("/dev/input/mice", "rb")
+    os.set_blocking(mouse.fileno(), False)
+
+    down = False
+    while not enterEvent.is_set():
+        buf = mouse.read(3)
+        if not (buf is None):
+            if ((buf[0] & 0x1) == 1):
+                down = True
+            if (((buf[0] & 0x1) == 0) and down):
+                break
+        time.sleep(0.1)
+    mouse.close()
+
+    mouseEvent.set()
+    with mouseOrEnterCondition:
+        mouseOrEnterCondition.notify()
+
+
+def enterKeyThread():
+    os.set_blocking(sys.stdin.fileno(), False)
+
+    while not mouseEvent.is_set():
+        input = sys.stdin.readline()
+        if (len(input) > 0):
+            break
+        time.sleep(0.1)
+
+    enterEvent.set()
+    with mouseOrEnterCondition:
+        mouseOrEnterCondition.notify()
+
+
+def waitMouseOrEnter():
+    with mouseOrEnterCondition:
+        threading.Thread(target=mouseClickThread, args=()).start()
+        threading.Thread(target=enterKeyThread, args=()).start()
+
+        mouseEvent.clear()
+        enterEvent.clear()
+        while not (mouseEvent.is_set() or enterEvent.is_set()):
+            mouseOrEnterCondition.wait()
 
 # --------------------------------------
 
@@ -709,7 +749,6 @@ def clean():
 
         if not has_curses:
             display_image("BAD")
-            wait_mouse_click()
         else:
             # print list of files
             cnt = 0
@@ -721,7 +760,8 @@ def clean():
                     break
             # wait for clean
             log('PRESS KEY TO CLEAN', flush=True)
-            wait_mouse_click()
+
+        waitMouseOrEnter()
 
         # TODO: check device is still present
 
@@ -793,8 +833,7 @@ def startup():
         logo = file1.readlines()
     # Print logo screen
     print_screen()
-    # First unmount remaining device
-    umount_device()
+
     return "WAIT"