save project

func.py

 total = []
 def new_input():
     while True:
-        unit= input("请输入：")
-        if unit== 'q':
+        try:
+            unit = int(input("请输入："))       
+        except:
+            print("请重新输入")
             break
         else:
             total.append(unit)