auto save

func.py

@@ -5,7 +5,12 @@ def new_input():    # 定义函数
         if unit== 'q':
             break
         else:
-            total.append(unit)
+            try:
+                unit = int(unit)  #转换成整数
+            except:
+                print("请输入一个整数！")
+            else:
+                total.append(unit)
     print(total)
 
 # 调用函数(运行函数)