1、作业需求：
模拟实现一个ATM + 购物商城程序
额度 15000或自定义
实现购物商城，买东西加入 购物车，调用信用卡接口结账
可以提现，手续费5%
支持多账户登录
支持账户间转账
记录每月日常消费流水
提供还款接口
ATM记录操作日志
提供管理接口，包括添加账户、用户额度，冻结账户等。。。
用户认证用装饰器
2、流程图
3、目录结构
|——员工信息查询系统 |——bin目录 | |—— _init.py | |__ Stary.py(开始程序) | |——core目录 | |—— __init__.py | |—— main.py（主模块程序） | |—— auth.py（登录认证程序） | |——db目录 | |—— shop_car(购物车文件目录) | |——XXX.txt(购物车文件) | |__ user_info(用户数据) | |——XXX.json(用户信息数据) |——log目录 | |—— card_log(信用卡日志目录) | |——XXX.log(信用卡日志文件) | |__ shop_log(购物日志目录) | |——XXX.log(购物日志文件) |——modules目录 | |—— __init__.py | |—— admincenter.py（管理中心程序） | |—— creditcard.py（信用卡程序） | |—— shopping.py（购物商城程序） |__ __init.py__
4、core目录
auth.py(登录认证模块)
#-*- Coding:utf-8 -*- # Author: D.Gray import os,sys BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASE_DIR) def accse_login(user_data): ''' 定义一个用户登录装饰器 :param user_data: :return: ''' def out_wrapper(func): #func接收 admin_info,shop_info,admin_info函数的返回值 def wrapper(*args,**kwargs): count = 0 if not user_data['is_authenticated'] and count < 3: print("33[32;0m用户登录认证33[0m".center(40, "-")) while count < 3: user = input(' 33[32;1m请输入用户名>>>: 33[0m') db_path = BASE_DIR + r'dbuser_info' db_path_user = db_path + '\%s.json' % user if os.path.isfile(db_path_user): #判断用户文件是否存在 with open(db_path_user, 'r', encoding='utf-8') as fh: user_datas = eval(fh.read()) #将用户文件中内容转换为字典形式 pwd = input(' 33[32;1m请输入用户密码>>>: 33[0m') if pwd == user_datas['password']: user_data['account_id'] = user_datas["cardid"] user_data['is_authenticated'] = True user_data['account_data'] = user_datas break else: print(' 33[31;1m密码错误请重新输入 33[0m') else: count += 1 print(' 33[31;1m该用户不存在,请重新输入还剩 %s 次机会 33[0m'% (3-count)) func(*args,**kwargs) return func return wrapper return out_wrapper
mian.py（主模块程序）
#-*- Coding:utf-8 -*- # Author: D.Gray import sys,os BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASE_DIR) from core import auth from modules import creditcard from modules import shopping from modules import admincenter #用户数据信息 user_data = { 'account_id':None, #帐号ID 'is_authenticated':False, #是否认证 'account_data':None #帐号数据 } def run(): ''' 定义一个主界面函数 :return: ''' while True: print(''' 33[35;1m ************欢迎来到ATM电子商务淫行************ 1.信用卡中心 2.购物中心 3.管理中心 4.退出 33[0m''') inputs = input(' 33[35;1m请选择操作方式>>>: 33[0m').strip() dict = { '1':card_info, '2':shop_info, '3':admin_info } if inputs in dict.keys(): dict[inputs]() elif inputs == '4': exit('程序退出 欢迎下次使用') else: print(' 33[31;1m请输入有效操作方式 33[0m') def admin_info(): ''' 定义一个管理员权限用户主函数 :return: ''' auth_user() #调用用户认证接口 if user_data['account_data']['type'] == 1: while True: print(''' 33[35;1m ---------------欢迎 %s 来到信用卡管理中心--------------- 1. 发行信用卡 2. 冻结信用卡 3. 解冻信用卡 4. 提升信用卡额度 5. 返回主菜单 6. 退出 33[0m'''%user_data['account_data']["username"]) inputs = input(' 33[35;1m请选择操作方式1>>>: 33[0m').strip() menu_dic = { "1": admincenter.banks, "2": admincenter.freezing, "3": admincenter.defrosting, "4": admincenter.limit, } if inputs in menu_dic.keys(): menu_dic[inputs](user_data['account_data']) elif inputs == '5': break elif inputs == '6': exit('程序退出 欢迎下次使用') else: print(' 33[31;1m请输入有效操作方式 33[0m') else: exit('对不起您的账号权限不足无法登录该模块') def shop_info(): ''' 定义一个用户购物的主函数 :return: ''' auth_user() #调用用户认证接口 while True: print(''' 33[35;1m ---------------欢迎来到购物中心--------------- 1.购物商城 2.查看购物车 3.查看购物记录 4.返回主菜单 5.退出 33[0m''') menu_dic = { '1':shopping.shopping, '2':shopping.shop_car, '3':shopping.center } inputs = input(' 33[35;1m请选择操作方式>>>: 33[0m').strip() if inputs in menu_dic.keys(): menu_dic[inputs](user_data['account_data']) elif inputs == '4': break elif inputs == '5': exit('程序退出 欢迎下次使用') else: print(' 33[31;1m请输入有效操作方式 33[0m') def card_info(): ''' 定义一个信用卡管理中心的主函数 :return: ''' auth_user() #调用用户认证接口 while True: print(''' 33[35;1m ---------------欢迎 %s 来到信用卡中心--------------- 1. 账户信息 2. 存款 3. 提现 4. 转账 5. 账单 6. 返回主菜单 7. 退出 33[0m'''%user_data['account_data']["username"]) user_account = user_data['account_data'] inputs = input(' 33[35;1m请选择操作方式1>>>: 33[0m').strip() menu_dic = { "1": creditcard.account_info, "2": creditcard.repay, "3": creditcard.withdraw, "4": creditcard.transfer, "5": creditcard.paycheck, } if inputs in menu_dic.keys(): menu_dic[inputs](user_account) elif inputs == '6': break elif inputs == '7': exit('程序退出 欢迎下次使用') else: print(' 33[31;1m请输入有效操作方式 33[0m') @auth.accse_login(user_data) #装饰器认证 def auth_user(): ''' 调用 auth模块中的accse_login装饰器对用户进行登录认证 :return: ''' print(' 33[32;1m用户 %s 登录认证成功 33[0m' % user_data['account_data']["username"])
5、db（数据库）
{ "status":1, "expire_date": "2021-01-01", "credit":5500, "pay_day":22, "balance":9028, "enroll_date": "2016-01-02", "cardid":1112, "password": "123456", "username": "admin", "type":1, "cardname":"招商银行-星耀" }
IPhone 1299 IWatch 2999 MacBo 1999 IPad 2199 Bicyc 999 X-box 1199 Letv 819 Book 599
6、log（日志目录）
#-*- Coding:utf-8 -*- # Author: D.Gray import logging,sys,os BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASE_DIR) def card_log(username,log): ''' 定义一个信用卡日志函数 :return: ''' log_path = BASE_DIR+r'logcard_log\%s_card.log'%username logger = logging.getLogger('Test_LOG') logger.setLevel(logging.INFO) fh = logging.FileHandler(log_path,encoding='utf-8') #将日志打印到log目录下的日志文件中 fh.setLevel(logging.INFO) fh_format = logging.Formatter('%(asctime)s %(message)s',datefmt='%m/%d/%Y %H:%M:%S') fh.setFormatter(fh_format) logger.addHandler(fh) logger.info(log) logger.removeHandler(fh) #避免打印重复日志 def shop_log(username,log): ''' 定义一个购物日志函数 :param username: :param log: :return: ''' log_path = BASE_DIR + r'logshop_log\%s_shop.log' % username logger = logging.getLogger('Test_LOG') logger.setLevel(logging.INFO) fh = logging.FileHandler(log_path,encoding='utf-8') fh.setLevel(logging.INFO) fh_format = logging.Formatter('%(asctime)s %(message)s', datefmt='%m/%d/%Y %H:%M:%S') fh.setFormatter(fh_format) logger.addHandler(fh) logger.info(log) logger.removeHandler(fh)
7、modules（模块程序）
admincenter.py
#-*- Coding:utf-8 -*- # Author: D.Gray import os,sys,logging BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASE_DIR) db_path = BASE_DIR + r'dbuser_info' def banks(account): ''' 定义一个查询发行信用卡函数 :param account: :return: ''' print(' 33[33;1m尊敬的用户您的发行信用卡是 【%s】' ' 卡号为:%s 开户日期为:%s 信用卡有效期至:%s' ' 我们将会真挚的为您服务！！！' %(account['cardname'],account["cardid"],account["enroll_date"],account["expire_date"])) def freezing(account): ''' 定义一个冻结信用卡函数 :return: ''' db_path_user = db_path+'\%s.json'%account['username'] with open(db_path_user,'r',encoding='utf-8') as fh: fr = fh.read() fd = eval(fr) if fd['status'] == 0: print(' 33[31;1m当前信用卡 【已冻结】 33[0m') if fd['status'] == 1: free = input(' 33[33;1m当前信用卡 【未冻结】 按任意键选择冻结 按b返回>>> 33[0m') if free != 'b': with open(db_path_user,'w',encoding='utf-8') as fw: res = fr.replace(str(fd["status"]),'0',1) fw.write(res) print(' 33[31;1m当前信用卡 【已冻结】 33[0m') def defrosting(account): ''' 定义一个解冻函数 :return: ''' db_path_user = db_path + '\%s.json' % account['username'] with open(db_path_user, 'r', encoding='utf-8') as fh: fr = fh.read() fd = eval(fr) if fd['status'] == 1: print(' 33[31;1m当前信用卡 【未冻结】 33[0m') if fd['status'] == 0: free = input(' 33[33;1m当前信用卡 【已冻结】 按任意键选择解冻 按b返回>>> 33[0m') if free != 'b': with open(db_path_user, 'w', encoding='utf-8') as fw: res = fr.replace(str(fd["status"]), '1', 1) fw.write(res) print(' 33[31;1m当前信用卡 【已解冻】 33[0m') def limit(account): ''' 定义一个提升信用额度函数 :return: ''' db_path_user = db_path + '\%s.json' % account['username'] with open(db_path_user, 'r', encoding='utf-8') as fh: fr = fh.read() fd = eval(fr) print(' 33[33;1m尊敬的用户您当前信用额度是 【%s元】'%account["credit"]) limit = input(' 33[34;1m是否选择提升信用额度 按任意键确认提示 按Q取消提升>>>') if limit.capitalize() != 'Q': while True: lines = input(' 33[35;1m请输入提升信用额度>>> 33[0m') if lines.isdigit(): lines = int(lines) if lines <= 2000: limits = fd['credit'] + lines with open(db_path_user, 'w', encoding='utf-8') as fw: res = fr.replace(str(fd["credit"]), str(limits)) fw.write(res) print(' 33[31;1m当前信用额度提升为： 【%s元】 33[0m'%limits) break else: print(' 33[31;1m提升额度超出提升范围 33[0m') else: print(' 33[31;1m请输入有效提升额度 33[0m')
creditcard.py
#-*- Coding:utf-8 -*- # Author: D.Gray import os,sys,logging BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASE_DIR) db_path = BASE_DIR + r'dbuser_info' log_path = BASE_DIR+r'logcard_log\%s_card.log' from log import loggers def account_info(user_account): ''' 定义一个查看用户信息的函数 :param user_account: user_account = user_data['account_data'] 接收用户字典信息 :return: ''' db_path_user = db_path + '\%s.json' % user_account["username"] with open(db_path_user, 'r',encoding='utf-8') as fh: fd = eval(fh.read()) print(" 33[33;1m我的账户信息： 持卡人: 【%s】 卡号： 【%s】 存款: 【￥%s】 可提现额度: 【￥%s】 33[0m" %(user_account["username"],user_account["cardid"],fd["balance"],fd["credit"])) inputs = input(' 33[33;1m按任意键返回上一级菜单>>>: 33[0m') def repay(user_account): ''' 定义一个用户存款函数 :param user_account: :return: ''' db_path_user = db_path + '\%s.json' % user_account["username"] with open(db_path_user, 'r',encoding='utf-8') as fh: fr = fh.read() fd = eval(fr) print(" 33[33;1m您的当前存款为:【￥%s】"% fd["balance"]) while True: repays = input(' 33[33;1m请输入存款金额并确认存款，按Q取消存款>>> 33[0m') if repays.capitalize() == 'Q': break else: if repays.isdigit(): repays = int(repays) user_balance = fd["balance"] + repays #当前存款=原存款+存款金额 with open(db_path_user,'w',encoding='utf-8') as fh: res = fr.replace(str(fd["balance"]),str(user_balance)) #修改用户文件操作 fh.write(res) print(' 33[33;1m尊敬的用户已为您成功存入 %s 元，您当前存款金额为 %s 元! 33[0m' %(repays,user_balance)) break else: print(' 33[31;1m请输入有效存款金额 33[0m') log = (' 33[31;1m尊敬的用户已为您成功存入: %s元，您当前存款金额: %s元! 33[0m' % (repays, user_balance)) loggers.card_log(user_account["username"], log) def withdraw(user_account): ''' 定义一个用户提现函数 :param user_account: :return: ''' db_path_user = db_path + '\%s.json' % user_account["username"] with open(db_path_user, 'r',encoding='utf-8') as fh: fr = fh.read() fd = eval(fr) print(" 33[33;1m您的当前存款为:【￥%s】" % fd["balance"]) print(" 33[33;1m您可提现额度为:【￥%s】" % fd["credit"]) while True: repays = input(' 33[33;1m请输入提现金额，按Q取消提现>>> 33[0m') if repays.capitalize() == 'Q': #判断用户输入 break else: if repays.isdigit(): repays = int(repays) if repays > fd["credit"]: #判断用户输入的提现金额是否大于现有存款 print(' 33[31;1m提现金额不得大于可提现额度 33[0m') else: print(' 33[31;1m提现金额:【￥%s】 手续费:【￥%s】 33[0m'%(repays,repays*0.05)) user_balance = fd["balance"] - (repays + repays * 0.05) #提现后的用户余额 inputs = input(' 33[33;1m请确认提现金额，按任意键提现>>> 33[0m') if user_balance <= 0: #判断结算后存款是否为负数，如结算后存款为负给出提示不让提现 print(' 33[31;1m当前存款不足以提现 33[0m') else: user_credit = user_account["credit"] - repays #提现后的可提现额度 print(' 33[31;1m尊敬的用户您已成功提现 %s 元! 33[0m'% repays) print(" 33[33;1m您的当前存款为:【￥%s】" % user_balance) print(" 33[33;1m您可提现额度为:【￥%s】" % user_credit) with open(db_path_user, 'w',encoding='utf-8') as fh: res1 = fr.replace(str(fd["balance"]), str(user_balance)) #先扣除用户的余额 res2 = res1.replace(str(fd["credit"]), str(user_credit)) #然后扣除用户的可提现额度 fh.write(res2) #将变更后的余额和提现额度重新写入json文本中 break else: print(' 33[31;1m请输入有效提现金额 33[0m') log = (" 33[31;1m尊敬的用户您已成功提现: %s元 所剩存款: %s元 可提现金额为: %s元! 33[0m" % (repays, user_balance, user_credit)) #打印日志内容 loggers.card_log(user_account["username"], log) #传递参数给card_log日志主函数 def transfer(user_account): ''' 定义一个用户转账的函数 :param user_account: :return: ''' count = 0 while count < 3: transfer = input(' 33[33;1m请输入需转账人用户名>>> 33[0m') db_path_user1 = db_path + '\%s.json' % transfer #需被转账用户 db_path_user2 = db_path + '\%s.json' % user_account["username"] #当前用户 if transfer == user_account["username"]: print(' 33[31;1m转账人不能是自己 33[0m') else: if os.path.isfile(db_path_user1): # 判断用户文件是否存在 with open(db_path_user1, 'r',encoding='utf-8') as fh: frh = fh.read() fd = eval(frh) #需被转账用户 with open(db_path_user2,'r',encoding='utf-8') as fw: frw = fw.read() fc = eval(frw) #当前用户 print(' 33[33;1m转账用户信用卡号为: 33[0m 33[32;1m %s 33[0m'%fd["cardid"]) while True: money = input(' 33[33;1m请输入需转账金额: 33[0m') if money.isdigit(): money = int(money) if money > fc["balance"]: #判断转账金额是否大于存款 print(' 33[31;1m对不起您的存款不足,无法转账 33[0m') else: print(' 33[31;1m转账用户卡号:【%s】 转账金额:【￥%s】 33[0m' % (fd["cardid"],money)) inputs = input(' 33[33;1m请再次确认转账信息数据:按Q取消>>> 33[0m') if inputs.capitalize() == 'Q': break else: user_balance1 = fc["balance"] - money #当前用户转账后余额 with open(db_path_user2,'w',encoding='utf-8') as fw: res1 = frw.replace(str(fc["balance"]),str(user_balance1)) fw.write(res1) print(' 33[33;1m您转账后存款为:【￥%s】'%user_balance1) user_balance2 = fd["balance"] + money #需被转账用户转账后余额 with open(db_path_user1,'w',encoding='utf-8') as fh: res2 = frh.replace(str(fd["balance"]), str(user_balance2)) fh.write(res2) log = (' 33[31;1m您为卡号:%s 用户转账: %s元 您当前存款金额: %s元! 33[0m' % (fd["cardid"], money, user_balance1)) loggers.card_log(user_account["username"], log) break else: print(' 33[31;1m请输入有效转账金额 33[0m') else: count += 1 print(' 33[31;1m该用户不存在,请重新输入还剩 %s 次机会 33[0m' % (3 - count)) break def paycheck(user_account): if not os.path.isfile(log_path % user_account["username"]): print(' 33[31;1m当前用户无流水记录 33[0m') else: with open(log_path % user_account["username"], 'r', encoding='utf-8') as fh: for line in fh: print(line)
shopping.py
#-*- Coding:utf-8 -*- # Author: D.Gray import os,sys,logging BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASE_DIR) shop_path = BASE_DIR+r'dbproduct_list' shop_car_path = BASE_DIR+r'dbshop_car' db_path = BASE_DIR + r'dbuser_info' log_path = BASE_DIR+r'logshop_log\%s_shop.log' from core import main from log import loggers def shopping(user_account): ''' 定义一个购物商城函数 :param user_account: :return: ''' shopcar_list,pro_list = [],[] #shopcar_list:购物车列表 pro_list:商品清单列表 with open(shop_path,'r',encoding='utf-8') as fh: for item in fh: pro_list.append(item.strip(' ').split()) def shop_info(): print(" 编号 商品 价格") for index,item in enumerate(pro_list): print(' %s %s %s'%(index,item[0],item[1])) while True: print((" 33[32;0m目前商城在售的商品信息 33[0m").center(40, "-")) shop_info() choice_id = input(" 33[34;0m选择要购买的商品编号 【购买 ID】/【返回 b】33[0m：") if choice_id.isdigit(): choice_id = int(choice_id) if choice_id < len(pro_list) and choice_id >= 0: #判断用户选择商品编号是否大于商品清单列表最大值 pro_item = pro_list[choice_id] #定义一个 pro_item 变量存储 用户选择的商品 信息和价格 num = input(' 33[34;1m选择商品数量>>> 33[0m') if num.isdigit(): num = int(num) if num > 0: print("33[31;0m商品 %s 加入购物车 价格%s 数量%s33[0m" % (pro_item[0], pro_item[1],num)) shopcar_list.append(pro_item) shop_car_paths = shop_car_path+'\%s_shopcar.txt'%user_account['username'] with open(shop_car_paths,'a',encoding='utf-8') as fc: fc.write(str('%s %s %s')%(pro_item[0],pro_item[1],num) +' ') else: print(' 33[31;1m购买数量不得为0 33[0m') else: print(' 33[31;1m请输入有效购买数量 33[0m') else: print("33[31;0m错误：没有相应的编号 请重新输入:33[0m ") elif choice_id == "b": main.shop_info() else: print("33[31;0m错误：没有相应的编号 请重新输入:33[0m ") #shopping() def shop_car(user_account): ''' 定义一个购物车函数 :param user_account: :return: ''' money_list,product_list= [],[] product_info = '' shop_car_paths = shop_car_path + '\%s_shopcar.txt' % user_account['username'] if not os.path.isfile(shop_car_paths): print(' 33[31;1m您还未有购物记录，请先进入商城购物 33[0m') shopping(user_account) else: with open(shop_car_paths,'r',encoding='utf-8')as fc: print((' 33[32;1m购物车清单 33[0m').center(32, '-')) print("编号 商品 价格 数量") lock_list = fc.readlines() for index ,lock_info in enumerate(lock_list): lock = lock_info.split() product_name = lock[0] money = lock[1] num = lock[2] print('%s %s %s %s'%(index,product_name,money,num)) moneys = int(money)*int(num) #定义moneys变量来计算 单个商品总金额 = 商品金额*数量 product_info = '%s商品%s件'%(product_name,num) #定义打印商品名称及数量 字符串 money_list.append(moneys) #将单个商品总金额 添加至 金额列表中 product_list.append(product_info) #将product_info添加至 购物信息列表中 if sum(money_list) == 0 : # sum(money_list) = 购物车所有商品总金额 print(' 33[31;1m购物车空空如也 33[0m') else: db_path_user = db_path + '\%s.json' % user_account["username"] with open(db_path_user, 'r',encoding='utf-8') as fh: fr = fh.read() fd = eval(fr) print(' 33[33;0m您当前余额为 %s 元 当前商品金额为 %s 元'%(fd['balance'],sum(money_list))) go_shop = input(" 33[34;0m是否选择购买 " "【任意键：购买】/【返回 b】33[0m：") #使用sum方法求出购物车商品总支付金额 if go_shop == 'b': main.shop_info() else: if sum(money_list) < fd["balance"]: #判断用户余额买得起购物商品 balance = fd["balance"] - sum(money_list) # 当前余额 = 原余额-商品总额 log = (' 33[31;1m尊敬的用户您已成功购物 %s ，购物总额为 %s 元，您购物后余额: %s元! 33[0m' %(str(product_list),sum(money_list),balance)) loggers.shop_log(user_account['username'],log) # 调用购物日志打印函数 print(" 33[33;1m购物成功！！！您购物后余额为:【￥%s】 33[0m" % balance) with open(db_path_user,'w',encoding='utf-8') as fh: res = fr.replace(str(fd["balance"]), str(balance)) # 修改用户文件操作 fh.write(res) with open(shop_car_paths,'w',encoding='utf-8') as fc: fc.truncate() #购物支付完成后 购物车文件清空 else: print(" 33[31;0m对不起您的余额不足无法购买: 33[0m ") main.shop_info() def center(user_account): ''' 定义一个查看购物记录的函数 :param user_account: :return: ''' if not os.path.isfile(log_path % user_account["username"]): print(' 33[31;1m当前用户无流水记录 33[0m') else: with open(log_path % user_account["username"], 'r', encoding='utf-8') as fh: for line in fh: print(line) main.shop_info()
Github