save project

func.py

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