Python如何连接数据库？SQL操作实战教程

import sqlite3

# 1. 建立连接：如果数据库文件不存在，它会自动创建
# 通常，更推荐使用上下文管理器（with语句），后面会提到
conn = sqlite3.connect('my_database.db')

# 2. 创建游标：游标是执行SQL命令的对象
cursor = conn.cursor()

try:
    # 3. 执行SQL：创建表（如果不存在）
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS users (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT NOT NULL,
            email TEXT UNIQUE NOT NULL
        )
    ''')

    # 插入数据，使用参数化查询防止SQL注入
    cursor.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('张三', 'zhangsan@example.com'))
    cursor.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('李四', 'lisi@example.com'))

    # 查询数据
    cursor.execute("SELECT id, name, email FROM users")
    rows = cursor.fetchall() # 获取所有结果
    print("所有用户:")
    for row in rows:
        print(row)

    # 更新数据
    cursor.execute("UPDATE users SET email = ? WHERE name = ?", ('zhangsan_new@example.com', '张三'))

    # 再次查询确认更新
    cursor.execute("SELECT id, name, email FROM users WHERE name = ?", ('张三',))
    print("\n更新后的张三:")
    print(cursor.fetchone()) # 获取一条结果

    # 删除数据
    cursor.execute("DELETE FROM users WHERE name = ?", ('李四',))

    # 提交事务：将更改永久保存到数据库
    conn.commit()
    print("\n数据已提交。")

except sqlite3.Error as e:
    print(f"数据库操作发生错误: {e}")
    # 发生错误时回滚事务，撤销所有未提交的更改
    conn.rollback()
    print("事务已回滚。")
finally:
    # 4. 关闭游标和连接：释放资源，非常重要
    if cursor:
        cursor.close()
    if conn:
        conn.close()
    print("数据库连接已关闭。")

# 错误示范：存在SQL注入风险，请勿在生产环境使用！
# user_input = "张三'; DROP TABLE users; --"
# cursor.execute(f"SELECT * FROM users WHERE name = '{user_input}'")

# 正确示范：使用参数化查询
# SQLite/psycopg2风格的参数占位符是问号 ?
user_input = "张三"
cursor.execute("SELECT * FROM users WHERE name = ?", (user_input,))

# 对于MySQL/PyMySQL，参数占位符通常是 %s
# cursor.execute("SELECT * FROM users WHERE name = %s", (user_input,))

import sqlite3

# 使用with语句，连接会在代码块结束时自动关闭（无论是否发生异常）
with sqlite3.connect('my_database.db') as conn:
    cursor = conn.cursor()
    cursor.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('王五', 'wangwu@example.com'))
    conn.commit()
    print("王五已添加。")
# 此时，conn和cursor都已自动关闭，无需手动调用close()

import sqlite3

conn = sqlite3.connect('my_database.db')
cursor = conn.cursor()

try:
    # 假设这是一个转账操作的简化示例
    # 从账户A扣款
    cursor.execute("UPDATE accounts SET balance = balance - ? WHERE id = ?", (100, 1))
    # 模拟一个可能失败的操作，比如账户B不存在或余额不足
    # raise ValueError("模拟错误：账户B操作失败")
    # 给账户B加款
    cursor.execute("UPDATE accounts SET balance = balance + ? WHERE id = ?", (100, 2))

    # 如果所有操作都成功，提交事务
    conn.commit()
    print("转账成功！")
except sqlite3.Error as e:
    print(f"转账失败: {e}")
    # 任何一步失败，回滚所有操作，保持数据一致性
    conn.rollback()
    print("事务已回滚。")
except ValueError as e: # 捕获模拟的业务逻辑错误
    print(f"业务逻辑错误: {e}")
    conn.rollback()
    print("事务已回滚。")
finally:
    cursor.close()
    conn.close()