save project

func.py

 def price(): 
 def price(): 
     total = []
     total = []
     while True:
     while True:
-        unit = input("请输入：")
+        unit = input("请输入：（输入“q”退出）")
         if unit == 'q':
         if unit == 'q':
             break
             break
         else:
         else:
+            try:
+                unit = int(unit)
+            except:
+                print('输入错误，请输入数字！')
+            else:
                 total.append(unit)
                 total.append(unit)
-    print('*'*30)
-    print(total)
-    print(' '*30)
+    return total
+
+def sum(alist):
+    count = 0
+    for i in alist:
+        count += i
+    return count
 
 
-price()
+p = price()
\ No newline at end of file
+print(p)
+s = sum(p)
+print(s)
\ No newline at end of file