save project

diy.py

@@ -4,6 +4,10 @@ def new_input()
         unit=input('请输入：')
         if unit=='q':
             break
+        try:
+            unit=int(unit)
+        except:
+            print("输入有误请输入数字")
         else:
             total.append(unit)
     print(total)