save project

diy.py

@@ -5,6 +5,12 @@ def store():
         if unit=='q':
             break
         else:
+            try:
+                unit=int(unit)
+            except:
+                print("请重输")
+            else:
                 total.append(unit)
+            print("-"*30)
         print(total)
-store()
\ No newline at end of file
+new_input()
\ No newline at end of file