save project

func.py

@@ -11,6 +11,10 @@ while True:
     unit = input("请输入:")
     if unit == "q":
         break
+    try:
+        unit = int(unit)
+    except:
+        print("黑小伙子，请输入数字呀！")
     else:
         total.append(unit)
-print(total)
\ No newline at end of file
+        
\ No newline at end of file