ATM三层架构思路：
启动文件
首先，创建一个start.py作为整个项目启动的启动文件。
然后导入os和sys模块，从core中导入src，也就是用户视图层。
然后添加解释器的环境变量，再开始执行项目函数。
在src.py 用户视图层中，先写好各个功能函数名，然后创建一个函数功能字典。
然后就是视图层的主程序，打印选择列表让用户选择，根据选择的编号来调用不同的功能函数。
配置信息
在conf的setting.py中，编写配置文件。
写好项目的根目录路径BASE_PATH。
写好商品信息路径GOODS_PATH。
写好用户信息路径USER_DATA_PATH
然后写上日志配置（日志格式可以复制粘贴过来）
公共方法
在lib的common.py中存放公共方法。
导入conf的settings配置。
写上用户的登录认证装饰器。
导入hashlib模块，写上密码的md5加密，以json格式存储密码。
导入logging.config，添加日志功能，在接口层使用。
商品信息
在db的goods_list.json中编写商品数据，以json的格式存储：
{
"0": {"goods_name": "logitech-G703", "price": 549},
"1": {"goods_name": "Antec-HCG750W", "price": 799},
"2": {"goods_name": "ROG-M11E-Z390", "price": 6999},
"3": {"goods_name": "G.SKILL-16G-RGB", "price": 769},
"4": {"goods_name": "Intel-i9-9900KF", "price": 3599},
"5": {"goods_name": "SAMSUNG-970Pro-1T", "price": 2699},
"6": {"goods_name": "Segotep-K1(ATX3.0)", "price": 499},
"7": {"goods_name": "ROG-RTX-2080Ti-11G", "price": 11999},
"8": {"goods_name": "CHERRY-MX6.0-BLUE RGB", "price": 1399},
"9": {"goods_name": "ASUS-ROG-360mm-RGB Aura", "price": 2299},
"10": {"goods_name": "SAMSUNG-27inch-Space-2K/144Hz", "price": 3499}
}
数据处理层
在db的db_handler.py 数据处理层中，导入json和os模块，从conf中导入settings配置。
编写查看数据函数，接收接口层传过来的username用户名，拼接用户json文件路径。
校验用户json文件，打开数据，并返回给接口层。
编写保存数据函数，用于添加新数据或者更新数据。
拼接用户的数据字典，保存用户数据。
编写获取商品数据函数，用于获取商品数据。
商品数据也是以json格式读取的。
注册功能
编写interface的user_interface.py的注册接口。
查看用户是否存在，若存在，返回“用户名已存在!”
若不存在，就编写一个用户字典保存为“用户名.json”。
其中，如果用户名是“admin”，就是管理员，字典信息中的“is_admin”为“True”。
然后调用数据处理层的db_handler的save功能进行保存。
在core的src.py 用户视图层中，编写注册函数。
先定义一个全局遍历login_user = None，用来记录用户是否已登录。
让用户输入用户名和密码进行注册，2次密码必须相同。
调用接口层的注册接口，将用户名与密码交给接口层来进行处理。
输入的用户名如果是admin，就是管理员，其余是普通用户。
根据flag判断用户注册是否成功，flag控制break的结束。
登录功能
编写interface的user_interface.py的登录接口。
先调用数据处理层的db_handler的select功能查看当前用户数据是否存在。
如果存在，就继续执行，如果不存在，就输出“用户不存在，请重新输入！”
再根据get用户字典的‘locker’的True或False判断用户是否被冻结。
然后给用户的密码做一次加密，和取过来的加密的用户密码比对，查看是否一致。
如果一致，就登录成功；如果不一致，登录失败。
在core的src.py 用户视图层中，编写登录函数。
让用户输入用户名和密码。
调用接口层的登录接口，将用户名与密码交给接口层来进行处理。
输入的用户名如果是admin，就是管理员，其余是普通用户。
根据flag判断用户注册是否成功，flag控制break的结束。
查看余额功能
编写interface的user_interface.py的查看余额接口。
编写user_dic调用数据处理层的db_handler的select功能查看当前用户的余额。
返回给用户视图层。
在core的src.py 用户视图层中，编写查看余额函数。
只有登录的用户才能查看。
调用接口层的查看余额接口，传入当前登录的用户名。
打印用户名和用户金额。
提现功能
编写interface的user_interface.py的查看提现接口。
编写user_dic调用数据处理层的db_handler的select功能查看当前用户的余额。
把当前余额变成整型，然后计算手续费。
判断用户金额是否足够，如果足够，就修改用户字典中的金额，如果不足够，就退出并提示“余额不足！”
记录流水，调用数据处理层的db_handler的save和update功能保存并更新数据。
将结果返回给用户视图层。
在core的src.py 用户视图层中，编写提现函数。
只有登录的用户才能提现。
让用户输入提现金额，判断用户输入的金额是否是数字，如果不是，就重新输入。
用户提现金额，将提现的金额交付给接口层来处理，传入当前登录的用户名。
调用接口层的查看余额接口，传入当前登录的用户名。
打印用户名和提现信息和余额信息。
还款功能
编写interface的user_interface.py的查看还款接口。
编写user_dic调用数据处理层的db_handler的select功能查看当前用户的余额。
把当前余额变成整型，然后计算还款额。
记录流水，调用数据处理层的db_handler的save和update功能保存并更新数据。
将结果返回给用户视图层。
在core的src.py 用户视图层中，编写还款函数。
只有登录的用户才能提现。
让用户输入提现金额，判断用户输入的金额是否是数字，如果不是，就重新输入。
用户还款金额，将还款的金额交付给接口层来处理，传入当前登录的用户名。
调用接口层的查看余额接口，传入当前登录的用户名。
打印用户名和还款信息和余额信息。
转账功能
编写interface的bank_interface.py的转账接口。
编写login_dic和to_user_dic调用数据处理层的db_handler的select功能。
获取“当前用户”和“目标用户”字典，判断目标用户是否存在，若用户存在，则判断“当前用户转账金额”是否足够。
若足够，给当前用户的数据，做减钱操作；给目标用户的数据，做加钱操作，然后记录双方流水。
调用数据处理层的db_handler的save功能保存数据。
将结果返回给用户视图层。
在core的src.py 用户视图层中，编写转账函数。
只有登录的用户才能提现。
判断用户输入金额是否是数字或者>0，如果不是，就重新输入。
让用户输入转账的用户和金额。将输入的信息交付给接口层来处理，传入当前登录的用户名、对方用户名、金额。
调用接口层的，打印用户名和还款信息和余额信息。
流水功能
编写interface的bank_interface.py的查看流水接口。
编写user_flow_dic字典，调用调用数据处理层的db_handler的select功能，传入当前用户。
将结果返回给用户视图层。
在core的src.py 用户视图层中，编写查看流水函数。
调用interface的bank_interface.py接口，传入当前用户。
调用流水接口，输出流水信息。
购物功能
编写interface的shop_interface.py的购物接口。
编写goods_dic字典，调用数据处理层的db_handler的select_goods功能查看商品信息。
将结果返回给用户视图层。
编写interface的shop_interface.py的add_shop_car_interface购物车添加接口。
将用户视图层传来的视频信息存入字典。
如果商品不在购物车中则直接加入，商品在购物车中，修改商品数量以及总价。
将结果返回给用户视图层。
在core的src.py 用户视图层中，编写购物函数。
调用interface的bank_interface.py接口，获取全部商品信息并打印。
将对应编号的商品加入购物车，调用add_shop_car_interface购物车添加接口。
调用购物车添加接口，已经入购物车的信息。
购物车结算功能
编写interface的shop_interface.py的查看购物车接口。
编写user_dic字典，调用调用数据处理层的db_handler的select功能，传入当前用户。
编写shop_car_dic字典，取出user_dic字典中的"shop_car"，然后记录流水信息。
将结果返回给用户视图层。
在core的src.py 用户视图层中，编写查看购物车函数。
调用interface的shop_interface.py接口，获取用户字典里的"shop_car"。
打印购物车内的信息，让用户选择要购买的购物车内的商品ID，输入c清空购物车，输入q退出该界面。
如果ID不存在，就跳过本次循环，输出信息“商品不存在”。
获取选择付款商品的商品信息并打印。
管理员功能
编写interface的admin_interface.py的查看管理员接口。
编写4种功能：添加用户直接调用注册函数；修改额度调用用户字典直接修改，冻结和解冻用户修改用户字典的“locker”的bool值。
将结果返回给用户视图层。
在core下单独编写一个admin.py。
编写4个功能函数，调用各个接口。
编写管理员功能字典：
admin_dic = {
'1': add_user,
'2': change_balance,
'3': lock_user,
'4': unlock_user
}
编写管理员视图和功能选项，打印选择列表让用户选择，根据选择的编号来调用不同的功能函数。
在core的src.py 用户视图层中，编写管理员函数。
直接调用core下的admin.py的admin.run()函数执行功能。
如果用户名是admin的话，就是管理员，如果不是，就是普通用户，没有资格使用该功能。
conf：
settings.py
存放配置信息
import os
获取项目的 根目录路径
BASE_PATH = os.path.dirname( os.path.dirname(file) )
goods_list文件目录路径
GOODS_PATH = os.path.join(BASE_PATH, 'db', 'goods_list.json')
获取 user_data 文件夹的 目录路径
USER_DATA_PATH = os.path.join( BASE_PATH, 'db', 'user_data' )
""" 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'
定义日志输出格式 结束
****************注意1: log文件的目录
BASE_PATH = os.path.dirname(os.path.dirname(file)) logfile_dir = os.path.join(BASE_PATH, 'log')
print(logfile_dir)
****************注意2: log文件名
logfile_name = 'log.log'
如果不存在定义的日志目录就创建一个
if not os.path.isdir(logfile_dir): os.mkdir(logfile_dir)
log文件的全路径
logfile_path = os.path.join(logfile_dir, logfile_name)
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）传递 }, }, }
core：
admin.py
from core import src from interface import admin_interface
添加用户
def add_user(): src.register()
修改用户额度
def change_balance(): while True: # 1.输入修改的用户名 change_user = input('请输入需要修改额度的用户：').strip()
# 2.输入修改用户额度
money = input('请输入需要修改的用户额度：').strip()
if not money.isdigit():
continue
# 3.调用修改额度接口
flag, msg = admin_interface.change_balance_interface(
change_user, money
)
if flag:
print(msg)
break
else:
print(msg)
冻结用户
def lock_user(): while True: # 1.输入懂的用户名 to_lock_user = input('请输入需要冻结的用户：').strip()
# 2.调用冻结用户接口
flag, msg = admin_interface.lock_user_interface(
to_lock_user
)
if flag:
print(msg)
break
else:
print(msg)
解冻用户
def unlock_user(): while True: # 1.输入懂的用户名 to_unlock_user = input('请输入需要解冻的用户：').strip()
# 2.调用冻结用户接口
flag, msg = admin_interface.unlock_user_interface(
to_unlock_user
)
if flag:
print(msg)
break
else:
print(msg)
管理员功能字典
admin_dic = { '1': add_user, '2': change_balance, '3': lock_user, '4': unlock_user }
def admin_run(): while True: print(''' ======== 管理员功能 ======== | 1.添加账户 | | 2.修改额度 | | 3.冻结账户 | | 4.解冻账户 | ========= THE END ========= ''') choice = input('请输入管理员功能编号（按q退出）：').strip()
if choice.lower() == 'q':
break
# 判断功能编号是否存在
if choice not in admin_dic:
print('请输入正确的功能编号!')
continue
# 调用用于选择的功能函数
admin_dic.get(choice)() # admin_dic.get('1')() ---> add_user()
src.py
''' 用户视图层 ''' from interface import user_interface from interface import bank_interface from interface import shop_interface from lib import common
全局变量，记录用户是否已登录
login_user = None
1.注册功能
def register(): print('注册功能执行中...') while True: # 1.让用户输入用户名与密码进行校验 username = input('请输入用户名: ').strip() password = input('请输入密码: ').strip() re_password = input('请确认密码: ').strip() # 可以输入自定义的金额
# 小的逻辑处理: 比如两次密码是否一致
if password == re_password:
# 2.调用接口层的注册接口，将用户名与密码交给接口层来进行处理
# res ---> (False, '用户名已存在!')
# res = user_interface.register_interface(
# flag, msg ---> (flag---> False, msg --> '用户名已存在!')
# (True, 用户注册成功)， (False, 注册失败)
flag, msg = user_interface.register_interface(
username, password
)
# 3.根据flag判断用户注册是否成功，flag控制break的结束
if flag:
print(msg)
break
else:
print(msg)
2.登录功能
def login(): print('登录功能执行中...') # 登录视图 while True: # 1.让用户输入用户名与密码 username = input('请输入用户名: ').strip() password = input('请输入密码: ').strip()
# 2.调用接口层，将数据传给登录接口
# (True, f'用户: [{username}] 登录成功!'),
# (return False, '密码错误'), (False, '用户不存在，请重新输入！')
flag, msg = user_interface.login_interface(
username, password
)
if flag:
print(msg)
# 记录用户信息，已登录
global login_user
login_user = username
break
else:
print(msg)
3.查看余额
@common.login_auth def check_balance(): print('查看余额功能执行中...') # 1.直接调用查看余额接口，获取用户余额 balance = user_interface.check_bal_interface( login_user )
print(f'用户[{login_user}] 账户余额为: [{balance}元]！')
4.提现功能
@common.login_auth def withdraw(): print('提现功能执行中...') while True: # 1.让用户输入提现金额 input_money = input('请输入提现金额: ').strip()
# 2.判断用户输入的金额是否是数字
if not input_money.isdigit():
print('请重新输入')
continue
# 3.用户提现金额，将提现的金额交付给接口层来处理
flag, msg = bank_interface.withdraw_interface(
login_user, input_money
)
if flag:
print(msg)
break
else:
print(msg)
5.还款功能
@common.login_auth def repay(): print('还款功能执行中...') while True: # 1.让用户输入还款金额 input_money = input('请输入还款金额: ').strip()
# 2.判断用户输入的金额是否是数字
if not input_money.isdigit():
print('请重新输入')
continue
input_money = int(input_money)
if input_money > 0:
# 3.用户还款金额，将还款的金额交付给接口层来处理
flag, msg = bank_interface.repay_interface(
login_user, input_money
)
if flag:
print(msg)
break
else:
print(msg)
else:
print('输入的金额不能小于等于0！')
6.转账功能
@common.login_auth def transfer(): print('转账功能执行中...') ''' 1.接收用户输入的 转账目标用户 2.接收用户输入的 转账金额 ''' while True:
# 1.让用户输入转账用户和金额
to_user = input('请输入转账目标用户：').strip()
money = input('请输入转账金额：').strip()
# 2.判断用户输入金额是否是数字或者>0
if not money.isdigit():
print('请输入正确的金额！')
continue
money = int(money)
if money > 0:
# 3.调用转账接口
flag, msg = bank_interface.transfer_interface(
# 当前用户，目标用户，转账金额
login_user, to_user, money
)
if flag:
print(msg)
break
else:
print(msg)
else:
print('请输入正确的金额！')
7.查看流水
@common.login_auth def check_flow(): print('查看流水功能执行中...') flow_list = bank_interface.check_flow_interface( login_user )
if flow_list:
for flow in flow_list:
print(flow)
else:
print('当前用户没有流水！')
8.购物功能
@common.login_auth def shopping(): print('购物功能执行中...') while True: # 获取全部商品信息并打印 goods_dic = shop_interface.get_goods_interface() print('=================== Welcome To Hardware World ===================
') for number, goods_info in goods_dic.items(): print(f'| 编号：[{number}] 商品名：[{goods_info.get("goods_name")}] 价格：[{goods_info.get("price")}]元') print('') print("============================ THE END ============================") goods_id = input("请输入要购买的商品编号(按q返回上一层)：").strip() if goods_id.lower() == 'q': break count = input("请输入购买的数量：").strip() if not count.isdigit() or count == '0': print("输入错误") continue if goods_id not in goods_dic: print('该商品不存在') continue # 将对应编号的商品加入购物车 shop_interface.add_shop_car_interface(goods_dic[goods_id], int(count)) print(f"{goods_dic[goods_id]['goods_name']}已加入购物车,请到购物车结算")
作业：清空购物车功能
清空购物车
def clean_shop_car(): inp_clear = input("是否清空购物车？（Y/N）:").strip() if inp_clear.lower() != 'y': print("清空购物车失败，用户取消操作") return False msg = shop_interface.clean_shop_car_interface(login_user) print(msg) return True
9.查看购物车
@common.login_auth def check_shop_car(): print('查看购物车功能执行中...') shop_car_dic = shop_interface.check_shop_car_interface(login_user) while True: shop_id = 0 print("======================== 购物车内容 =========================") print('') for goods_name, goods_info in shop_car_dic.items(): print(f'| ID：[{shop_id}] 商品名 {goods_name} 数量 {goods_info.get("count")}个 总价 {goods_info.get("total")}元') shop_id += 1 print('') print("========================== THE END ==========================") pay_id = input("请选择需要付款的商品ID(按q离开购物车,按c清空购物车)：").strip() if pay_id.lower() == 'c': flog = clean_shop_car() if flog: return continue if pay_id.lower() == 'q': break if int(pay_id) not in range(len(shop_car_dic)): print("商品不存在") continue # 获取选择付款商品的商品信息并打印 buy_goods_name = list(shop_car_dic.keys())[int(pay_id)] buy_goods_count = shop_car_dic[buy_goods_name].get('count') buy_goods_total = shop_car_dic[buy_goods_name].get('total') print(f"您将付款的商品为：{buy_goods_name},数量为：{buy_goods_count}个, 总价为：{buy_goods_total}元") # 付款 inp_buy = input("确认付款(Y/n)：").strip() if inp_buy.lower() == 'y': flog, msg = shop_interface.payment_interface(login_user, buy_goods_name, buy_goods_total) if flog: print(msg) break else: print('msg') print("付款失败，用户取消支付")
@common.login_auth
10.管理员功能
def admin(): print('管理员功能执行中...') from core import admin if login_user == 'admin': admin.admin_run() else: print('你没有资格！')
创建函数功能字典
func_dic = { '1': register, '2': login, '3': check_balance, '4': withdraw, '5': repay, '6': transfer, '7': check_flow, '8': shopping, '9': check_shop_car, '10': admin, }
视图层主程序
def run(): while True: print(''' ======= ATM + 购物车 ======= | 1.注册功能 | | 2.登录功能 | | 3.查看余额 | | 4.提现功能 | | 5.还款功能 | | 6.转账功能 | | 7.查看流水 | | 8.购物功能 | | 9.查看购物车 | | 10.管理员功能 | =========== END =========== ''')
choice = input('请输入功能编号: ').strip()
if choice not in func_dic:
print('请输入正确的功能编号!')
continue
func_dic.get(choice)() # func_dic.get('1')() ---> register()
db：
userdata
db_handler.py
''' 数据处理层 '''
import os import json from conf import settings
查看数据
def select(username): # 1.接收接口层传过来的username用户名，拼接用户json文件路径 user_path = os.path.join( settings.USER_DATA_PATH, f'{username}.json' )
# 2.校验用户json文件是否存在
if os.path.exists(user_path):
# 3.打开数据，并返回给接口层
with open(user_path, mode='rt', encoding='UTF-8') as f:
user_dic = json.load(f)
return user_dic
保存数据（添加新数据或者更新数据）
def save(user_dic): # 1.拼接用户的数据字典 username = user_dic.get('username')
user_path = os.path.join(
settings.USER_DATA_PATH, f'{username}.json'
)
# 2.保存用户数据
with open(user_path, mode='wt', encoding='UTF-8') as f:
# ensure_ascii=False让文件中的中文数据，显示更美观
json.dump(user_dic, f, ensure_ascii=False)
获取商品数据
def select_goods(): with open(settings.GOODS_PATH, mode='rt', encoding='UTF-8') as f: goods_dic = json.load(f) return goods_dic
goods_list.json
{ "0": {"goods_name": "logitech-G703", "price": 549}, "1": {"goods_name": "Antec-HCG750W", "price": 799}, "2": {"goods_name": "ROG-M11E-Z390", "price": 6999}, "3": {"goods_name": "G.SKILL-16G-RGB", "price": 769}, "4": {"goods_name": "Intel-i9-9900KF", "price": 3599}, "5": {"goods_name": "SAMSUNG-970Pro-1T", "price": 2699}, "6": {"goods_name": "Segotep-K1(ATX3.0)", "price": 499}, "7": {"goods_name": "ROG-RTX-2080Ti-11G", "price": 11999}, "8": {"goods_name": "CHERRY-MX6.0-BLUE RGB", "price": 1399}, "9": {"goods_name": "ASUS-ROG-360mm-RGB Aura", "price": 2299}, "10": {"goods_name": "SAMSUNG-27inch-Space-2K/144Hz", "price": 3499} }
interface：
admin_interface.py
from db import db_handler from lib import common
根据不同的接口类型传入不同的日志对象
admin_logger = common.get_logger(log_type='admin')
修改额度接口
def change_balance_interface(username, money): user_dic = db_handler.select(username)
if user_dic:
# 修改额度
user_dic['balance'] = int(money)
# 保存修改后用户的数据
db_handler.save(user_dic)
msg = f'管理员修改用户: [{username}]额度修改成功!'
admin_logger.info(msg)
return True, '额度修改成功！'
return False, '该用户不存在！'
冻结接口
def lock_user_interface(username): user_dic = db_handler.select(username)
if user_dic:
if not user_dic['locked']:
# 将locked的默认值改为True
user_dic['locked'] = True
db_handler.save(user_dic)
msg = f'用户：[{username}]冻结成功!'
admin_logger.info(msg)
return True, msg
else:
return False, '该用户已被冻结，无需再次冻结！'
return False, '冻结用户不存在！'
解冻接口
def unlock_user_interface(username): user_dic = db_handler.select(username)
if user_dic:
if user_dic['locked']:
# 将locked的默认值改为False
user_dic['locked'] = False
db_handler.save(user_dic)
msg = f'用户：[{username}]解冻成功!'
admin_logger.info(msg)
return True, msg
else:
return False, '该用户未被冻结，无需解冻！'
return False, '解冻用户不存在！'
bank_interface.py
银行相关业务的接口
import time from db import db_handler from lib import common
根据不同的接口类型传入不同的日志对象
bank_logger = common.get_logger(log_type='bank')
记录当前操作的时间
start_time = time.strftime('%Y-%m-%d %H:%M:%S')
提现接口(手续费5%)
def withdraw_interface(username, money): # 1.先获取用户字典 user_dic = db_handler.select(username)
# 账户中的金额
balance = int(user_dic.get('balance'))
# 提现本金 + 手续费
money2 = int(money) * 1.05 # ---> float
# 判断用户金额是否足够
if balance >= money2:
# 2.修改用户字典中的金额
balance -= money2
user_dic['balance'] = balance
# 3.记录流水
flow = f'{start_time} 用户：[{username}] 提现金额：[{money}元]成功，手续费为: [{money2 - float(money)}元]'
user_dic['flow'].append(flow)
# 4.再保存数据，或更新数据
db_handler.save(user_dic)
bank_logger.info(flow)
return True, flow
return False, '提现金额不足，请重新输入!'
还款接口(无需手续费)
def repay_interface(username, money): # 1.先获取用户字典 user_dic = db_handler.select(username)
# 2.直接做加钱操作
# user_dic['balance'] ==> int
user_dic['balance'] += money
new_money = user_dic['balance']
# 3.记录流水
flow = f'{start_time} 用户：[{username}] 还款：[{money}]成功!'
user_dic['flow'].append(flow)
# 4.调用数据处理层，将修改后的数据更新
db_handler.save(user_dic)
bank_logger.info(f'{start_time} 用户：{username} 还款成功，还款金额：[{money}]元')
return True, flow
转账接口
def transfer_interface(login_user, to_user, money): ''' 1.获取“当前用户”数据 2.获取“目标用户”数据 3.获取“转账金额”数据 :return: '''
# 1.获取“当前用户”字典
login_user_dic = db_handler.select(login_user)
# 2.获取“目标用户”字典
to_user_dic = db_handler.select(to_user)
# 3.判断目标用户是否存在
if not to_user_dic:
bank_logger.warning(f'{start_time} 用户：[{login_user}] 转账失败，用户：[{to_user}]不存在')
return False, '目标用户不存在！'
# 4.若用户存在，则判断“当前用户转账金额”是否足够
if login_user_dic['balance'] >= money:
# 5.若足够，则开始给目标用户转账
# 5.1.给当前用户的数据，做减钱操作
login_user_dic['balance'] -= money
# 5.2.给目标用户的数据，做加钱操作
to_user_dic['balance'] += money
# 5.3.记录当前用户 与 目标用户 的流水
# 当前用户流水
login_user_flow = f'{start_time} 用户：[{login_user}] 给 用户：[{to_user}] 转账：[{money}元]'
login_user_dic['flow'].append(login_user_flow)
# 目标用户流水
to_user_flow = f'{start_time} 用户：[{to_user}] 接收 用户：[{login_user}] 转账：[{money}元]'
to_user_dic['flow'].append(to_user_flow)
# 6.调用数据处理层的save功能，保存用户数据
# 6.1.保存当前用户数据
db_handler.save(login_user_dic)
# 6.2.保存目标用户数据
db_handler.save(to_user_dic)
bank_logger.info(f'{login_user} 转账成功，对方用户：[{to_user}]，转账金额[{money}]元')
return True, login_user_flow
else:
bank_logger.warning(f'{start_time} 用户：[{login_user}] 转账失败，余额不足')
return False, '余额不足，转账失败！'
查看流水接口
def check_flow_interface(login_user): user_flow_dic = db_handler.select(login_user) return user_flow_dic['flow']
支付接口
def pay_interface(login_user, cost): user_dic = db_handler.select(login_user)
# 判断用户金额是否足够
if user_dic.get('balance') >= cost:
# 扣费
user_dic['balance'] -= cost
# 记录消费流水
flow = f'用户消费金额：[{cost}元]'
user_dic['flow'].append(flow)
# 保存数据
db_handler.save(user_dic)
# return True 或者 return False 交给购物接口来操作
return True
return False
shop_interface.py
商品购买接口
import time from db import db_handler from lib import common
根据不同的接口类型传入不同的日志对象log_type=
shop_logger = common.get_logger(log_type='shop')
start_time = time.strftime('%Y-%m-%d %H:%M:%S')
获取商品信息接口
def get_goods_interface(): goods_dic = db_handler.select_goods() return goods_dic
商品准备结算接口
def payment_interface(login_user, buy_goods_name, buy_goods_total): user_dic = db_handler.select(login_user) # 获取用户购物车字典 shop_car_dic = user_dic['shop_car'] # 获取用户余额 balance = user_dic['balance'] if buy_goods_total > balance: shop_logger.warning(f'用户：[{login_user}] 支付失败，余额不足') return False, '支付失败，余额不足' balance -= buy_goods_total shop_car_dic.pop(buy_goods_name) # 修改用户余额 user_dic['balance'] = balance # 删除已付款商品 user_dic['shop_car'] = shop_car_dic # 增加流水 user_dic['flow'].append(f'{start_time} 用户：[{login_user}] 购买商品：[{buy_goods_name}] 消费：[{buy_goods_total}]元') # 更新用户数据 db_handler.save(user_dic) shop_logger.info(f'{start_time} 用户：{login_user} 付款成功！ 商品：[{buy_goods_name}] 等待发货') return True, '支付成功'
购物车添加接口
def add_shop_car_interface(goods_dic, count): from core import src total = goods_dic["price"] * count goods_name = goods_dic['goods_name'] shop_dic = { f"{goods_name}": {"count": count, "total": total} } user_dic = db_handler.select(src.login_user) # 用户购物车内商品信息 shop_car_dic = user_dic['shop_car'] # 如果购物车为空则直接加入 if user_dic.get('shop_car') == {}: user_dic['shop_car'] = shop_dic else: # 如果商品不在购物车中则直接加入 if goods_name not in shop_car_dic: shop_car_dic.update(shop_dic) user_dic['shop_car'] = shop_car_dic else: # 商品在购物车中，修改商品数量以及总价 user_dic['shop_car'].get(goods_name)['count'] += count user_dic['shop_car'].get(goods_name)['total'] += total db_handler.save(user_dic) shop_logger.info(f'{start_time} 用户：{src.login_user} 添加商品：[{goods_name}]至购物车成功')
return True, '添加购物车成功!'
获取购物车信息接口
def check_shop_car_interface(login_user): user_dic = db_handler.select(login_user) shop_car_dic = user_dic["shop_car"] shop_logger.info(f'{start_time} 用户：[{login_user}] 查看购物车') return shop_car_dic
清空购物车
def clean_shop_car_interface(login_user): user_dic = db_handler.select(login_user) user_dic['shop_car'] = {} db_handler.save(user_dic) shop_logger.info(f'{start_time} 用户：[{login_user}] 清空了购物车') return '清空购物车成功'
userface_interface.py
逻辑接口层
用户接口
from db import db_handler from lib import common
根据不同的接口类型传入不同的日志对象
user_logger = common.get_logger(log_type='user')
注册接口
def register_interface(username, password, balance=15000): # 2.查看用户是否存在 # 2.1.调用 数据处理层 中的 select函数，会返回 用户字典 或 None user_dic = db_handler.select(username)
# {user: user, pwd: pwd...} or None
# 若用户存在，则return，告诉用户重新输入
if user_dic:
# return (False, '用户名已存在!')
return False, '用户名已存在!'
# 3.若用户不存在，则保存用户数据
# 做密码加密
password = common.get_pwd_md5(password)
if username == 'admin':
user_dic = {
'username': username,
'password': password,
'balance': balance,
# 用于记录用户流水的列表
'flow': [],
# 用于记录用户购物车
'shop_car': {},
# locked：用于记录用户是否被冻结
# False: 未冻结 True: 已被冻结
'is_admin': True
}
# 3.1.组织用户的数据的字典信息
user_dic = {
'username': username,
'password': password,
'balance': balance,
# 用于记录用户流水的列表
'flow': [],
# 用于记录用户购物车
'shop_car': {},
# locked：用于记录用户是否被冻结
# False: 未冻结 True: 已被冻结
'locked': False,
'is_admin': False
}
# 3.2.保存数据
db_handler.save(user_dic)
msg = f'{username} 新用户注册成功, 初始额度为：{balance}元'
# 3.3.记录日志
user_logger.info(msg)
return True, msg
登录接口
def login_interface(username, password): # 1.先查看当前用户数据是否存在 # {用户数据字典} or None user_dic = db_handler.select(username)
# 2.判断用户是否存在
if user_dic:
# 若有冻结用户，则需要判断是否被锁定
if user_dic.get('locked'):
return False, '当前用户已被冻结，好好反省一下自己做了什么！'
# 给用户输入的密码做一次加密
password = common.get_pwd_md5(password)
# 3.校验密码是否一致
if password == user_dic.get('password'):
msg = f'用户: [{username}] 登录成功!'
user_logger.info(msg)
return True, msg
else:
msg = f'用户: [{username}]密码错误!'
user_logger.warn(msg)
return False, msg
msg = f'用户: [{username}]用户不存在，请重新输入！!'
return False, msg
查看余额接口
def check_bal_interface(username): user_dic = db_handler.select(username) return user_dic['balance']
lib：
common.py
存放公共方法
import hashlib from conf import settings import logging.config
md5加密
def get_pwd_md5(password): md5_obj = hashlib.md5() md5_obj.update(password.encode('UTF-8')) salt = '急急如律令!' md5_obj.update(salt.encode('UTF-8')) return md5_obj.hexdigest()
登录认证装饰器
def login_auth(func): from core import src
def inner(*args, **kwargs):
if src.login_user:
res = func(*args, **kwargs)
return res
else:
print('未出示证明，无法享受美好的功能服务!')
src.login()
return inner
添加日志功能: (日志功能在接口层 使用)
def get_logger(log_type): # log_type ---> user ''' :param log_type: 比如是 user日志，bank日志，购物商城日志 :return: ''' # 1.加载日志配置信息 logging.config.dictConfig( settings.LOGGING_DIC )
# 2.获取日志对象
logger = logging.getLogger(log_type)
return logger