Python 连接 MySQL 数据库基础操作与银行管理系统实战

3. 数据类型说明：char 与 varchar

• char(n)：长度固定。例如存储 'abc'，使用 char(10)，实际占用 10 个字节（包含填充的空字符）。适合存储长度固定的数据，如身份证号、手机号。
• varchar(n)：长度可变。例如存储 'abc'，使用 varchar(10)，只占 3 个字节 + 1 个长度标记字节。10 是最大值。适合存储长度不确定的文本。

三、项目实战：银行管理系统

本案例模拟一个简单的银行系统，实现以下功能：

1. 用户登录（账号密码匹配）
2. 余额查询
3. 存款
4. 取款
5. 退出系统

1. 数据库设计

创建数据库 bank，并在其中创建账户表 account。

预置数据：

• 001 | 123456 | 1000.00
• 002 | 456789 | 5000.00

2. 完整代码实现

import pymysql
from decimal import Decimal

# 全局变量用于存储当前登录用户信息
current_user = None

def connect_db():
    """建立数据库连接"""
    try:
        db = pymysql.connect(
            host='localhost',
            user='root',
            password='1234',
            charset='utf8',
            database='bank'
        )
        return db
    except Exception as e:
        print(f"连接失败：{e}")
        return None

def login(db):
    """用户登录验证"""
    global current_user
    z = input("请输入账号：")
    m = input("请输入密码：")

    if not z or not m:
        print("输入不能为空")
        return False

    try:
        cursor = db.cursor()
        # 使用参数化查询防止 SQL 注入
        sql = 'select * from account where account_id=%s and account_passwd=%s'
        cursor.execute(sql, (z, m))
        result = cursor.fetchone()

        if result:
            print("登录成功")
            current_user = {'id': result[0], 'money': result[2]}
            return True
        else:
            print("登录失败")
            return False
    except Exception as e:
        print(f"登录过程出错：{e}")
        return False

def query_balance(db):
    """查询余额"""
    if not current_user:
        print("请先登录")
        return
    
    try:
        cursor = db.cursor()
        sql = 'select money from account where account_id=%s'
        cursor.execute(sql, (current_user['id'],))
        res = cursor.fetchone()
        if res:
            print(f"当前余额：{res[0]}")
    except Exception as e:
        print(f"查询失败：{e}")

def deposit(db):
    """存款"""
    if not current_user:
        print("请先登录")
        return
    
    try:
        amount = float(input("请输入存款金额："))
        if amount <= 0:
            print("金额必须大于 0")
            return

        cursor = db.cursor()
        new_money = current_user['money'] + amount
        sql = 'update account set money=%s where account_id=%s'
        cursor.execute(sql, (new_money, current_user['id']))
        db.commit()
        current_user['money'] = new_money
        print("存款成功")
    except Exception as e:
        print(f"存款失败：{e}")
        db.rollback()

def withdraw(db):
    """取款"""
    if not current_user:
        print("请先登录")
        return
    
    try:
        amount = float(input("请输入取款金额："))
        if amount <= 0:
            print("金额必须大于 0")
            return
        if amount > current_user['money']:
            print("余额不足")
            return

        cursor = db.cursor()
        new_money = current_user['money'] - amount
        sql = 'update account set money=%s where account_id=%s'
        cursor.execute(sql, (new_money, current_user['id']))
        db.commit()
        current_user['money'] = new_money
        print("取款成功")
    except Exception as e:
        print(f"取款失败：{e}")
        db.rollback()

def main():
    """主程序入口"""
    db = connect_db()
    if not db:
        return

    while True:
        print("\n=== 银行管理系统 ===")
        print("1. 登录")
        print("2. 查询余额")
        print("3. 存款")
        print("4. 取款")
        print("5. 退出")
        choice = input("请选择功能：")

        if choice == '1':
            if login(db):
                continue
        elif choice == '2':
            query_balance(db)
        elif choice == '3':
            deposit(db)
        elif choice == '4':
            withdraw(db)
        elif choice == '5':
            print("感谢使用，再见！")
            break
        else:
            print("无效选择")

    finally:
        db.close()

if __name__ == '__main__':
    main()

四、最佳实践建议

1. 安全性：生产环境中不要硬编码密码，建议使用环境变量或配置文件管理敏感信息。
2. SQL 注入：始终使用参数化查询（如 %s），避免直接拼接字符串。
3. 事务控制：涉及资金变动的操作（存/取）必须开启事务，并在异常时回滚 (rollback)。
4. 资源管理：尽量使用上下文管理器 (with 语句) 自动管理连接和游标的关闭。
5. 异常处理：捕获具体异常类型，避免吞掉所有错误信息，便于调试。

通过上述示例，您可以掌握 Python 操作 MySQL 的核心流程，并能在此基础上扩展更复杂的企业级应用。