save project

func.py

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