从零开始的Python开发日记（6）：如何使用SQLAlchemy

SQLAlchemy ORM 使用教程

SQLAlchemy 是 Python 生态系统中非常流行的 SQL 工具包和对象关系映射（ORM）库。它提供了一种高效的方法来处理数据库操作，使开发者能够通过 Python 对象与数据库进行交互。本文将详细介绍 ORM 的定义和含义，Python 中最常用的 ORM，SQLAlchemy 的优点，并逐步展示如何使用 SQLAlchemy ORM 进行数据库操作。

什么是 ORM？

对象关系映射（Object Relational Mapping，简称 ORM）是一种用于将关系数据库中的数据映射到面向对象编程语言中的对象的技术。它通过将数据库表映射到类，将表中的行映射到类的实例，从而使开发者能够通过编程语言的对象来操作数据库，而不需要编写复杂的 SQL 语句。

ORM 的核心思想是：

• 表 映射到 类。
• 行 映射到 类的实例。
• 列 映射到 类的属性。

Python 中最常用的 ORM

在 Python 生态系统中，有多个流行的 ORM 库：

1. Django ORM：Django 框架自带的 ORM，功能强大且与 Django 无缝集成。
2. SQLAlchemy：功能强大且灵活性高的 ORM，支持多种数据库，适用于各种规模的项目。
3. Peewee：一个小巧且易于使用的 ORM，适用于小型项目。
4. Tortoise ORM：一个现代的异步 ORM，专为 Python 的 asyncio 设计。

为什么选择 SQLAlchemy？

SQLAlchemy 是一个功能强大的 ORM，具有以下优点：

• 灵活性：SQLAlchemy 提供了 ORM 和 Core（SQL 表达式语言）两种模式，适用于不同需求的开发者。
• 性能：通过延迟加载、批量操作等优化技术，SQLAlchemy 能够高效地进行数据库操作。
• 兼容性：支持多种数据库，包括 SQLite、PostgreSQL、MySQL、Oracle 等。
• 扩展性：可以通过插件和扩展来增强功能，满足复杂的需求。
• 社区和文档：拥有活跃的社区和详尽的文档，方便开发者查找和解决问题。

SQLAlchemy ORM 的使用

下面将逐步展示如何使用 SQLAlchemy ORM 进行数据库操作。

1. 安装 SQLAlchemy

首先，使用 pip 安装 SQLAlchemy：

pip install sqlalchemy

2. 创建数据库引擎

创建一个数据库引擎，用于连接数据库。这里我们使用 SQLite 数据库：

from sqlalchemy import create_engine

# 创建 SQLite 数据库引擎
engine = create_engine('sqlite:///example.db', echo=True)

echo=True 参数会在控制台输出生成的 SQL 语句，方便调试。

3. 定义模型

定义一个基类，并使用它来定义数据库表的模型：

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String

# 创建一个基类
Base = declarative_base()

# 定义 User 类来表示 users 表
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    age = Column(Integer)

4. 创建表

使用模型类来创建数据库表：

# 创建所有表
Base.metadata.create_all(engine)

5. 创建会话

使用 sessionmaker 创建一个会话，用于进行数据库操作：

from sqlalchemy.orm import sessionmaker

# 创建会话
Session = sessionmaker(bind=engine)
session = Session()

6. 添加数据

将数据添加到数据库中：

# 创建一个新用户
new_user = User(name='Alice', age=25)

# 添加到会话
session.add(new_user)

# 提交会话
session.commit()

7. 查询数据

从数据库中查询数据：

# 查询所有用户
users = session.query(User).all()
for user in users:
    print(user.name, user.age)

8. 更新数据

更新数据库中的数据：

# 查询用户
user_to_update = session.query(User).filter_by(name='Alice').first()

# 更新数据
user_to_update.age = 26
session.commit()

9. 删除数据

删除数据库中的数据：

# 查询用户
user_to_delete = session.query(User).filter_by(name='Alice').first()

# 删除数据
session.delete(user_to_delete)
session.commit()

结论

SQLAlchemy 是一个功能强大且灵活的 ORM 库，适用于各种规模的项目。通过本文的介绍，你应该能够理解 ORM 的基本概念，并掌握使用 SQLAlchemy 进行数据库操作的基本方法。SQLAlchemy 提供了丰富的功能和优化技术，使你能够高效地进行数据库开发。如果你对数据库操作有更多的需求，可以查阅 SQLAlchemy 的官方文档，进一步学习和探索。