auto save

func.py

 def a():
 def a():
     total = []
     total = []
     while True:
     while True:
-        unit= input("请输入：")
+        unit= int(input("请输入："))
         try:
         try:
             if unit== 'q':
             if unit== 'q':
                 break
                 break