save project

func.py

@@ -4,6 +4,10 @@ def input_unit():
         unit= input("请输入：")
         if unit== 'q':
             break
+        try:
+            unit=int(unit)
+        except:
+            print('请输入数字')
         else:
             total.append(unit)
     print(total)