auto save

func.py

@@ -14,12 +14,23 @@
 def new_input():
     total = []
     while True:
-        unit= input("请输入：")
-        if unit== 'q':
-            break
+        try:
+            unit= input("请输入：")
+            if unit == 'q':
+                break
+            unit = int(unit)
+        except:
+            print("请输入整数")
         else:
             total.append(unit)
-    print(total)
-
+    return total
 
+def sum(money):
+    count = 0
+    for i in money:
+        count += i
+    return count
 
+total = new_input()
+num = sum((total))
+print(num)