auto save

func.py

@@ -5,6 +5,20 @@ def new_input():
         if unit== 'q':
         if unit== 'q':
             break
             break
         else:
         else:
+            try:
+                unit=int(unit)
+            except:
+                print("请重新输入一个数字")
+            else:
                 total.append(unit)
                 total.append(unit)
-    print(total)
-new_input()
+    return total
+
\ No newline at end of file
+def sum(money):
+    count=0
+    for i in money:
+        count=count+i
+    return count
+
+price=new_input()
+pay=sum(price)
+print("您一共消费了"+str(pay)+"元!扫码还是现金，或者信用卡？")
\ No newline at end of file

p.py

+import func
+data=func.new_input()
+score=func.sum(data)
+print(score)
\ No newline at end of file