save project

func.py

 total = []
 while True:
     unit= input("请输入：")
+    try:
+        unit = int(unit)
+    except Exception:
+        print("请输入整数！")
     if unit== 'q':
         break
     else: