【一】需求
# ATM ： 提款姬
# def save_data(file_path,)
# user_pwd_dict ={}
# 文本 ： dream-521
# dream,521
# user_info.txt : 每一个人的信息就是一行
# dream_flow.txt :
# print(time.localtime())
# 注册 ： 存储到文件中 用户名 - 登录密码 - 年龄 - 银行卡号(1314) - 取款密码 - 余额(1000)
# 登录 ： 直接将用户信息从文件中取出，然后进行比对 用户名 - 密码
# 取款 : 验证你的取款密码，更改余额 余额(1000) ，记录你的提款信息 -- 文件里 - 加时间
# 转账 ： 验证你的取款密码，更改余额 目标银行卡号去转 记录你的提款信息 -- 文件里
【二】代码
import time
login_user_dict = {'username': '',
'role': '',
"bank_id": '',
"bank_pay_pwd": "",
"balance": ""
}
def create_data_type(tag, *args, **kwargs):
# print(kwargs) # {'username': 'dream', 'password': '521', 'role': 'admin'}
if tag == 'pwd':
data = '|'.join(kwargs.values()) # dream|521|admin
return data
elif tag == 'bank':
data = '|'.join([str(value) for value in kwargs.values()])
return data
def read_data(path, tag=None):
user_pwd_dict = {}
user_bank_dict = {}
with open(path, 'r', encoding='utf8') as fp:
data = fp.read()
data_list = data.split('\n')
data_list.pop()
if tag == 'pwd':
for data in data_list:
username, password, role = data.split('|')
user_pwd_dict[username] = {'password': password, 'role': role}
# {'dream': {'password': '521', 'role': 'admin'}}
return user_pwd_dict
elif tag == 'bank':
for data in data_list:
username, pay_pwd, bank_id, balance = data.split('|')
user_bank_dict[username] = {'pay_pwd': pay_pwd, 'bank_id': bank_id, 'balance': balance}
return user_bank_dict
def save_data(path, data, mode='a'):
with open(path, mode, encoding='utf8') as fp:
fp.write(data + '\n')
def get_user_pwd():
# 【1】要获取用户名和密码
username = input("用户名 :>>>> ").strip()
password = input("密码 :>>>> ").strip()
return username, password
def register():
username, password = get_user_pwd()
# 【2】拼接存储数据的格式
if username == 'dream' and password == '521':
role = 'admin'
else:
role = 'normal'
pwd_data = create_data_type(tag='pwd', username=username, password=password, role=role)
bank_data = create_data_type(tag='pwd', username=username, bank_password='None', bank_id='0', balance='0')
# print(pwd_data) # dream|521|admin
# 【3】存储数据
# [3.1]用户名和密码
save_data(path='user_pwd.txt', data=pwd_data)
save_data(path='user_log.txt', data=f'当前用户 {username} 注册成功 | {time.strftime("%x %X")}')
# [3.2]银行信息
save_data(path='user_bank.txt', data=bank_data)
save_data(path='user_log.txt', data=f'当前用户 {username} 初始化银行卡成功 | {time.strftime("%x %X")}')
# print(bank_data)# dream|None|0|0
return True, f'{username} 注册成功!'
def init_bank_data(username):
user_bank_dict = read_data(tag='bank', path='user_bank.txt')
user_bank_data = user_bank_dict.get(username)
login_user_dict['bank_id'] = user_bank_data.get('bank_id')
login_user_dict['bank_pay_pwd'] = user_bank_data.get('pay_pwd')
login_user_dict['balance'] = user_bank_data.get('balance')
def login():
username, password = get_user_pwd()
user_pwd_dict = read_data(tag='pwd', path='user_pwd.txt')
user_data = user_pwd_dict.get(username)
if not user_data:
return False, '没有注册'
if password != user_data.get('password'):
return False, '密码失败!'
login_user_dict['username'] = username
login_user_dict['role'] = user_data.get('role')
init_bank_data(username)
save_data(path='user_log.txt', data=f'当前用户 {username} 登录成功 | {time.strftime("%x %X")}')
return True, f'用户 {username} 登陆成功'
def login_auth(func):
def inner(*args, **kwargs):
# print(func.__name__) # add_bank_info
# 【1】执行原函数的前提是已经登录过
if not login_user_dict['username']:
return False, f'必须先登录'
if not login_user_dict['bank_id'] and func.__name__ != 'add_bank_info':
return False, '必须先激活银行卡'
return func(*args, **kwargs)
return inner
@login_auth
def add_bank_info():
username = login_user_dict['username']
bank_pay_pwd = input("请输入支付密码 :>>>> ").strip()
bank_id = input("请输入银行卡号 :>>>> ").strip()
bank_data = create_data_type(tag='bank', username=username, bank_pay_pwd=bank_pay_pwd, bank_id=bank_id,
balance=1000)
save_data(path='user_bank.txt', data=bank_data)
init_bank_data(username)
print(login_user_dict)
return True, f'初始化银行信息完成!'
def input_money_info():
bank_pwd = input("请输入支付密码 :>>>> ").strip() # 确认是数字并且是三位
balance = input("请输入金额 :>>>> ").strip() # 余额是数字类型并且余额大于0小于你的余额
bank_id = input("请输入银行卡号 :>>>> ").strip() # 确认是数字并且是六位
return bank_pwd, balance, bank_id
# 在装饰器内部完成 --- 格式校验。位数校验。数字余额校验
def charge_auth(flag):
def wrapper(func):
def inner(*args, **kwargs):
if flag == 'charge':
bank_pwd, balance, bank_id = input_money_info()
if len(bank_pwd) != 3:
return False, f'支付密码必须是3位'
if not balance.isdigit():
return False, f'必须是数字'
if balance < '0' or balance > login_user_dict.get('balance'):
return False, f'输入金额格式错误'
if not bank_id.isdigit():
return False, f'必须是数字'
if len(bank_id) != 6:
return False, f'必须是六位'
result = func(*args, bank_pay_pwd=bank_pwd, bank_id=bank_id, balance=balance)
return result
if flag == 'recharge':
balance = input("请输入金额 :>>>> ").strip() # 余额是数字类型并且余额大于0小于你的余额
if not balance.isdigit():
return False, f'必须是数字'
if balance < '0':
return False, f'输入金额格式错误'
bank_id = input("请输入银行卡号 :>>>> ").strip() # 确认是数字并且是六位
if not bank_id.isdigit():
return False, f'必须是数字'
if len(bank_id) != 6:
return False, f'必须是六位'
result = func(*args, bank_id=bank_id, balance=balance)
return result
return inner
return wrapper
@login_auth
@charge_auth(flag='charge')
def charge(*args, **kwargs):
bank_info_dict = read_data('user_bank.txt', tag='bank')
# {'dream': {'password': '521', 'role': 'admin'}} 银行信息类似这种
bank_data_1 = bank_info_dict.get(login_user_dict.get('username'))
balance = int(bank_data_1.get('balance')) - int(kwargs.get('balance'))
bank_data = create_data_type(tag='bank', username=login_user_dict.get('username'),
bank_pay_pwd=kwargs.get('bank_pay_pwd'),
bank_id=kwargs.get('bank_id'), balance=balance)
save_data(path='user_bank.txt', data=bank_data)
save_data(path='user_log.txt', data=f"当前用户 {login_user_dict.get('username')} 取{kwargs.get('balance')}元子成功 | {time.strftime('%x %X')}")
return True, f"当前用户 {login_user_dict.get('username')} 取了{kwargs.get('balance')}元子"
@login_auth
@charge_auth(flag='recharge')
def recharge(*args, **kwargs):
user_bank_info = read_data('user_bank.txt', tag='bank')
bank_data_2 = user_bank_info.get(login_user_dict.get('username'))
balance = int(bank_data_2.get('balance')) + int(kwargs.get('balance'))
bank_data = create_data_type(tag='bank', username=login_user_dict.get('username'),
bank_pay_pwd=login_user_dict.get('bank_pay_pwd'),
bank_id=kwargs.get('bank_id'), balance=balance)
save_data(path='user_bank.txt', data=bank_data)
save_data(path='user_log.txt', data=f"当前用户 {login_user_dict.get('username')} 存{kwargs.get('balance')}元子成功 | {time.strftime('%x %X')}")
return True, f"当前用户 {login_user_dict.get('username')} 存了{kwargs.get('balance')}元子"
@login_auth
def check_flow():
with open('user_log.txt', 'r', encoding='utf8') as fp:
for line in fp:
print(line)
return True, f'打印成功'
@login_auth
def check_bank_info():
user_bank_info_dict = read_data('user_bank.txt', tag='bank')
bank_info_dict = user_bank_info_dict.get(login_user_dict.get('username'))
return True, f'''
姓名：{login_user_dict.get('username')}
支付密码：{bank_info_dict.get('pay_pwd')}
银行卡号：{bank_info_dict.get('bank_id')}
余额：{bank_info_dict.get('balance')}
'''
def money_auth(func):
def inner(*args, **kwargs):
bank_pwd, balance, bank_id = input_money_info()
target_bank_card_number = input('请输入目标银行卡号：').strip()
if target_bank_card_number == bank_id:
return False, f'不能转账给相同卡号'
if len(bank_pwd) != 3:
return False, f'支付密码必须是3位'
if not balance.isdigit():
return False, f'必须是数字'
if balance < '0' or balance > login_user_dict.get('balance'):
return False, f'输入金额格式错误'
if not bank_id.isdigit():
return False, f'必须是数字'
if len(bank_id) != 6:
return False, f'必须是六位'
result = func(*args, balance=balance, target_bank_card_number=target_bank_card_number)
return result
return inner
@login_auth
@money_auth
def transfer(*args, **kwargs):
user_bank_info = read_data('user_bank.txt', tag='bank')
bank_data = user_bank_info.get(login_user_dict.get('username'))
balance = int(bank_data.get('balance')) - int(kwargs.get('balance'))
bank_data = create_data_type(tag='bank', username=login_user_dict.get('username'),
bank_pay_pwd=login_user_dict.get('bank_pay_pwd'),
bank_id=kwargs.get('bank_id'), balance=balance)
save_data(path='user_bank.txt', data=bank_data)
save_data(path='user_log.txt',
data=(f"当前用户 {login_user_dict.get('username')} 给{kwargs.get('target_bank_card_number')}转了{kwargs.get('balance')}元子 | {time.strftime('%x %X')}"))
return True,f'转账成功'
def over():
return False, f'谢谢使用，欢迎下次光临'
func_menu = '''
===================用户功能菜单=====================
1.注册
2.登陆
3.初始化银行信息
4.取款
5.存款
6.查看流水
7.查看银行信息(查看自己的卡号、余额、流水等信息)
8.转账
9.退出
======================欢迎使用=======================
'''
func_dict = {
'1': register,
'2': login,
'3': add_bank_info,
'4': charge,
'5': recharge,
'6': check_flow,
'7': check_bank_info,
'8': transfer,
'9': over,
}
def main():
while True:
print(func_menu)
func_id = input("功能ID :>>>> ").strip()
if func_id not in func_dict.keys():
print(f"不存在")
continue
func = func_dict[func_id]
flag, msg = func()
if flag:
print(msg)
else:
print(msg)
break
continue
main()
浙公网安备 33010602011771号