auto save

func.py

@@ -2,9 +2,10 @@ def new_input():
     total = []
     while True:
         try:
-            unit= int(input("请输入："))
+            k =input("请输入：")
+            unit= int(k)
         except:
-            if unit=='q':
+            if k == 'q':
                 break
             else:
                 print("请输入整数")