save project

func.py

-
 total = []
 total = []
 def A():
 def A():
-while True:
+    while True:
         try:
         try:
             unit= int(input("请输入:"))
             unit= int(input("请输入:"))
         except:
         except:
@@ -9,7 +8,7 @@ while True:
         else:
         else:
             if unit == 1:
             if unit == 1:
                 break
                 break
-        else:
             total.append(unit)
             total.append(unit)
+return unit
 A()
 A()
 print(total)
 print(total)