save project

demo.py

+try:
+    age = int(input("How old are you?"))
+except:
+    print("You should print number")
+else:
+    if age < 18:
+        print("You cannot drink 酒")
+print("finish")
\ No newline at end of file

func.py

@@ -5,6 +5,12 @@ def new_input():
         if unit== 'q':
             break
         else:
-            total.append(unit)
+            try:
+                unit1 = int(unit)
+            except:
+                print("You should print number")
+            else:
+                total.append(unit1)
+            print("finish")
     print(total)
 new_input()
\ No newline at end of file