save project

func.py

 def wfj():
 def wfj():
     total = []
     total = []
     while True:
     while True:
+        unit = input('请输入：')
+        if unit == 'q':
+            break
         try:
         try:
-            unit= int(input("请输入："))
+            a = int(unit)
         except:
         except:
             print("请重新输入")
             print("请重新输入")
-        if unit== 'q':
-            break
         else:
         else:
             total.append(unit)
             total.append(unit)
     print(total)
     print(total)