Python操作MySQL的两种方法详解

pip install pymysql

import pymysql
from pymysql.cursors import DictCursor 可选，用于获取字典格式的结果

1. 建立数据库连接

connection = pymysql.connect(

    host='localhost',      数据库服务器地址，本地可用 localhost 或 127.0.0.1

    user='your_username',  用户名

    password='your_password', 密码

    database='your_database_name', 要连接的数据库名

    charset='utf8mb4',     字符集，支持中文和表情符号推荐 utf8mb4

    cursorclass=DictCursor 可选：设置游标类型，返回字典而非元组

)

try:

    2. 创建一个游标对象，用于执行SQL语句

    with connection.cursor() as cursor:

        示例1: 执行一条 SELECT 查询

        sql = "SELECT `id`, `name` FROM `users` WHERE `email` = %s"

        cursor.execute(sql, ('user@example.com',)) 使用参数化查询，防止SQL注入！

        

        获取结果

        result = cursor.fetchone() 获取单条记录

        print(result) 如果是DictCursor，输出: {'id': 1, 'name': 'John'}

        如果用默认游标，输出: (1, 'John')

        

        获取所有记录

        results = cursor.fetchall()

        for row in results:

           print(row)

    示例2: 执行一条 INSERT 语句

    with connection.cursor() as cursor:

        sql = "INSERT INTO `users` (`name`, `email`) VALUES (%s, %s)"

        cursor.execute(sql, ('Alice', 'alice@example.com'))

    

    3. 提交事务！对于INSERT/UPDATE/DELETE等写操作，必须提交才会生效。

    connection.commit()

    示例3: 获取刚插入数据的主键ID

    with connection.cursor() as cursor:

        sql = "INSERT INTO `users` (`name`, `email`) VALUES (%s, %s)"

        cursor.execute(sql, ('Bob', 'bob@example.com'))

        connection.commit()

        print(f"新插入记录的ID是: {cursor.lastrowid}")

finally:

    4. 关闭连接，释放资源

    connection.close()

import pymysql

使用 with 语句自动管理连接和游标的关闭

try:

    with pymysql.connect(host='localhost', user='root', password='password', database='test') as conn:

        with conn.cursor() as cursor:

            cursor.execute("SELECT VERSION()")

            result = cursor.fetchone()

            print(f"Database version: {result[0]}")

        连接结束时，如果没有异常，会自动 commit(); 如果有异常，会自动 rollback()

except pymysql.Error as e:

    print(f"Database error: {e}")

pip install sqlalchemy pymysql

from sqlalchemy import create_engine, Column, Integer, String

from sqlalchemy.ext.declarative import declarative_base

from sqlalchemy.orm import sessionmaker

1. 定义数据库连接字符串

格式: dialect+driver://username:password@host:port/database

DB_URI = 'mysql+pymysql://username:password@localhost:3306/your_database?charset=utf8mb4'

2. 创建引擎 (Engine)，它是ORM和数据库的连接核心

engine = create_engine(DB_URI, echo=True) echo=True 会打印执行的SQL，调试时有用

3. 创建基类

Base = declarative_base()

4. 定义映射类（模型）

class User(Base):

    __tablename__ = 'users' 指定映射的表名

    id = Column(Integer, primary_key=True, autoincrement=True)

    name = Column(String(50), nullable=False)

    email = Column(String(100), unique=True)

5. 创建所有表（如果不存在）

Base.metadata.create_all(engine)

6. 创建会话工厂 (Session)

Session = sessionmaker(bind=engine)

7. 使用会话进行增删改查

新增数据

with Session() as session:

    new_user = User(name='Charlie', email='charlie@example.com')

    session.add(new_user)

    session.commit() 提交事务

    print(f"New user ID: {new_user.id}") 提交后，id自动赋值

查询数据

with Session() as session:

    查询所有用户

    users = session.query(User).all()

    for user in users:

        print(user.id, user.name, user.email)

    条件查询

    user = session.query(User).filter_by(name='Charlie').first()

    if user:

        print(f"Found user: {user.name}")

        pip install python-dotenv

        .env 文件

        DB_HOST=localhost

        DB_USER=root

        DB_PASSWORD=your_secure_password

        DB_NAME=my_app

        ```

        ```python

        app.py

        from dotenv import load_dotenv

        import os

        import pymysql

        load_dotenv() 加载 .env 文件中的环境变量

        connection = pymysql.connect(

            host=os.getenv('DB_HOST'),

            user=os.getenv('DB_USER'),

            password=os.getenv('DB_PASSWORD'),

            database=os.getenv('DB_NAME')

        )