save project

func.py

@@ -2,6 +2,9 @@ def jiage():
     total = []
     total = []
     while True:
     while True:
         unit= input("请输入：")
         unit= input("请输入：")
+        try:
+            unit=int(unit)
+        except:
             if unit== 'q':
             if unit== 'q':
                 break
                 break
         else:
         else: