save project

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
+
+def sum(b):
+    s=0
+    for i in b:
+        s=s+i
+    return s
+
+        
\ No newline at end of file

swim.py

+import func
+a=func.new_input()
+s=func.sum(a)
+print('您一共消费'+str(s)+'元,现金支付还是扫码支付呢')
\ No newline at end of file