save project

func.py

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