save project

func.py

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