save project

func.py

 total = []
 def new_input():
     while True:
-        unit= input("请输入：")
-        if unit== 'q':
-            break
+        try:
+            unit= int(input("请输入："))
+        except:
+            print("请输入整数")
         else:
-            total.append(unit)
-    print(total)
\ No newline at end of file
+            if unit== 'q':
+                break
+            else:
+                total.append(unit)
+        print(total)
\ No newline at end of file