auto save

func.py

-total = []
-while True:
+def new_input():
+    total = []
+    while True:
         unit= input("请输入：")
         unit= input("请输入：")
         if unit== 'q':
         if unit== 'q':
             break
             break
         else:
         else:
+            try:
+                a = int(unit)
+            except:
+                print('重新输入数字')
+            else:
                 total.append(unit)
                 total.append(unit)
-print(total)
+    return total
\ No newline at end of file
+def sum(unit):
+    count = 0
+    for i in unit:
+        count = count + int(i)
+    return count
+price = new_input()
+pay = sum(price)
+print('您一共消费'+str(pay)+'元')
\ No newline at end of file