save project

func.py

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