Reset BT callback and send ko message to user when progarm downloading fails

lib/menu.py

@@ -190,7 +190,6 @@ def main():
                     program_name = cmd[4:]
                     if program_name in program_list:
                         program_num = program_list.index(program_name)
-                    print("BT user command: ", cmd, program_name)
                 elif cmd[:4] == "prg ":
                     program_name = cmd[4:]
                     if program_name.endswith(".py"):
@@ -208,8 +207,12 @@ def main():
                     robot.esp.bt_write("oc prg ok")
                     downloading = 2
                     return
-                with open(OC_TMP_FILE, "a") as f:
-                    f.write(next_line + "\n")
+                try:
+                    with open(OC_TMP_FILE, "a") as f:
+                        f.write(next_line + "\n")
+                except:
+                    downloading = 0
+                    robot.esp.bt_write("oc prg ko")
 
     # Initialize variables
     robot_state = 0
@@ -301,16 +304,19 @@ def main():
                         menu_debounce = 1
                 else:
                     if menu_move_v != 0:
-                        if current_menu >= Menu.COUNT:
+                        if current_menu >= Menu.COUNT: # == Menu.DOWNLOADING
                             pass
                         else:
                             menu_list[current_menu] += menu_move_v
                             menu_list[current_menu] %= menu_list_size[current_menu] 
                             menu_debounce = 1
                     elif menu_move_h != 0:
-                        if current_menu >= Menu.COUNT:
+                        if current_menu >= Menu.COUNT: # == Menu.DOWNLOADING
                             current_menu = Menu.PROGRAM
                             menu_list[current_menu] = 0
+                            downloading = 0
+                            robot.esp.bt_reset()
+                            robot.esp.bt_write("oc prg ko")
                         else:
                             menu_list[current_menu] = 0
                             current_menu += menu_move_h
@@ -446,7 +452,12 @@ def main():
 
         elif current_menu == Menu.DOWNLOAD:
             display_fill_downloading(robot, program_name)
-            if downloading == 2:
+            if downloading == 0:
+                robot.esp.bt_reset()
+                current_menu = Menu.PROGRAM
+                menu_list[current_menu] = 0
+                menu_debounce = 1
+            elif downloading == 2:
                 robot.esp.bt_reset()
                 downloading = 0
                 try: