Python实现ATM系统

下面是详细讲解“Python实现ATM系统”的完整攻略。

1. 需求分析

在实现ATM系统之前，需要先进行需求分析。具体来说，我们需要考虑以下几个方面的需求：

• 用户登录和认证；
• 查看余额；
• 存款和取款；
• 转账和还款。

2. 类和数据库设计

在了解完需求之后，我们需要对ATM系统进行类和数据库设计。具体来说，我们可以设计以下几个类：

• 用户类；
• 账户类；
• ATM机类。

同时，我们还需要为ATM系统设计一个数据库来存储用户和账户信息。

3. 实现代码

在进行了类和数据库设计之后，我们就可以开始编写实现ATM系统的代码了。具体来说，我们可以按照以下步骤逐步实现：

3.1 创建数据库

我们需要创建一个数据库来存储用户和账户信息。我们可以使用SQLite作为数据库。

import sqlite3

conn = sqlite3.connect('atm.db')
c = conn.cursor()

c.execute('''CREATE TABLE IF NOT EXISTS users
             (id INTEGER PRIMARY KEY, name TEXT, password TEXT, balance INTEGER)''')

c.execute('''CREATE TABLE IF NOT EXISTS accounts
             (id INTEGER PRIMARY KEY, user_id INTEGER, account_type TEXT, balance INTEGER)''')

conn.commit()
conn.close()

3.2 注册和登录功能

为了让用户能够使用ATM系统，我们需要提供注册和登录功能。具体来说，我们可以编写以下两个方法：

def register(self, name, password):
    sql = 'INSERT INTO users (name, password, balance) VALUES (?, ?, ?)'
    self.cursor.execute(sql, (name, password, 0,))
    self.connection.commit()

def login(self):
    name = input('请输入用户名：')
    password = input('请输入密码：')
    sql = 'SELECT * FROM users WHERE name = ? AND password = ?'
    data = self.cursor.execute(sql, (name, password,))
    row = data.fetchone()
    if row:
        self.current_user = row[1]
        print('登录成功')
    else:
        print('用户名或密码错误')

3.3 查看余额功能

用户可以通过ATM系统查看自己的余额。我们可以编写以下方法来实现：

def check_balance(self):
    sql = 'SELECT balance FROM users WHERE name = ?'
    data = self.cursor.execute(sql, (self.current_user,))
    row = data.fetchone()
    if row:
        print(f'当前账户余额为 {row[0]}')
    else:
        print('获取余额失败')

3.4 存款和取款功能

用户可以通过ATM系统进行存款和取款操作。我们可以编写以下两个方法来实现：

def deposit(self, amount):
    sql = 'UPDATE users SET balance = balance + ? WHERE name = ?'
    self.cursor.execute(sql, (amount, self.current_user,))
    self.connection.commit()
    print(f'已成功存入 {amount} 元')

def withdraw(self, amount):
    sql = 'UPDATE users SET balance = balance - ? WHERE name = ? AND balance >= ?'
    self.cursor.execute(sql, (amount, self.current_user, amount,))
    self.connection.commit()
    print(f'已成功取出 {amount} 元')

3.5 转账和还款功能

用户可以通过ATM系统进行转账和还款操作。我们可以编写以下两个方法来实现：

def transfer(self, amount, recipient):
    sql1 = 'UPDATE users SET balance = balance - ? WHERE name = ? AND balance >= ?'
    sql2 = 'UPDATE users SET balance = balance + ? WHERE name = ?'
    self.cursor.execute(sql1, (amount, self.current_user, amount,))
    self.cursor.execute(sql2, (amount, recipient,))
    self.connection.commit()
    print(f'向 {recipient} 成功转账 {amount} 元')

def repay(self, amount):
    sql = 'UPDATE users SET balance = balance + ? WHERE name = ?'
    self.cursor.execute(sql, (amount, self.current_user,))
    self.connection.commit()
    print(f'已成功还款 {amount} 元')

4. 示例说明

下面是两个示例，分别演示了存款和转账操作的实现：

示例1：存款操作

atm = ATM()
atm.login()
atm.check_balance()

amount = 500
atm.deposit(amount)
atm.check_balance()

输出结果：

请输入用户名：alice
请输入密码：123456
登录成功
当前账户余额为 0
已成功存入 500 元
当前账户余额为 500

示例2：转账操作

atm = ATM()
atm.login()
atm.check_balance()

amount = 300
recipient = 'bob'
atm.transfer(amount, recipient)
atm.check_balance()

输出结果：

请输入用户名：alice
请输入密码：123456
登录成功
当前账户余额为 1000
向 bob 成功转账 300 元
当前账户余额为 700

至此，我们就成功实现了Python版的ATM系统。