readme.txt
ATM + 购物车 # 一个项目是如何从无到有的 1.需求分析 注册,登陆,查看余额,支付,购物车, 提现,还款,转账,查看流水,注销,管理员, 查看购物车,登陆认证装饰器, 密码加密 2.程序的架构设计 三层架构: 用户功能层: 接收用户输入的内容,展示给用户的内容. 小的逻辑判断,例如两次密码是否一致. 接口层: 处理业务逻辑. 数据处理层: 对数据进行增删查改. 3.分任务开发 4.测试 5.上线运行 注册,登陆,查看余额,提现,还款,转账,查看流水,注销,登陆认证装饰器,密码加密. 今日内容: 购物车, 查看购物车, 添加日志, 管理员 ATM + 购物车 前戏: 一个项目是如何从无到有的. 一 需求分析 对项目需求进行分析,并提取出相应的功能. - 额度15000或自定义 ---> 注册功能 - 实现购物商城，买东西加入购物车，调用信用卡接口结账 ---> 购物车, 支付功能 - 可以提现，手续费5% ---> 提现 - 支持多账户登录 ---> 登陆 - 支持账户间转账 ---> 转账 - 记录每月日常消费流水 ---> 记录流水 - 提供还款接口 ---> 还款 - ATM记录操作日志 ---> 日志功能 - 提供管理接口，包括添加账户、用户额度，冻结账户等... ---> 管理员功能 - 用户认证功能 ---> 登陆认证,使用装饰器 二 程序的架构设计 用户功能层: 负责接收用户输入的内容,并返回结果给用户. 做一些小的逻辑判断. 接口层: 处理业务逻辑. 数据处理层: 对文件进行处理. 增,删,查,改操作. 三层架构的好处: 1.代码结构清晰 2.可扩展性强. 3.易维护,管理. 三 分任务开发 - CTO - 技术总监 - 架构师 - 项目经理 - 普通开发 UI: 用户界面设计师 前端: 网页的开发 后端: 写业务逻辑,写接口 测试: 测试软件 运维: 部署项目. 四 测试 - 手动测试 传统人工去手动测试. - 自动化测试 通过脚本模拟人的行为,自动化执行测试. - 黑盒测试: 对用户界面进行测试. - 白盒测试: 对软件的性能进行测试,例如每分钟能接收多少并发量. 五 上线运行 把测试好的代码交给运维人员, 部署上线.
bin/start.py
import os import sys BASE_DIR = os.path.dirname(os.path.dirname(__file__)) sys.path.append(BASE_DIR) from core import src if __name__ == '__main__': src.run()
conf/settings.py
import os BASE_PATH = os.path.dirname(os.path.dirname(__file__)) # 拼接db目录路径 DB_PATH = os.path.join(BASE_PATH, 'db') """ logging配置 """ # 定义三种日志输出格式 开始 standard_format = '[%(asctime)s][%(threadName)s:%(thread)d][task_id:%(name)s][%(filename)s:%(lineno)d]' \ '[%(levelname)s][%(message)s]' # 其中name为getlogger指定的名字 simple_format = '[%(levelname)s][%(asctime)s][%(filename)s:%(lineno)d]%(message)s' id_simple_format = '[%(levelname)s][%(asctime)s] %(message)s' # 定义日志输出格式 结束 logfile_dir = os.path.join(BASE_PATH, 'log') # log文件的目录 logfile_name = 'atm_shop_log.log' # log文件名 # 如果不存在定义的日志目录就创建一个 if not os.path.isdir(logfile_dir): os.mkdir(logfile_dir) # log文件的全路径 logfile_path = os.path.join(logfile_dir, logfile_name) # log配置字典 LOGGING_DIC = { 'version': 1, 'disable_existing_loggers': False, 'formatters': { 'standard': { 'format': standard_format }, 'simple': { 'format': simple_format }, }, 'filters': {}, 'handlers': { # 打印到终端的日志 'console': { 'level': 'DEBUG', 'class': 'logging.StreamHandler', # 打印到屏幕 'formatter': 'simple' }, # 打印到文件的日志,收集info及以上的日志 'default': { 'level': 'DEBUG', 'class': 'logging.handlers.RotatingFileHandler', # 保存到文件 'formatter': 'standard', 'filename': logfile_path, # 日志文件 'maxBytes': 1024 * 1024 * 5, # 日志大小5M 'backupCount': 5, 'encoding': 'utf-8', # 日志文件的编码，再也不用担心中文log乱码了 }, }, 'loggers': { # logging.getLogger(__name__)拿到的logger配置 '': { 'handlers': ['default', 'console'], # 这里把上面定义的两个handler都加上，即log数据既写入文件又打印到屏幕 'level': 'DEBUG', 'propagate': True, # 向上（更高level的logger）传递 }, }, }
db/db_handler.py
import os import json from conf import settings # 保存用户数据 def save(user_dic): # 拼接用户保存文件 user_path = f'{settings.DB_PATH}/{user_dic["user"]}.json' # 把用户信息写入文件中 with open(user_path,'w',encoding='utf-8') as f: json.dump(user_dic,f,ensure_ascii=False) f.flush() # 查看用户数据 def select(user): # base_path = os.path.dirname(os.path.dirname(__file__)) # db_path = os.path.join(base_path,'db') user_path = f'{settings.DB_PATH}/{user}.json' # 调用数据处理层的select函数,查看用户信息 if os.path.exists(user_path): # 若存在, 读取用户信息 with open(user_path,'r',encoding='utf-8') as f: user_dic = json.load(f) return user_dic
lib/common.py
import hashlib import logging.config from conf import settings # MD5 加密功能 def get_md5(pwd): # 加盐 val = '天山折梅手' md5 = hashlib.md5() md5.update(pwd.encode('utf-8')) md5.update(val.encode('utf-8')) res = md5.hexdigest() return res # 装饰器 def login_auth(func): from core import src def inner(*args,**kwargs): # 判断用户是否存在 if not src.user_info['user']: print('请先登录') src.register() pass res = func(*args,*kwargs) return res return inner # 日志功能 def get_logger(type_name): logging.config.dictConfig(settings.LOGGING_DIC) logger = logging.getLogger(type_name) return logger if __name__ == '__main__': res = get_md5('123') print(res) logger = get_logger('tank')
core/src.py
from interface import user_interface,bank_interface,shop_interface,admin_interface from lib import common user_info = {'user':None} # 1.注册 def register(): print('注册功能...') while True: user = input('plealse your name:>>>').strip() # 调用接口判断用户是否存在 flag = user_interface.check_user_interface(user) # 若存在,则让用户重新输入 if flag: print('user 已经 存在') continue pwd = input("please your pwd:>>>").strip() re_pwd = input('please your re_pwd:>>>') if pwd == re_pwd: # 业务逻辑 # 调用接口层注册接口保存用户信息 msg = user_interface.register_interface(user,pwd) if msg: print(msg) break else: print('注册失败') else: print('密码错误') # 2.登录 def login(): print('登录功能...') while True: user = input('plealse your name:>>>').strip() # 调用接口判断用户是否存在 flag = user_interface.check_user_interface(user) # 若不存在,则让用户重新输入 if not flag: print('请重新输入') continue pwd = input("please your pwd:>>>").strip() # 调用登录接口 flag,msg = user_interface.login_interface(user,pwd) if flag: print(msg) # 登陆成功后做一个记录 user_info['user'] = user break else: print(msg) continue # 3.查看余额 @common.login_auth def check_bal(): print('余额接口') bal = user_interface.check_bal_interface(user_info['user']) print(bal) # 4.提现 @common.login_auth def withdraw(): print('提现功能') while True: money = input('please input withdraw money:>>>').strip() if not money.isdigit(): print('请输入数字!') continue money = int(money) flag,msg = bank_interface.withdraw_interface(user_info['user'],money) if flag: print(msg) break else: print(msg) # 5.还款 @common.login_auth def repay(): print('还款功能') while True: money = input('请输入还款金额: ').strip() if not money.isdigit(): print('请输入数字') continue money = int(money) flag,msg = bank_interface.repay_interface(user_info['user'],money) if flag: print(msg) break # 6.转账 @common.login_auth def transfer(): print('转账功能') while True: to_user = input('请输入目标用户').strip() # 判断目标用户是否存在 flag = user_interface.check_user_interface(to_user) if not flag: print('目标用户不存在!') continue # 输入转账金额 money = input('请输入转账金额:').strip() if not money.isdigit(): print('请输入数字!') continue money = int(money) flag, msg = bank_interface.transfer_interface( to_user, user_info['user'], money) if flag: print(msg) break else: print(msg) # 7.查看流水 @common.login_auth def check_flow(): print('查看流水') flow_list = bank_interface.check_flow_interface(user_info['user']) for flow in flow_list: print(flow) # 8.购物车功能 @common.login_auth def shopping(): good_list = [ ['J哥牌辣椒酱', 5], ['Egon牌公仔', 10], ['nick牌T-shirt', 100], ['围城', 39], ['tank牌坦克', 50000], ] user_bal = user_interface.check_bal_interface(user_info['user']) shop_cart = {} cost = 0 while True: # 打印商品信息 for index, good_price in enumerate(good_list): print(index, good_price) # 选择商品编号或者输入q退出 choice = input('请选择商品编号或者输入q退出').strip() if choice.isdigit(): choice = int(choice) if choice >= 0 and choice < len(good_list): good_name, good_price = good_list[choice] if user_bal >= good_price: if good_name in shop_cart: shop_cart[good_name] += 1 else: shop_cart[good_name] = 1 cost += good_price else: print('*穷*, 请打工赚钱再来购买!') else: print('请输入正确商品编号.') elif choice == 'q': commit = input('是否确认结账,请输入y/n: ').strip() if commit == 'y': # 调用购车商城支付功能, 并调用银行支付接口 flag, msg = shop_interface.buy_shop_interface( user_info['user'], cost) if flag: print(msg) break else: print(msg) elif commit == 'n': # 调用添加购物车接口 shop_interface.add_shop_cart_interface(user_info['user'], shop_cart) break # 9.查看购物车 @common.login_auth def check_shop_cart(): shop_cart = shop_interface.check_shop_cart_interface(user_info['user']) if not shop_cart: print('购物车是空的,请选择购物车功能.') print(shop_cart) # 10.注销 @common.login_auth def logout(): print('注销功能') flag,msg = user_interface.logout_interface() if flag: print(msg) # 冻结用户 def lock_user(): print('冻结用户...') user = input('请输入需要冻结的用户: ').strip() flag = user_interface.check_user_interface(user) if flag: # 调用冻结接口 msg = admin_interface.lock_interface(user) print(msg) else: print('用户不存在!') def unlock_user(): print('解冻用户...') user = input('请输入需要解冻的用户: ').strip() flag = user_interface.check_user_interface(user) if flag: # 调用冻结接口 msg = admin_interface.unlock_interface(user) print(msg) else: print('用户不存在!') def change_limit(): print('修改用户额度') user = input('请输入需要修改额度的用户: ').strip() flag = user_interface.check_user_interface(user) if not flag: print('用户不存在!') return limit = input('请输入修改额度').strip() if limit.isdigit(): limit = int(limit) # 调用修改额度接口 msg = admin_interface.change_limit_interface(user, limit) print(msg) else: print('请输入数字') admin_func_dic = { '1': lock_user, '2': unlock_user, '3': change_limit, } # 11.管理员功能 def admin_sys(): while True: print(''' 1.冻结用户 2.解冻用户 3.用户额度 ''') choice = input('请选择管理员功能: ').strip() if choice not in admin_func_dic: print('请重新输入') continue admin_func_dic[choice]() func_dic = { '1': register, '2': login, '3': check_bal, '4': withdraw, '5': repay, '6': transfer, '7': check_flow, '8': shopping, '9': check_shop_cart, '10': logout, '11': admin_sys, } def run(): while True: print(""" 1.注册 2.登录 3.查看余额 4.提现 5.还款 6.转账 7.查看流水 8.购物车功能 9.查看购物车 10.注销 11.管理员功能 """) choice = input('please your choice:>>>').strip() if choice == 'q':break if choice in func_dic: func_dic[choice]() else: print('please True number') continue
interface/user_interface.py
from db import db_handler from lib import common user_logger = common.get_logger('user') # 查看用户接口 def check_user_interface(user): # 调用数据处理层的select函数,查看用户信息 user_dic = db_handler.select(user) if user_dic: return True # 注册接口 def register_interface(user,pwd,balance=15000): # 密码加密 pwd = common.get_md5(pwd) user_dic = { 'user':user, 'pwd':pwd, 'balance':15000, 'flow':[], 'shop_cart':{}, 'lock':False } # 调用保存数据功能 db_handler.save(user_dic) user_logger.info(f'{user_dic["user"]}用户注册成功') return f'{user_dic["user"]}用户注册成功' # 登录接口 def login_interface(user,pwd): user_dic = db_handler.select(user) pwd = common.get_md5(pwd) if user_dic['lock']: user_logger.info(f'{user}已经被冻结,请联系管理员') return False,'用户已被冻结,请联系管理员!' if user_dic['pwd'] == pwd: user_logger.info(f'{user}登录成功') return True,f'{user}登录成功' user_logger.info(f'{user}输入密码错误') return False,f'{user}密码错误' # 查看余额接口 def check_bal_interface(user): user_dic = db_handler.select(user) return user_dic['balance'] # 查看注销接口 def logout_interface(): from core import src user = src.user_info['user'] src.user_info['user'] = None return True,f'{user}用户注销成功.欢迎下次再来'
interface/bank_interface.py
from db import db_handler from lib import common bank_logger = common.get_logger('bank') # 提现接口 def withdraw_interface(user,money): user_dic = db_handler.select(user) if user_dic['balance'] >= money * 1.05: user_dic['balance'] -= money * 1.05 msg = f'用户{user}提现{money}元成功' user_dic['flow'].append(msg) db_handler.save(user_dic) return True,msg else: msg = f'用户{user}余额不足' return False,msg # 还款接口 def repay_interface(user,money): user_dic = db_handler.select(user) user_dic['balance'] += money msg = f'用户{user}, 还款{money}元成功!' user_dic['flow'].append(msg) db_handler.save(user_dic) return True,msg # 转账接口 def transfer_interface(to_user,from_user,money): to_user_dic = db_handler.select(to_user) from_user_dic = db_handler.select(from_user) if from_user_dic['balance'] >= money: # 加减钱操作 from_user_dic['balance'] -= money to_user_dic['balance'] += money # 拼接流水信息 to_user_flow = f'{to_user}用户接收到用户{from_user}转账{money}元' from_user_flow = f'{from_user}用户给到用户{to_user}转账{money}元' # 记录流水 from_user_dic['flow'].append(from_user_flow) to_user_dic['flow'].append(to_user_flow) # 保存用户信息 db_handler.save(from_user_dic) db_handler.save(to_user_dic) return True, from_user_flow return False, '余额不足,请充值!' # 查看流水接口 def check_flow_interface(user): user_dic = db_handler.select(user) return user_dic['flow'] # 银行支付接口 def pay_interface(user, cost): user_dic = db_handler.select(user) if user_dic['balance'] >= cost: user_dic['balance'] -= cost user_dic['flow'].append(f'{user}用户支付{cost}成功!') db_handler.save(user_dic) return True else: return False
interface/admin_interface.py
from db import db_handler from lib import common admin_logger = common.get_logger('admin') # 冻结用户接口 def lock_interface(user): user_dic = db_handler.select(user) user_dic['lock'] = True db_handler.save(user_dic) return f'{user}用户冻结成功!' def unlock_interface(user): user_dic = db_handler.select(user) user_dic['lock'] = False db_handler.save(user_dic) return f'{user}用户解冻成功!' # 修改额度接口 def change_limit_interface(user, limit): user_dic = db_handler.select(user) user_dic['balance'] = limit db_handler.save(user_dic) return f'修改用户[{user}]额度成功!'
interface/shop_interface.py
from interface import bank_interface from db import db_handler from lib import common shop_logger = common.get_logger('shop') # 商城结账接口 def buy_shop_interface(user,cost): flag = bank_interface.pay_interface(user, cost) if flag: return True,'购物成功' else: return False,'余额不足,支付失败' # 添加购物车接口 def add_shop_cart_interface(user, shop_cart): user_dic = db_handler.select(user) old_cart = user_dic['shop_cart'] # 循环当前购物车 for shop in shop_cart: if shop in old_cart: # 获取当前购物车中的商品数量 number = shop_cart[shop] # 给用户信息中的商品数量做 增值运算 old_cart[shop] += number else: # 获取当前购物车中的商品数量 number = shop_cart[shop] old_cart[shop] = number user_dic['shop_cart'].update(old_cart) db_handler.save(user_dic) return True, '添加购物车成功!' # 查看购物车接口' def check_shop_cart_interface(user): user_dic = db_handler.select(user) return user_dic['shop_cart']
log
