save project

func.py

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