Python学习1，代码

  看了好久的网上视频，今天尝试着写了一串代码：

 _author_ = "Happyboy"
 produce_list = [
     ('Iphone',5800),
     ('Mac Pro',9800),
     ('Bike',800),
     ('Watch',10600),
     ('Coffee',31),
     ('Book',120)
     ]
 shopping_list = []
 salary = input("Input your salary:")
 if salary.isdigit():
      salary = int(salary)
      while True:
          for index,item in enumerate(produce_list):
             print(index,item)
          user_choice = input("选择要买吗？>>>:?")
          if user_choice.isdigit():
             user_choice = int(user_choice)
             if  user_choice < len(produce_list) and user_choice >=0:
                p_item = produce_list[user_choice]
                if p_item[1] <= salary : #买得起
                    shopping_list.append(p_item)
                    salary -= p_item[1]
                    print("Added %s into shopping cart,your current balance is \033[31;1m%s\033[0m"%(p_item,salary))
                else:
                    print("\033[41;1m你的余额只剩[%s]\033[0m"% salary)
             else:
                    print("prodct code [%s] is not exist"% user_choice)
          elif user_choice =="q":
               print("--------shooping list-------")
               for p in shopping_list:
                   print(p)
               print("Your current balance:",salary)
               exit()
          else:
               print("invalid option")

如果可以缩短，或者精炼的麻烦各位大牛帮我看下。。