# -*- coding: utf-8 -*-
"""
Created on Sat May 6 20:16:34 2023
@author: 北辰远_code
I love python.快乐每一天！
"""
import csv
import datetime
# 创建账户
def create_account():
while True:
account = input("请输入账户名：")
password = input("请输入密码：")
balance = input("请输入初始余额：")
if not account or not password or not balance:
print("账户名、密码和初始余额不能为空！")
else:
with open("accounts.csv", "a", newline="") as f:
writer = csv.writer(f)
writer.writerow([account, password, balance])
print("账户创建成功！")
break
# 登录
def login():
account = input("请输入账户名：")
password = input("请输入密码：")
with open("accounts.csv", "r") as f:
reader = csv.reader(f)
for row in reader:
if row[0] == account and row[1] == password:
print("登录成功！")
return account, float(row[2])
print("账户名或密码错误！")
return None, None
# 存钱
def deposit(account, balance):
amount = input("请输入存款金额：")
try:
amount = int(amount)
if amount <= 0 or amount%100 != 0:
print("无效金额！")
return balance
balance += amount
date = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
with open(f"{account}.csv", "a", newline="") as f:
writer = csv.writer(f)
writer.writerow([date, "存款", amount, balance])
print("存款成功！")
return balance
except ValueError:
print("无效金额！")
return balance
# 取钱
def withdraw(account, balance):
amount = input("请输入取款金额：")
try:
amount = int(amount)
if amount <= 0 or amount > balance or amount%100 != 0:
print("无效金额！")
return balance
balance -= amount
date = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
with open(f"{account}.csv", "a", newline="") as f:
writer = csv.writer(f)
writer.writerow([date, "取款", amount, balance])
print("取款成功！")
return balance
except ValueError:
print("无效金额！")
return balance
# 查看余额
def check_balance(balance):
print(f"当前余额为：{balance}")
# 查看存款和取款记录
def check_records(account):
with open(f"{account}.csv", "r") as f:
reader = csv.reader(f)
for row in reader:
print(row)
# 主函数
def main():
while True:
print("1. 创建账户")
print("2. 登录")
print("3. 存钱")
print("4. 取钱")
print("5. 查看余额")
print("6. 查看存款和取款记录")
print("0. 退出")
choice = input("请选择功能：")
if choice == "1":
create_account()
elif choice == "2":
account, balance = login()
if account:
while True:
print(f"当前余额为：{balance}")
print("1. 存钱")
print("2. 取钱")
print("3. 查看余额")
print("4. 查看存款和取款记录")
print("0. 退出")
choice = input("请选择功能：")
if choice == "1":
balance = deposit(account, balance)
elif choice == "2":
balance = withdraw(account, balance)
elif choice == "3":
check_balance(balance)
elif choice == "4":
check_records(account)
elif choice == "0":
break
else:
print("无效选择！")
elif choice == "0":
break
else:
print("无效选择！")
if __name__ == "__main__":
main()
