save project

func.py

 def new_input():
     total = []
     while True:
-        unit= input("请输入：")
+        unit = input("请输入：")
+        try:
+            unit = int(unit)
+        except:
+            print("请输入数字，要整数！")
+        else:
+            total.append(unit)
         if unit== 'q':
             break
         else: