第三周作业

作业需求：

模拟实现一个ATM + 购物商城程序

1. 额度 15000或自定义
2. 实现购物商城，买东西加入 购物车，调用信用卡接口结账
3. 可以提现，手续费5%
4. 每月22号出账单，每月10号为还款日，过期未还，按欠款总额 万分之5 每日计息
5. 支持多账户登录
6. 支持账户间转账
7. 记录每月日常消费流水
8. 提供还款接口
9. ATM记录操作日志 
10. 提供管理接口，包括添加账户、用户额度，冻结账户等。。。
11. 用户认证用装饰器

源代码：

其他文件格式：

#!/usr/bin/env python  
#-*- coding:utf-8 -*-
import time
import os
import re
ab_info ={'shop_user':None,'bank_user':'msj'}

def t_stamp():
    """返回一个时间戳"""
    return time.strftime('%Y-%m-%d %X')

def add_user(name,pwd,type):
    """
    添加账户
    :param name:
    :param pwd:
    :param type: 账户种类 两种  account  和  bank
    :return:
    """
    if type =='account':
        file = 'account_info.txt'
        user_info ='%s|%s
'%(name,pwd)
    elif type =='bank':
        file = 'bank_info.txt'
        user_info ='%s|%s|0|15000
'%(name,pwd)
    else:
        print('非法操作2')
        return
    with open(r'%s'%file,mode='a',encoding='utf-8') as f:
        f.write(user_info)

