python练习_module012购物车

购物车练习

一、需求：

1、启动程序后，输入用户名密码后，如果是第一次登录，让用户输入工资，然后打印商品列表

2、允许用户根据商品编号购买商品

3、用户选择商品后，检测余额是否够，够就直接扣款，不够就提醒

4、可随时退出，退出时，打印已购买商品和余额

5、在用户使用过程中， 关键输出，如余额，商品已加入购物车等消息，需高亮显示

6、用户下一次登录后，输入用户名密码，直接回到上次的状态，即上次消费的余额什么的还是那些，再次登录可继续购买

7、允许查询之前的消费记录

二、实现需求：

- 达成7项作业需求
- 自动去除用户输入前后的空格
- 简单修饰商品列表,购物车列表
- 用户输入 s 显示消费记录后，程序暂停。可输入任意键，按回车继续

 

三、代码

　　1. 生成包含 账号信息 的pickle文件

import  pickle

acc = {
    'Jailly':
        {'pwd':'123',
         'salary':None,
         'shopping cart':[]},
    'Lily':
        {'pwd':'abc',
         'salary':None,
        'shopping cart':[]}
}

with open('acc.pkl','wb') as f:
    pickle.dump(acc,f)

　　2. 主程序

#! /usr/bin/env python3
# -*- encoding:utf-8 -*-
# Author:Jailly

import pickle,getpass

# 用户登陆
def login():
    count = 0
    while count < 3:
        name_input = input('用户名：').strip()
        pwd_input = getpass.getpass('密码：')

        if name_input in acc:
            if acc[name_input]['pwd'] == pwd_input:
                return name_input
            else:
                print('\033[1;31m密码错误\033[0m')
                count += 1
        else:
            print('\033[1;31m您输入的用户名不存在，请重新输入\033[0m')
    else:
        print('错误次数太多，程序关闭...')
        exit()


# 欢迎信息 & 判断是否第一次登陆
def welcome(name_input):
    if acc[name_input]['salary']:
        salary = acc[name_input]['salary']
        print('欢迎，%s'%name_input)
        print('您的当前余额为\033[1;31m%d\033[0m,愿您购物愉快~' % salary)
    else:
        print('欢迎，%s' % name_input)
        salary = input('请输入您的工资：').strip()

        # 工资输入并判断输入是否正确
        while 1:
            if salary.isdigit() and int(salary) >= 0:
                salary = int(salary)

                break
            else:
                salary = input('工资必须为正整数，请重新输入：')
    return salary

# 消费记录
def show_recode(sc,salary):
    print('-' * 51)
    print('|', 'shopping cart'.center(47, ' '), '|')
    print('-' * 51)
    for i in sc:
        print('|', i[0].center(22, ' '), '|', str(i[1]).center(22, ' '), '|')
        print('-' * 51)
    print('|',('您的余额为%d' % salary).center(42,' '),'|')
    print('-' * 51)

# 选购商品
def shopping(name_input,salary):
    sc = acc[name_input]['shopping cart']

    while 1:
        print(' Product List '.center(50,'-'))
        for k,v in enumerate(product_list):
            print(('| %s. %s'%(k,v)).ljust(48,' '),'|')
        print('-'*50)

        choice = input('请输入您想要购买的商品编号：').strip()
        if choice.isdigit():
            choice = int(choice)
            if 0 < choice < len(product_list):
                p_item = product_list[choice]
                salary -= p_item[1]
                sc.append(p_item)
                print('您刚购买了\033[;31m%s\033[0m，当前账户余额为\033[1;31m%d\033[0m'%(str(p_item),salary))
            else:
                print('抱歉，没有找到您选择的商品')

        elif choice == 's':
            show_recode(sc,salary)
            input('输入回车键继续...')

        elif choice == 'q':
            show_recode(sc,salary)

            acc[name_input]['salary'] = salary
            acc[name_input]['shopping cart'] = sc
            with open('acc.pkl','wb') as f:
                pickle.dump(acc,f)

            print('感谢您的支持，再见~')
            exit()

        else:
            print('\033[1;31m请输入正确的商品编号\033[0m')

# 主程序
def main():
    name_input = login()
    salary = welcome(name_input)
    shopping(name_input,salary)

if __name__ == '__main__':
    product_list = [
        ('IPhone', 5000),
        ('Mac Pro', 12000),
        ('xbox360', 2500),
        ('PlayStation3', 2000),
        ('bike', 1000),
        ('kindle', 150)
    ]

    with open('acc.pkl', 'rb') as f:
        acc = pickle.load(f)

    main()