save project

func.py

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