auto save

func.py

-total = []
-while True:
-    unit= input("请输入：")
-    if unit== 'q':
-        break
-    else:
-        total.append(unit)
-print(total)
\ No newline at end of file
+def input_unit(): 
+    total = []
+    while True:
+        unit= input("请输入：")
+        if unit== 'q':
+            break
+        try:
+            unit = int(unit)
+        except:
+            print("请输入数字")
+        else:
+            total.append(unit)
+    return total
+
+def sum(list):
+    count = 0
+    for i in list:
+        count += i
+    return count
+
+

swim.py

+import func
+
+list = func.input_unit()
+score = fnuc.sum(list)
+print(score)
\ No newline at end of file