auto save

func.py

 def r():    
     total = []
     while True:
-        try:
-            unit= int(input("请输入："))
-        except:
-            print('请输入整数')
-        else:    
-            if unit=='q':
-                break
-            else:
+        unit= (input("请输入："))
+        if unit=="q":
+            break
+        else:
+            try:    
+                unit1=int(unit)
+            except:
+                print('请输入整数')
+            else:    
                 total.append(unit) 
-    print(total)  
-r()
\ No newline at end of file
+    return unit1  
+print()
\ No newline at end of file