save project

func.py

@@ -2,8 +2,12 @@ def est_2():
     total = []
     while True:
         unit= input("请输入：")
-        if unit== 'q':
-            break
+        if unit=="q":
+            break    
+        try:
+            unit=int(unit)
+        except:
+            print("请输入整数")    
         else:
             total.append(unit)
     print(total)