save project

func.py

@@ -2,10 +2,13 @@
 def new_input():
 def new_input():
     total = []
     total = []
     while True:
     while True:
-        try:
-            unit = int(input("请输入："))
+        
+        unit = input("请输入：")
         if unit== 'q':
         if unit== 'q':
             break
             break
+        else:
+            try:
+                unit=int(unit)
             except:
             except:
                 print('输入错误')
                 print('输入错误')
             else:
             else: