auto save

func.py

@@ -5,6 +5,12 @@ def a():
         if unit== 'q':
             break
         else:
-            total.append(unit)
-    print(total)
-a()
\ No newline at end of file
+            try:
+                unit=int(unit)
+            except:
+                print('请输入一个数字')
+            else:
+                total.append(unit)
+    return(total)
+d=a()
+print(d)
\ No newline at end of file