save project

func.py

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