save project

func.py

@@ -2,10 +2,14 @@ def new_input():
     total = []
     total = []
     while True:
     while True:
         unit= input("请输入：")
         unit= input("请输入：")
-        if unit == 'q':
-            break
+        try:
+            unit = int(unit)
+        except:  
+            print("请输入整数")
         else:
         else:
             total.append(unit)
             total.append(unit)
+        if unit == 'q':
+            break        
     print(total)
     print(total)
 
 
 new_input()
 new_input()
\ No newline at end of file