save project

func.py

@@ -3,7 +3,11 @@ def suanzhang():
     while True:
         unit = input('请输入：')
         if unit=='q':
-            break
+                break
+        try:
+            unit = int(unit)
+        except:
+            print('要输入整数')
         else:
             total.append(unit)
     print(total)