save project

func.py

 def menu():
 def menu():
     total = []
     total = []
     while True:
     while True:
-        unit= input("请输入：")
+        try:
+            unit= int(input("请输入："))
+        except:
+            print("你输入有误,输入数字")
+        else:
             if unit== 'p':
             if unit== 'p':
                 break
                 break
             else:
             else: