auto save

func.py

-total = []
-while True:
-    unit= input("请输入：")
-    if unit== 'q':
-        break
-    else:
-        total.append(unit)
-print(total)
\ No newline at end of file
+def new_input():
+    total = []
+    while True:
+        unit= input("请输入：")
+        if unit== 'q':
+            break
+        else:
+            try:
+                unit = int(unit)
+            except:
+                print("请输入整数：")
+            else:
+                total.append(unit)
+    print(total)
+    #返回的是价格列表
+    return total
+    
+a = new_input()
+for i in a:
+    print(i)
+
+for i in range(len(a)):
+    print(a[i])
\ No newline at end of file