购物程序

今天写了个购物小程序，有的地方不够严谨，还需进一步完善，继续努力加油！


 1 #!/usr/bin/env python
#Author: Tony Cab

users = (['cab','123'],['tony','234'])
shop_mall = [
    ['Thinkpad',5999],
    ['Mac Proc',8000],
    ['Mi note 2',3000],
    ['Hua Wei',4000],
    ['Suo lan tuo',300000],
    ['Bicycle',3000],
    ['Clothes',500],
    ['Coffe',20],
    ['Chocolate',10]
]

shop_car = []
shop_price = []
total = 0

print('Who are you? Please login. Thanks！')
name = input('Account: ').strip()
passw = input('Password: ')

if [name,passw] in users:
    print('Welcome %s to cabell shopping mall' %name)
else:
    exit('Invalid account or password.')

salary = input('How much money do you have? ').strip()
if salary.isdigit(): sala = int(salary)

print('This is the list of goods. Please input the number of which do you want to buy.')
for item in enumerate(shop_mall):
    index = item[0]
    shop = item[1][0]
    price = item[1][1]
    print(index,':',shop,price)


flag = False
while not flag:
    pri = 'This list \033[1;31;40m %s \033[0m is your goods. they take you \033[1;31;40m %s \
           \033[0m dollars'
    num = input("Press 'q' to quit. Press number of good to buy. ").strip()
    if num == 'q' or num == 'Q':
        print(pri %(shop_car,total))
        flag = True
    else:
        numb = int(num)
        shop_car.append(shop_mall[numb])
        total += shop_mall[numb][1]
        if total > sala:
            shop_car.pop()
            total -= shop_mall[numb][1]
            print(pri %(shop_car,total))
            exit('Your money is not enough. Please recharge.'.center(60,'*'))
        else:
            print(shop_car)
            print(pri %(shop_car,total))

 经过修改优化后：

#!/usr/bin/env python
#Author: Tony Cab

import getpass

flag = False

users = [['cab','123'],['tony','234']]  #用户列表

new_acc = []
shop_car = []        #定义变量
shop_price = []
total = 0


while not flag:
    acc = input('Do you have a account? if you have please press \'y\'. else press \'a\' to register. and \
    press \'q\' to quit.')
    if acc == 'y':
        break
    elif acc == 'a':
        uname = input('New Account: ').strip()      #新用户注册
        upassw = input('New Password: ')
        upassd = input('Reput Password: ')
        if upassw == upassd:
            new_acc.append(uname)
            new_acc.append(upassw)
            users.append(new_acc)
            print('Congratulations! register success.Please login')
            break
        else:continue
    elif acc == 'q':
        exit('Good Bye'.center(40,'-'))
    else:
        print('Please do it according introductions')
        continue

print('Who are you? Please login. Thanks！')
name = input('Account: ').strip()               #用户登录
passw = input('Password: ')

if [name,passw] in users:
    print('Welcome %s to cabell shopping mall' %name)
else:
    exit('Invalid account or password.')

salary = input('How much dollars do you have? ').strip()            #输入可购买金额
if salary.isdigit(): sala = int(salary)
else:
    exit('Error! Please input number.')

shop_mall = [
    ['Thinkpad',5999],
    ['Mac Proc',8000],                #购物商场商品列表
    ['Mi note 2',3000],
    ['Hua Wei',4000],
    ['Suo lan tuo',300000],
    ['Bicycle',3000],
    ['Clothes',500],
    ['Coffe',20],
    ['Chocolate',10]
]

print('This is the list of goods. Please input the number of which do you want to buy.')
for item in enumerate(shop_mall):
    index = item[0]
    shop = item[1][0]                      #待购商品列表
    price = item[1][1]
    print(index,':',shop,price)



while not flag:
    pri = 'This list \033[1;31;40m %s \033[0m is your goods. they take you \033[1;31;40m %s \
           \033[0m dollars and left %s dollars'
    num = input("Press 'q' to quit. Press \'s\' to reset which to remove from your shopping cart \
     Press number of good to buy. Press \'l\' to list your goods. Press 'e' to exit with out buy \
     anything").strip()
    if num == 'q' or num == 'Q':
        print(pri %(shop_car,total,sala-total))
        print('We hope you have a plesent everyday. See you later.')
        flag = True

    elif num.isdigit() and int(num) < len(shop_mall):
        numb = int(num)
        shop_car.append(shop_mall[numb])
        total += shop_mall[numb][1]                     #选择商品并购买
        if total > sala:
            shop_car.pop()
            total -= shop_mall[numb][1]
            print(pri %(shop_car,total,sala-total))
            print('Your money is not enough. Please recharge.'.center(60,'*'))
        else:
            print(shop_car)
            print(pri %(shop_car,total,sala-total))
    elif num == 's':
        for items in enumerate(shop_car):
            inde = items[0]
            goo = items[1][0]                         #已购商品
            pric = items[1][1]
            print(inde,':',goo,pric)
        renum = input('Press the number of good to remove and press q to return shop_mall list')
        if renum.isdigit() and int(renum) < len(shop_car):
            renumb = int(renum)
            total -= shop_car[renumb][1]                  #已购商品退货
            shop_car.pop(renumb)
            print(pri %(shop_car,total,sala-total))
        elif renum.isdigit() and int(renum) >= len(shop_car):
            print('The number out of shopping number')
            continue
        else:
            continue
    elif num == 'l':
        for list in enumerate(shop_car):
            good_nam = list[1][0]
            good_pri = list[1][1]                     #已购商品列表
            print(good_nam,':',good_pri)
        print('Total: %s' %total)
    elif num.isdigit() and int(num) >= len(shop_mall):
        print('The number out of shopping number')
        continue
    elif num == 'e':
        total = 0
        shop_car.clear()
        print(pri %(shop_car,total,sala))
    else:
        print('Please input according to the Useage')
        continue