跟alex学python之购物车样例 列表，元组，函数使用

和alex学python之列表，元组，函数使用实例

#!/usr/bin/env python
#_*_ coding:utf-8 _*_
#__author__ = "Jimmy"

import sys

product_list = [
    ('Iphone',7800),
    ('Mac Pro',10000),
    ('Bike',1200),
    ('Watch',4500),
    ('Coffee',25),
    ('Alex Python',80),
    ('lenovo', 7500),
]

def shopping():
    shopping_list = []
    salary = input("Input your salary:")
    if salary.isdigit():
        salary = int(salary)
        while True:
            for index,item in enumerate(product_list):
                #print(product_list.index(item),item)
                print(index+1,item)
            user_choice = input("Choose items you want, or 'q' to quit>>>:")
            if user_choice.isdigit():
                user_choice = int(user_choice) - 1
                if user_choice < len(product_list) and user_choice >=0:
                    p_item = product_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 Your balance is [%s]，Cann't afford any items\033[0m" % salary)
                else:
                    print("product code [%s] is not exist!"% user_choice)
            elif user_choice == 'q':
                print("--------shopping list------")
                for p in shopping_list:
                    print(p)
                print("Your current balance:",salary)
                exit()
            else:
                print("invalid option")

if __name__ == "__main__":
    print("In main function")
    shopping()