首先我们要做的就是登陆的方法了。
因为涉及到从数据库中查找用户的操作，所以我们需要用到前面的 DBHelper 类，因为每一个方法都要使用数据库，所以我把 db 作为 class 的一个属性。
from User import User
from Item import Item
from DBHelper import DBHelper
class ATM:
"""定义ATM机的类"""
def __init__(self):
self._current_user = None
self._db = DBHelper()
# ------------登录--------------------------
def _login(self):
while True:
print("登录ATM")
number = input("请输入卡号:\n")
db_user = self._db.get_user(number)
if db_user:
password = input("请输入密码：\n")
if db_user.get_password() == password:
user = User(db_user.get_name(), db_user.get_number(),
password,
db_user.get_balance())
self._current_user = user
print("欢迎",user,"登录")
return number
print("账户或密码错误")
经过调试，最终实现了初步的登录功能：
image.png
用的是我室友名
正常来说，用户登录也是应该做记录的，不过这里我就不写这么多了。
简单实现就好。
在用户进行登录之后，系统应该提供菜单选项，用来给用户选择操作选项。
后来想想之前的记录的问题，填写信息的时候好像要求的是真实姓名，所以姓名是不能改的所以那个问题也就没有了。
在线注册功能：
是实现用户直接注册一个新的账户的功能。用户的卡号现在是由用户自己输入，而不是有系统自动分配，所以还有待改进。
def open_an_account(self):
"""在线开户功能"""
while True:
name = input("请输入用户名：")
if name == "quit":
return False
password = input("请输入密码")
re_password = input("请再次输入密码")
number = input("请输入卡号")
# TODO 先让用户输入卡号, 正常来讲应该由系统分配
if re_password == password:
user = User(name, number, password, 0)
self._db.insert_user(user)
return True
else:
print("两次输入的密码不符！请重新注册")
这里倒是还没出什么大问题。
image.png
image.png
现在有一点遗憾就是输入密码的那一块，现在把密码输入变成*号，那样会更有安全性，现在输入密码基本上是不会错的。
在线销户
有了开户功能，就有销户功能。
销户是有个重要的事，所以需要再三确认：
于是就有了下面的代码：
def closing_an_account(self):
"""在线销户功能，删除user表中额数据和log中的日志"""
n = 3
while n:
password = input("请输入密码")
if self._current_user.get_password() == password:
confirm = input("请输入“注销账户”")
if confirm == "注销账户":
self._db.del_user(self._current_user.get_number())
return True
else:
print("输入错误！/")
else:
print("密码错误，请重新输入")
n -= 1
从数据库把室友的号删了，，
你号没了！
接下来就是更改用户摩玛的方法了：幸亏当时写好了更改数据库的方法，现在写起来事半功倍。
def change_password(self):
"""在ATM机上一般是改不了密码的，其他的几个属性也是改不了的，但是
我在DBHelper里写的相关方法却可以更改所有的属性。"""
n = 3
while n:
old_password = input("请输入旧密码")
new_password = input("请输入新密码")
re_new_password = input("请再次输入新密码")
if old_password == self._current_user.get_password():
if new_password == re_new_password:
self._db.update_user(self._current_user, "password",
new_password)
return True
else:
print("新密码不一致")
else:
print("密码不正确")
n = n-1
image.png
接下来是存款的功能：
中间有点小插曲，不过还是完成了，本来不想实现日志添加的，但是还是实现了，为了试一试，为了给室友加点存款，就先写个循环。
def deposit(self):
"""存款余额"""
money = input("请输入金额：")
if self._db.update_user(self._current_user, "balance", int(money) +
self._current_user.get_balance()):
status = "成功"
print("存款成功")
else:
status = "失败"
print("存款失败")
item = Item(self._current_user, "存款", 1000, status)
if self._db.insert_log(item):
return True
image.png
image.png
。。。。。。emmmmm不对，为什么都是1000
奥原来是前面记录写的有问题，我都写1000，，，但是也不对啊，余额都没变呀。。。。但是
image.png
user表中的数据却变了。。。
这是怎么回事、、、、、、
奥！~~~我明白了，因为我改变了数据库中的数据，但是当前的程序运行时，我的用户的信息是么变的所以。。。
。所以我在改变用户数据的时候就把ATM机的当前用户的属性改变。于是就有了下面的代码：
def deposit(self):
"""存款"""
money = input("请输入金额：")
if self._db.update_user(self._current_user, "balance", int(money) +
self._current_user.get_balance()):
self._current_user.set_balance(int(money) +
self._current_user.get_balance())
status = "成功"
print("存款成功")
else:
status = "失败"
print("存款失败")
item = Item(self._current_user, "存款", int(money), status)
if self._db.insert_log(item):
return True
image.png
现在好了。。睡觉了.