save project

func.py

@@ -5,6 +5,12 @@ def new_input():
         if unit== 'q':
             break
         else:
-            total.append(unit)
-    print(total)
-new_input()
\ No newline at end of file
+            try:
+                un = int(unit)
+            except:
+                print("请重新输入！")
+            else:    
+                total.append(unit)
+    return total
+result = new_input()
+print(result)
\ No newline at end of file