作业13
周末作业
编写ATM程序实现下述功能，数据来源于文件db.txt
0、注册功能：用户输入账号名、密码、金额，按照固定的格式存入文件db.txt
1、登录功能：用户名不存在，要求必须先注册，用户名存在&输错三次锁定，登录成功后记录下登录状态（提示：可以使用全局变量来记录）
下述操作，要求登录后才能操作
1、充值功能：用户输入充值钱数，db.txt中该账号钱数完成修改
2、转账功能：用户A向用户B转账1000元，db.txt中完成用户A账号减钱，用户B账号加钱
3、提现功能：用户输入提现金额，db.txt中该账号钱数减少
4、查询余额功能：输入账号查询余额
# 0、注册功能：用户输入账号名、密码、金额，按照固定的格式存入文件db.txt
import os
def logon():
tag = 1
while tag:
print("开始注册")
inp_user = input("请输入要注册的账号:")
inp_password = input("请输入注册账号的密码:")
with open("db.txt", mode="r", encoding="utf-8") as f:
for line in f:
user, password, money = line.strip().split(":")
if user == inp_user:
print("用户名已存在,注册失败")
break
with open("db.txt", mode="a", encoding="utf-8") as f:
f.write("{}:{}:{}\n".format(inp_user, inp_password, 0))
print("注册成功")
tag = 0
break
# logon()
online_user = None
# 1、登录功能：用户名不存在，要求必须先注册，用户名存在&输错三次锁定，登录成功后记录下登录状态（提示：可以使用全局变量来记录）
def login():
global online_user
tag = 1
while tag:
inp_user = input("请输入账号:")
with open("lock_list.txt", mode="r", encoding="utf-8") as f:
for line in f:
user, count = line.strip().split(":")
count = int(count)
if user == inp_user and count == 3:
print("该账号已被锁定")
break
elif user == inp_user and count < 3:
tag = 0
break
else:
count = 0
break
while count < 3:
inp_password = input("请输入密码:")
with open("db.txt", mode="r", encoding="utf-8") as f:
for line in f:
user, password, money = line.strip().split(":")
if user == inp_user and password == inp_password:
online_user = inp_user
print("登陆成功")
return online_user
elif user == inp_user and password != inp_password:
print("密码错误,请重新输入")
count += 1
break
else:
print("您输入的用户名不存在,请先注册")
logon()
break
with open("lock_list.txt", mode="r", encoding="utf-8") as f1, \
open("lock_list.txt.swap", mode="w", encoding="utf-8") as f2:
for line in f1:
if line.startswith(inp_user):
f2.write("{}:{}\n".format(inp_user, str(count)))
else:
f2.write(line)
if tag == 1:
f2.write("{}:{}\n".format(inp_user, str(count)))
tag = 0
os.remove("lock_list.txt")
os.rename("lock_list.txt.swap", "lock_list.txt")
# login()
# print(online_user)
# 1、充值功能：用户输入充值钱数，db.txt中该账号钱数完成修改
online_user = "wu"
import os
def invest():
if online_user:
print("开始充值")
invest_money = input("请输入要充值的钱为多少:")
if not invest_money.isdigit():
print("只能输入数字")
return
else:
invest_money = float(invest_money)
with open("db.txt","r",encoding="utf-8") as f1,\
open("db.txt.swap","w",encoding="utf-8") as f2:
for line in f1:
user, password, balance = line.strip().split(":")
if user == online_user:
balance = str(float(balance)+ invest_money)
f2.write("{}:{}:{}\n".format(user,password,balance))
else:
f2.write(line)
os.remove("db.txt")
os.rename("db.txt.swap","db.txt")
print("充值成功")
else:
print("请先登录")
invest()
# 2、转账功能：用户A向用户B转账1000元，db.txt中完成用户A账号减钱，用户B账号加钱
online_user = "wu"
import os
def transfer():
if online_user:
print("开始转账")
transfer_money = input("请输入要转账的钱为多少:")
if not transfer_money.isdigit():
print("只能输入数字")
return
else:
if float(check_balance())-float(transfer_money)<0:
print("余额不足,无法转账")
return
transfer_user = input("请输入要转账的账号:")
transfer_money = float(transfer_money)
with open("db.txt", "r", encoding="utf-8") as f1:
for line in f1:
user, _,_ = line.strip().split(":")
if user == transfer_user:
break
else:
print("账号不存在")
return
with open("db.txt","r",encoding="utf-8") as f1,\
open("db.txt.swap","w",encoding="utf-8") as f2:
for line in f1:
user, password, balance = line.strip().split(":")
if user == online_user :
balance = str(float(balance)- transfer_money)
f2.write("{}:{}:{}\n".format(user,password,balance))
elif user == transfer_user :
balance = str(float(balance) + transfer_money)
f2.write("{}:{}:{}\n".format(user, password, balance))
tag = 1
else:
f2.write(line)
os.remove("db.txt")
os.rename("db.txt.swap","db.txt")
print("转账成功")
else:
print("请先登录")
# 3、提现功能：用户输入提现金额，db.txt中该账号钱数减少
online_user = "wu"
import os
def with_draw():
if online_user:
print("开始转账")
with_draw_money = input("请输入要提现的钱为多少:")
if not with_draw_money.isdigit():
print("只能输入数字")
return
else:
with_draw_money = int(with_draw_money)
if float(check_balance()) - float(with_draw_money) < 0:
print("余额不足,无法提现")
return
with open("db.txt", "r", encoding="utf-8") as f1, \
open("db.txt.swap", "w", encoding="utf-8") as f2:
for line in f1:
user, password, balance = line.strip().split(":")
if user == online_user:
balance = str(float(balance) - with_draw_money)
f2.write("{}:{}:{}\n".format(user, password, balance))
else:
f2.write(line)
os.remove("db.txt")
os.rename("db.txt.swap", "db.txt")
print("提现成功")
else:
print("请先登录")
# 4、查询余额功能：输入账号查询余额
def check_balance():
if online_user:
with open("db.txt", "r", encoding="utf-8") as f:
for line in f:
user, _, balance = line.strip().split(":")
if user == online_user:
print("您的余额为{}".format(balance))
return balance
else:
print("请先登录")
# check_balance()
# transfer()
# with_draw()
1