51,411
社区成员
发帖
与我相关
我的任务
分享
import csv
import time
import bank_main
user_info = []
flag = 0
#显示功能模块
def show_menu():
print("欢迎来到【银行资金管理系统】,请登录:")
print("Press<Enter>")
user_select = int(input('直接登录请按【1】,注册请按【2】:'))
if(2 == user_select):
user_register()
else:
user_login(user_info)
#传入用户注册信息进行登录
def user_login(user_info):
print('登录成功!')
funds_trade(user_info[0])
#提供用户注册功能
def user_register():
global user_info
user_info.append(input('请输入用户名:'))
user_info.append(input('请输入用户密码:'))
print('您已注册成功!\n即将跳转至登录')
#资金存取
def funds_trade(user_name):
while True:
print('-'*20 + '【银行资金管理系统】' + '-'*20)
print('1:资金存入或取出')
print('2:交易账单打印')
print('3:退出系统')
print('-'*60)
num = int(input('请输入相关数字进行相关操作:'))
#用户输入3退出
if(3 == num):
global flag
flag = 1
break
left = 0
while True:
#存取资金
if(1 == num):
money = int(input('请输入您存(+)取(-)的金额:'))
print(money)
if(money >= 0):
left += money
print('数据更新成功!')
print('数据保存成功!')
with open('user_funds.csv', 'a+') as file:
data = [user_name, time.ctime(), money, left]
writer = csv.writer(file)
writer.writerow(data)
print('你已成功存入{0}元!,当前余额:{1}'.format(money,left))
user_select = int(input('继续存取请按【1】,返回主菜单请按【2】'))
#回到主菜单
if(2 == user_select):
break
elif (1 == user_select):
pass
else:
print('您的输入有误!请重新输入')
else:
left += money
print('数据更新成功!')
print('数据保存成功!')
with open('user_funds.csv', 'a+') as file:
data = [user_name, time.ctime(), money, left]
writer = csv.writer(file)
writer.writerow(data)
print('你已成功取出{0}元!,当前余额:{1}'.format(money.__abs__(), left))
user_select = int(input('继续存取请按【1】,返回主菜单请按【2】'))
if (2 == user_select):
break
elif (1 == user_select):
pass
else:
print('您的输入有误!请重新输入')
#打印存取信息
elif(2 == num):
funds_print()
break
else:
print('您的输入有误,请重新输入')
#打印交易详情
def funds_print():
with open('user_funds.csv', 'r') as file:
reader = csv.reader(file)
for rows in reader:
if(rows==[]):
continue
print(rows)
book_main.py代码
import bank_tools
import csv
if __name__ == '__main__':
with open('user_funds.csv', 'w', encoding='gbk') as file:
data = ['用户名', '时间', '存入/支出', '余额']
writer = csv.writer(file)
writer.writerow(data)
while True:
bank_tools.show_menu()
if(bank_tools.flag):
break