def acc_info():
    """
    :return: 返回account_info.txt中以字典形式{
    'msj': '123', 'scg': '123', 'lhy': '123', 'egon': '1234', 'flower': '123', 'fgh': '123', 'hua': '123'}
    """
    account_info = {}
    with open(r'account_info.txt',mode='r',encoding='utf-8') as f:
        for line in f:
            line = line.strip('
')
            if line != '':
                u,v = line.split('|')
                account_info[u] = v
    return account_info

def bk_info():
    '''
    钱和信用额度都是浮点数类型
    :return: {'msj': {'pwd': '123', 'money': 4600.0, 'credit': 15000.0}}
    '''
    bank_user ={}
    with open(r'bank_info.txt',mode='r',encoding='utf-8') as f:
        for line in f:
            line = line.strip('
')
            if line != '':
                u,p,m,c = line.split('|')
                bank_user[u]={'pwd':p,'money':float(m),'credit':float(c)}
        return bank_user

def alter_money(name,money):
    '''
    改变账号中的余额
    :param name: 用户名
    :param money: int 钱
    :return:
    '''
    user_info = bk_info()
    user_info[name]['money']=user_info[name]['money']+money
    # print(user_info[name]['money'])
    with open(r'bank_info.txt',mode='w',encoding='utf-8') as f:
        for k,v in user_info.items():
            msg = '{}|{}|{}|{}
'.format(k,v['pwd'],v['money'],v['credit'])
            f.write(msg)

def alter_credit(name,money):
    '''
    改变账号中的信用额
    :param name: 用户名
    :param money: int 钱
    :return:
    '''
    user_info = bk_info()
    user_info[name]['credit']=user_info[name]['credit']+money
    # print(user_info[name]['money'])
    with open(r'bank_info.txt',mode='w',encoding='utf-8') as f:
        for k,v in user_info.items():
            msg = '{}|{}|{}|{}
'.format(k,v['pwd'],v['money'],v['credit'])
            f.write(msg)

def auth(kind):
    def auth1(func):
        def wrapper(*args,**kwargs):
            if kind == 'shop':
                account =ab_info['shop_user']
            elif kind =='bank':
                account =ab_info['bank_user']
            if account is None:
                print('please log in first')
            else:
                res = func(*args, **kwargs)
                return res
        return wrapper
    return auth1

def is_number(num):
    '''
    判断字符串是否为数字，包含整形和浮点型。
    :param num:
    :return:
    '''
    pattern = re.compile(r'^[-+]?[-0-9]d*.d*|[-+]?.?[0-9]d*$')
    result = pattern.match(num)
    if result:
        return True
    else:
        return False

@auth('bank')
def check_balance():
    '''查看余额'''
    username = ab_info['bank_user']
    user_info= bk_info()
    res = user_info[username]['money']
    print('当前账户余额：%s'%res)
    return res

@auth('bank')
def check_credit():
    '''查看信用'''
    username = ab_info['bank_user']
    user_info= bk_info()
    res = user_info[username]['credit']
    print('当前账户信用额度：%s'%res)
    return res

def log(type,msg):
    """
    type 选项account用户日志文件、选项bank 银行日志文件、shopping 购物车文件、buy 写入shopping.log.txt
    :param type: 选择写入的文件
    :param msg: 写入的内容
    :return:
    """
    if type == 'account':
        file = 'info_account.log.txt'
    elif type == 'bank':
        file = 'info_bank.log.txt'
    elif type == 'buy':
        file = 'shopping.log.txt'
    elif type == 'shopping':
        file = 'shoppingcart.txt'
    else:
        print('不符合要求操作')
        return
    with open(r'%s'%file,mode='a',encoding='utf-8') as f:
        f.write(msg)

def read_hmd(type):
    if type == 'shop':
        file = 'shop_hmd.txt'
    else:
        file = 'bank_hmd.txt'
    hmd = []
    with open('%s'%file,mode='r',encoding='utf-8') as f:
        for i in f:
            i = i.strip('
')
            if i != '':
                hmd.append(i)
    return hmd

@auth('bank')
def recharge():
    '''余额充值'''
    current_user= ab_info['bank_user']
    check_balance()
    while True:
        choice = input('是否充值 Y/N>>:').strip()
        if choice == 'Y' or choice =='y':
            break
        elif choice == 'N' or choice =='n':
            print('exit')
            return
        else:
            print('非法操作')
            continue
    while True:
        money = input('请输入充值金额>>:').strip()
        if is_number(money):
            money = float(money)
            if money<=0:
                print('充值金额需要大于0')
                continue
            else:
                t =t_stamp()
                msg = '[INFO] [{}] [bank] {} 充值了 {} 元
'.format(t,current_user,money)
                alter_money(current_user,money)
                log('bank',msg)
                print('recharge success')
                return
        else:
            print('非法输入')
            continue

def regist():
    '''
    注册功能
    :return:
    '''
    choice = input('请问你想要注册的账户：1、商城，2、银行。>>').strip()
    if choice == '1':
        users = acc_info()
    elif choice == '2':
        users = bk_info()
    else:
        print('非法操作1')
        return
    tag = True
    while tag:
        username = input('please input username>>').strip()
        if username in users:
            print('user is exist')
            continue
        break
    while tag:
        pwd1 = input('please input password>>').strip()
        pwd2 = input('please config password>>').strip()
        if pwd1 == pwd2:
            if choice == '1':
                add_user(username,pwd1,type='account')#添加account添加用户
                t = t_stamp()
                msg = '[INFO] [{}] [user] {} 注册了
'.format(t,username)#添加日志
                log('account',msg)
                print(msg)
                return
            else:
                add_user(username, pwd1, type='bank')#添加bank添加用户
                t = t_stamp()
                msg = '[INFO] [{}] [bank] {} 注册了
'.format(t, username)#添加日志
                log('bank', msg)
                print(msg)
                return

        else:
            print('password is difference')

def login():
    '''登录'''
    choice = input('请问你想要登录的账户：1、商城，2、银行。>>').strip()
    if choice == '1':
        user_inf = acc_info()
        t = 'shop_user'
        type = 'shop'
    elif choice == '2':
        user_inf = bk_info()
        t='bank_user'
        type = 'bank'
    else:
        print('非法操作')
        return
    if ab_info[t] is  None:
        name = input('please input username>>:').strip()
        pwd = input('please input password>>:').strip()
        hmd = read_hmd(type)
        if name in hmd:
            print('该账户在黑名单')
            return
        if name in user_inf :
            if choice == '1' and pwd == user_inf[name]:
                    print('login success')
                    ab_info[t] = name
            elif pwd == user_inf[name]['pwd']:
                    print('login success')
                    ab_info[t] = name
            else:
                print('user or password is error')
        else:
            print('user or password is error')
    else:
        print('Please logoff and log in.')

@auth('bank')
def transfer():
    '''转账'''
    user_info = bk_info()
    while True:
        name = input('please input transfer user(q to exit)>>:').strip()
        if name == ab_info['bank_user']:
            print('cannot transfer to yourself')
            continue
        elif name == 'q':
            print('exit')
            return
        elif name in bk_info():
            money = input('please input transfer money>>: ').strip()
            if is_number(money):
                money = float(money)
            else:
                print('输入金额不正确')
                return
            if money > user_info[ab_info['bank_user']]['money']:
                print('余额不足，无法转账')
                return
            t= t_stamp()
            msg = '[INFO] [{}][bank] {} 向 {} 转账 {}
'.format(t,ab_info['bank_user'],name,money)
            alter_money(ab_info['bank_user'],-money)
            alter_money(name,money)
            log('bank',msg)
            print(msg)
            return
        else:
            print('user do not exist')

@auth('bank')
def repay():
    '''还款'''
    user_info= bk_info()
    current_user=ab_info['bank_user']
    credit_money = user_info[current_user]['credit']
    if credit_money>=15000:
        print('没有使用信用额度')
        return
    print('需要还款额度：%s'%(15000-credit_money))
    while True:
        repay_money = input('please input repay money(q to exit)>>:').strip()
        if repay_money == 'q':
            print('exit')
            return
        elif is_number(repay_money):
            repay_money = float(repay_money)
            if repay_money<=0:
                print('please input number')
                continue
            elif repay_money>(15000-credit_money):
                print('超过最大还款额度，请重新输入金额')
                continue
            else:
                t= t_stamp()
                msg = '[INFO][{}] [bank] {} 还款了 {} 元
'.format(t,current_user,repay_money)
                alter_credit(current_user,repay_money)
                log('bank',msg)
                return
        else:
            print('非法操作')
            continue

@auth('bank')
def withdraw():
    '''取款'''
    user_info = bk_info()
    current_user = ab_info['bank_user']
    while True:
        draw_money = input('please input withdraw(q to exit)>>:').strip()
        if draw_money == 'q':
            print('exit')
            return
        elif is_number(draw_money):
            draw_money = float(draw_money)
            if draw_money > user_info[current_user]['money']:
                print('money not enough')
                continue
            else:
                print('withdraw %s yuan success' % draw_money)
                t = t_stamp()
                draw_money = draw_money * 1.05
                msg = '[INFO] [{}] [bank] {} 取款 {} 元
'.format(t, current_user, draw_money)
                alter_money(current_user, -draw_money)
                log('bank', msg)
                print(msg)

        else:
            print('非法操作')

@auth('bank')
def turnover():
    '''查看流水'''
    current_user = ab_info['bank_user']
    with open(r'info_bank.log.txt',mode='r',encoding='utf-8') as f:
        for line in f:
            line=line.strip('
')
            if line != '':
                l=line.rsplit(']',1)
                if current_user in l[1]:
                    print(l[1])


def add_shoppingcart(name,good,price,count):
    '''添加购物车'''
    log_msg = "%s|{'%s':{'price':%s,'count':%s}}
"%(name,good,price,count)
    log('shopping',log_msg)

@auth('shop')
def check_shoppingcart():
    '''
    检查当前用户购物车中的商品，并将商品信息返回
    :return: 返回，为字典内包含列表形式
    '''
    current_user = ab_info['shop_user']
    shopping_list = []
    with open(r'shoppingcart.txt',mode='r',encoding='utf-8') as f:
        for line in f:
            line = line.strip('
')
            if line != '':
                l = line.split('|')
                if l[0] == current_user:
                    shopping_list.append(eval(l[1]))
    return shopping_list

def sum_money(shopping_list):
    '''根据check_shoppingcart 返回的列表计算需要支付的钱款的总和'''
    sum_m = 0
    for item in shopping_list:
       for i in item.values():
           sum_m = sum_m + i['price']*i['count']
    return sum_m

def clear_cart(name):
    '''
    清空购物车
    :param name:
    :return:
    '''
    with open(r'shoppingcart.txt',mode='r',encoding='utf-8') as rf,
        open(r'shoppingcart.txt.swap',mode='w',encoding='utf-8') as wf:
        for line in rf:
            line = line.strip('
')
            if line != '':
                if name in line:
                    line = ''
                else:
                    line = line +'
'
                wf.write(line)
    os.remove('shoppingcart.txt')
    os.rename('shoppingcart.txt.swap','shoppingcart.txt')

@auth('shop')
@auth('bank')
def buy():
    shop_user= ab_info['shop_user']
    bank_user = ab_info['bank_user']
    shopping_list = check_shoppingcart()
    if shopping_list is []:
        print('购物车中没有商品')
        return
    print('%s 购物车中 %s'%(shop_user,shopping_list))
    money = sum_money(shopping_list)
    print('需支付 %s'%money)
    while True:
        choice = input('是否支付 Y/N>>:')
        if choice == 'Y' or choice =='y':
            break
        elif choice == 'N' or choice =='n':
            print('退出支付')
            return
        else:
            print('非法操作')
            continue
    while True:
        pay_wey = input('请选择支付方式：1.余额  2.信用额度(q to exit)>>:')
        if pay_wey == '1':
            paymoney = check_balance()
        elif pay_wey =='2':
            paymoney = check_credit()
        elif pay_wey == 'q':
            print('exit')
            return
        else:
            print('非法操作')
            continue
        if paymoney <money:
            print('账户中金额无法支付，请重新选择支付方式。')
            continue
        else:
            if pay_wey == '1':
                alter_money(bank_user,-money)
            else:
                alter_credit(bank_user,-money)
            t = t_stamp()
            bank_info = '[INFO] [{}] [bank] {} 消费 {} 元
'.format(t,bank_user,money)
            log('bank',bank_info)
            shop_info = '[INFO] [{}] [shop] {} 花费 {} 元购买了{}
'.format(t,shop_user,money,shopping_list)
            log('buy',shop_info)
            clear_cart(shop_user)
            print('buy success')
            return

@auth('shop')
def shoppingcart():
    current_user=ab_info['shop_user']
    good_msg = [['coffee',30],['chicken',20],['iphone',8000],['macbook',12000],['car',100000]]
    print('购物')
    while True:
        i = 0
        for i in range(len(good_msg)):
            print(i,good_msg[i])
        choice = input('please choice goods or exit(q)>>:').strip()
        if choice == 'q':
            print('exit')
            # 显示购物车信息，提示是否购买
            shopping_list = check_shoppingcart()
            buy()
            return
        elif choice.isdigit():
            choice = int(choice)
        else:
            print('非法操作')
            continue
        if choice < len(good_msg):
            while True:
                count = input('please input goods number>>:').strip()
                if count.isdigit():
                    count = int(count)
                    if count <= 0:
                        print('数量必须大于0')
                    else:
                        show_msg = '{} {} add to shopping cart'.format(count,good_msg[choice][0])
                        print(show_msg)
                        add_shoppingcart(current_user,good_msg[choice][0],good_msg[choice][1],count)
                        break

                else:
                    print('非法操作')
                    continue

        else:
            print('没有对应的商品')
            continue

@auth('shop')
def check_shoppinglog():
    current_user = ab_info['shop_user']
    with open(r'shopping.log.txt',mode = 'r',encoding='utf-8') as f:
        for line in f:
            line= line.strip('
')
            if line != '':
                if current_user in line:
                    l=line.split(']',3)
                    print(l[3])

def log_off():
    if ab_info['shop_user'] and ab_info['bank_user']:
        while True:
            choice = input('log off 1.shop 2.bank').strip('
')
            if choice =='1':
                ab_info['shop_user'] = None
            elif choice =='2':
                ab_info['bank_user'] = None
            else:
                print('非法操作')
                continue
            return
    elif ab_info['shop_user']:
        ab_info['shop_user'] = None
    elif ab_info['bank_user']:
        ab_info['bank_user'] = None
    else:
        print('no account is login')

init_msg = """
    1.登录
    2、注册
    3、查看余额
    4、购物
    5、充值
    6、转账
    7、还款
    8、取款
    9、查看流水
    10、查看购物车
    11、查看信用额度
    12、查看购买信息
    13、支付
    14、注销
    0、退出
"""

init_dic={
    '1':login,
    '2':regist,
    '3':check_balance,
    '4':shoppingcart,
    '5':recharge,
    '6':transfer,
    '7':repay,
    '8':withdraw,
    '9':turnover,
    '10':check_shoppingcart,
    '11':check_credit,
    '12':check_shoppinglog,
    '13':buy,
    '14':log_off
}

def chief():
    while True:
        print(init_msg)
        choice = input('请输入你想要的操作').strip()
        if choice in init_dic:
            init_dic[choice]()
        elif choice == '0':
            print('exit shopping')
            return
        else:
            print('非法操作，请重新输入操作')

chief()

info_account.log.txt

[INFO] [2018-09-30 13:06:33] [user] lll 注册了
[INFO] [2018-09-30 13:09:46] [user] lhy 注册了
[INFO] [2018-09-30 14:22:43] [user] we 注册了

shopping.log.txt

[INFO] [2018-09-30 11:08:19] [shop] msj 花费 100 元购买了[{'chicken': {'price': 20, 'count': 2}}, {'coffee': {'price': 30, 'count': 2}}]
[INFO] [2018-09-30 13:49:02] [shop] scg 花费 100 元购买了[{'coffee': {'price': 30, 'count': 2}}, {'chicken': {'price': 20, 'count': 2}}]
[INFO] [2018-09-30 13:59:14] [shop] scg 花费 90 元购买了[{'coffee': {'price': 30, 'count': 1}}, {'chicken': {'price': 20, 'count': 3}}]

shoppingcart.txt

msj|{'coffee':{'price':30,'count':2}}
msj|{'car':{'price':100000,'count':1}}
scg|{'iphone':{'price':8000,'count':1}}

account_info.txt

msj|123
scg|321
lll|123456
lhy|123
we|123

bank_info.txt

msj|123|4600|15000
scg|123|6700|15000
swb|123456|200|15000
cxm|123|0|15000
lhy|321|0|15000
mwl|123|0|15000

info_bank.log.txt

[INFO] [2018-09-30 13:45:54] [bank] scg 充值了 2000 元
[INFO] [2018-09-30 13:46:31] [bank] scg 充值了 100 元
[INFO] [2018-09-30 13:47:17][bank] scg 向 msj 转账 500
[INFO] [2018-09-30 13:49:02] [bank] scg 消费 100 元
[INFO] [2018-09-30 13:59:14] [bank] scg 消费 90 元
[INFO][2018-09-30 14:01:10] [bank] scg 还款了 90 元
[INFO] [2018-09-30 14:02:38] [bank] scg 充值了 5000 元