第二周作业

购物车实现功能并函数封装，
购物车
1. 注册
2. 登陆
3. 购物
4. 支付
5. 充值
退出

用户表格记录，用户名:密码:余额 user.txt

msj:123456:10000
swb:123:911900
lyh:123:1000
scg:321:3000

记录购物车信息，用户名:商品名:价格:数量

msj:apple:10:1
msj:tesla:10:1
msj:apple:1000:2
scg:mac:3000:2
scg:chicken:10:10

函数代码：

def getname():
    '''
    得到用户输入的用户名，交由登录和注册使用
    :return: 返回整理后用户
    '''
    name = input('请输入用户名：').strip()
    return name

def getpwd():
    '''
    获得用户输入的密码，交由登录和注册使用
    :return: 整理过的密码
    '''
    pwd = input('请输入密码:').strip()
    return pwd

def is_db_get_u(name):
    """
    从数据文本下的获取用户名,判断键入用户名是否在user.txt
    :return: 如果在，返回True，如果不在返回False
    """
    with open(r'user.txt',mode='r',encoding='utf-8') as rf:
        for line in rf:
            u = line.strip('\n').split(':')[0]
            if name == u:
                return True
        else:return False

def is_pwd_corret(name,pwd):
    '''
    检测对应的用户名和密码是否正确
    :param name:
    :param pwd:
    :return: 正确的话返回True，否则返回False
    '''
    with open(r'user.txt',mode='r',encoding='utf-8')as rf:
        for line in rf:
            u,p,*_=line.strip('\n').split(':')
            if name ==u and pwd ==p:
                return True
        else:return False

def register():
    '''
    注册，检测用户名是否存在，
    存在，打印用户名存在，不存在，将用户名:密码:0
    :return:
    '''
    name=getname()
    pwd = getpwd()
    if is_db_get_u(name):
        print('用户已存在')
    else:
        msg = "%s:%s:0\n"%(name,pwd)
        with open(r'user.txt',mode='a',encoding='utf-8') as wf:
            wf.write(msg)
        print('注册完成')

def login():
    '''
    登录
    :return:登录成功返回name，登陆失败返回None
    '''
    name = getname()
    pwd = getpwd()
    if is_pwd_corret(name,pwd):
        print('登陆成功')
        return name
    else:
        print('用户名或密码错误，请重新输入')
        return None

def shopping(db_name):
    '''
    同展示内容，让用户输入自己想要的东西 和数量，并将数据添加到购物车shopping.txt当中
    :param db_name:用户名
    :return:None
    '''
    msg_dic = {
        'apple': 1000,
        'tesla': 100000,
        'mac': 3000,
        'lenovo': 30000,
        'chicken': 10,
    }
    while True:
        for key, price in msg_dic.items():
            print('商品名：%s  价格：%s' % (key, price))
        choice = input('请输入想要的商品名：').strip()
        if not choice or choice not in msg_dic:
            print('请重新输入商品名')
            continue
        while True:
            count = input('请输入购买的数量:').strip()
            if not count.isdigit():
                continue
            else:
                msg = '%s:%s:%s:%s\n' % (db_name, choice, msg_dic[choice], count)
                with open(r'shoppingcart.txt', mode='a', encoding='utf-8') as f:
                    f.write(msg)
                break
        break

def alter_money(db_name,money):
    """
    用于修改账户余额的功能，提供给充值和支付功能使用，根据用户找到相应的账户信息，并增加或减少相应的金额，并打印信息
    :param db_name: str 来自login 确定账户信息
    :param money: int 支付和充值信息，正负由支付功能和充值功能确定,还需要检测钱是不是整形
    :return:
    """
    with open(r'user.txt',mode='r',encoding='utf-8')as f,\
            open(r'user.txt.swap',mode='w',encoding='utf-8')as wf:
        for line in f:
            n,p,m=line.strip('\n').split(':')
            m=int(m)
            if db_name == n:
                line='%s:%s:%s\n'%(n,p,m+int(money))
            wf.write(line)
    os.remove('user.txt')
    os.rename('user.txt.swap','user.txt')

def total_price(db_name):
    '''
    用于计算购物车，某位用户购买货物的总价值，用于支付函数payment使用
    :param db_name: 用户名
    :return: 总价 int 类型
    '''
    tprice = 0
    with open(r'shoppingcart.txt',mode='r',encoding='utf-8') as f:
        for line in f:
            info = line.strip('\n').split(':')
            if  info:
               if db_name == info[0]:
                   tprice += int(info[2])*int(info[3])
    return tprice

def get_db_money(db_name):
    '''
    根据用户名获得账户的余额，交由支付payment使用
    :param db_name: 用户名
    :return: 余额   int类型
    '''
    with open(r'user.txt',mode='r',encoding='utf-8') as f:
        for line in f:
            u = line.strip('\n').split(':')[0]
            if db_name == u:
                return int(line.strip('\n').split(':')[2])

def clear_cart(db_name):
    '''
    用户支付后，清空shoppingcart的购物信息，交支付函数payment函数使用，会清空购车对应用户名的每行信息
    :param db_name: 用户
    :return:
    '''
    with open(r'shoppingcart.txt',mode='r',encoding='utf-8') as f,\
            open(r'shoppingcart.txt.swap',mode='w',encoding='utf-8') as wf:
        for line in f:
            u = line.strip('\n').split(':')[0]
            if db_name == u:
                line =''
            wf.write(line)
    os.remove('shoppingcart.txt')
    os.rename('shoppingcart.txt.swap','shoppingcart.txt')

def payment(db_name):
    """
    通过购物车文件内相同用户名的计算物品价格，在通过用户名检查user.txt下的第三个值为账户余额，
    判断余额是否大于等于购物车合计，如果大于等于提示支付，并修改账户内余额，否则提示余额不足请充值
    :param db_name:用户名，需要来自login
    :return:
    """
    if total_price(db_name)>get_db_money(db_name):
        print('账户余额不足，请充值')
    else:
        alter_money(db_name,-total_price(db_name))
        clear_cart(db_name)
        print('支付成功')

def recharge(db_name):
    '''
    充值功能，获得金钱数额，输错重新再输，正确就调用alter_money方法填写进数据表格中
    :param db_name:
    :return:
    '''
    while True:
        money = input('请输入充值的金额:').strip()
        if money.isdigit() and int(money)>0:
            alter_money(db_name,money)
            break
        print('请重新输入金额')

主代码

db_name = None
while True:
    print("""
    0. 退出
    1. 登录
    2. 注册
    3. 购物
    4. 支付
    5. 充值
    """)
    choice = input('请输入你想要的操作：').strip()
    if choice =='0':#退出
        print('退出成功')
        break
    elif choice =='1':#d登录
        db_name = login()
    elif choice =='2':#注册
        register()
    elif choice =='3':#购物
        if db_name:
            shopping(db_name)
        else:
            print('请登录')
    elif choice =='4':#支付
        if db_name:
            payment(db_name)
        else:
            print('请登录')
    elif choice=='5':#充值
        if db_name:
            recharge(db_name)
        else:
            print('请登录')
    else:
        print('请重新输入想要的操作')

上述函数还涉及使用了os模块，用于修改txt内存放的信息。