save project

func.py

@@ -4,6 +4,10 @@ def suanzhang():
         unit = input('请输入：')
         if unit=='q':
                 break
+        try:
+            unit = int(unit)
+        except:
+            print('要输入整数')
         else:
             total.append(unit)
     print(total)