auto save

func.py

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