save project

func.py

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