save project

func.py

-
 total = []
 def A():
-while True:
-    try:
-        unit= int(input("请输入:"))
-    except:
-        print('请输入数字')
-    else:
-        if unit == 1:
-            break
+    while True:
+        try:
+            unit= int(input("请输入:"))
+        except:
+            print('请输入数字')
         else:
+            if unit == 1:
+                break
             total.append(unit)
+return unit
 A()
-print(total)
\ No newline at end of file
+print(total)