python实现银行ATM功能

原创

mob649e815e9bc9 2024-10-18 10:37:36 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e815e9bc9的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python 实现银行 ATM 功能指南

在这篇文章中，我们将学习如何使用 Python 实现一个简单的银行 ATM (自动取款机) 功能。我们将分步骤介绍整个开发过程，详细阐述每一步需要的代码，并为你提供清晰的注释，帮助你理解每段代码的功能。为了帮助你更好地理解整个流程，我们还将使用甘特图和表格来展示任务安排和步骤。

开发流程概览

下面是实现 ATM 功能的一般开发流程：

步骤编号	步骤描述	预计完成时间
1	需求分析	1 天
2	系统设计	1 天
3	编写代码（基础功能）	2 天
4	编写代码（扩展功能）	2 天
5	测试与验证	1 天
6	文档编写	1 天
7	发布与维护	持续进行

gantt
    title 开发 ATM 功能的时间安排
    dateFormat  YYYY-MM-DD
    section 需求分析
    需求分析           :a1, 2023-10-01, 1d
    section 系统设计
    系统设计           :a2, 2023-10-02, 1d
    section 编写代码
    基础功能           :a3, 2023-10-03, 2d
    扩展功能           :a4, 2023-10-05, 2d
    section 测试与验证
    测试与验证         :a5, 2023-10-07, 1d
    section 文档编写
    文档编写           :a6, 2023-10-08, 1d
    section 发布与维护
    发布与维护         :after a6  , 1d

各步骤详解

1. 需求分析

在需求分析阶段，我们要明确 ATM 的基本功能。基本功能包括：

用户登录
查询余额
存款
取款
退出

2. 系统设计

为实现这些功能，我们将使用一个简单的类来存储用户的基本信息，包括用户名、密码、余额。

3. 编写代码（基础功能）

下面是 ATM 系统的基础功能代码。

class ATM:
    def __init__(self):
        # 初始化用户信息
        self.username = "user"
        self.password = "pass"
        self.balance = 1000  # 初始余额设为1000

    def authenticate(self, input_username, input_password):
        """
        验证用户登录信息。
        """
        if input_username == self.username and input_password == self.password:
            return True
        else:
            return False

    def check_balance(self):
        """
        查询余额。
        """
        return self.balance

    def deposit(self, amount):
        """
        存款。
        """
        if amount > 0:
            self.balance += amount
            return f"存款成功，当前余额: {self.balance}"
        else:
            return "存款金额需大于0"

    def withdraw(self, amount):
        """
        取款。
        """
        if amount > self.balance:
            return "余额不足"
        elif amount <= 0:
            return "取款金额需大于0"
        else:
            self.balance -= amount
            return f"取款成功，当前余额: {self.balance}"


# 示例代码，模拟用户操作
atm = ATM()
if atm.authenticate("user", "pass"):
    print(atm.check_balance())
    print(atm.deposit(500))
    print(atm.withdraw(300))
else:
    print("用户名或密码错误")

4. 编写代码（扩展功能）

我们可以为这个 ATM 系统添加更多的功能，比如限额，交易记录等。下面的代码展示了如何添加交易记录功能。

class ATMWithHistory:
    def __init__(self):
        self.username = "user"
        self.password = "pass"
        self.balance = 1000
        self.transaction_history = []  # 存储交易记录

    def log_transaction(self, transaction):
        """
        记录每一笔交易
        """
        self.transaction_history.append(transaction)

    def deposit(self, amount):
        if amount > 0:
            self.balance += amount
            self.log_transaction(f"存入: {amount}")  # 记录存款交易
            return f"存款成功，当前余额: {self.balance}"
        else:
            return "存款金额需大于0"

    def withdraw(self, amount):
        if amount > self.balance:
            return "余额不足"
        elif amount <= 0:
            return "取款金额需大于0"
        else:
            self.balance -= amount
            self.log_transaction(f"取出: {amount}")  # 记录取款交易
            return f"取款成功，当前余额: {self.balance}"

    def view_history(self):
        """
        显示交易记录。
        """
        return self.transaction_history

# 示例代码，模拟用户操作
atm_with_history = ATMWithHistory()
if atm_with_history.authenticate("user", "pass"):
    print(atm_with_history.check_balance())
    print(atm_with_history.deposit(500))
    print(atm_with_history.withdraw(300))
    print("交易记录:", atm_with_history.view_history())
else:
    print("用户名或密码错误")