首先我们要做的就是登陆的方法了。

因为涉及到从数据库中查找用户的操作，所以我们需要用到前面的 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

现在好了。。睡觉了.