auto save

func.py

 def new_input():
     total = []
     while True:
-        unit = input('请输入（q退出）：')
-        if unit == 'q':
+        unit= input('请输入（q退出）：')
+        if unit== 'q':
             break
+        try:
+            unit=int(unit)
+        except:
+            print("输入一个数字")
         else:
-            try:
-                unit = int(unit)
-            except:
-                print("请重新输入一个数字")
-            else:
-                total.append(unit)
-            print("-"*30)
-    print(total)
+            total.append(unit)
+    return total
 
 
-new_input()      
-    
+print(new_input())
\ No newline at end of file