save project

diy.py

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