save project

func.py

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