SQLAlchemy 入门:Python 中的 SQL 工具包和 ORM

SQLAlchemy 是 Python 中一款非常流行的数据库工具包,它对底层的数据库操作提供了高层次的抽象。在本篇文章中,我们将介绍 SQLAlchemy 的两个主要组成部分:SQL 工具包 (SQL Toolkit) 和对象关系映射器 (Object-Relational Mapper, ORM) 的基本使用。

一、SQLAlchemy 的安装和设置

安装 SQLAlchemy 很简单,可以通过 pip 工具进行安装:

pip install sqlalchemy

接下来,我们需要设置一个 SQLAlchemy Engine。Engine 是 SQLAlchemy 中的一个核心接口,它为 SQL 数据库提供了一种统一的方式来与 Python 程序进行交互。以下是一个 SQLite 数据库的 Engine 设置示例:

from sqlalchemy import create_engine

# 创建一个内存中的 SQLite 数据库
engine = create_engine('sqlite:///:memory:', echo=True)

二、SQLAlchemy SQL 工具包的使用

SQLAlchemy 的 SQL 工具包提供了一种 Pythonic 的方式来生成和执行 SQL 语句。例如,我们可以使用以下代码来创建一个新的数据库表:

from sqlalchemy import Table, Column, Integer, String, MetaData

metadata = MetaData()

users = Table(
   'users', metadata, 
   Column('id', Integer, primary_key=True), 
   Column('name', String), 
   Column('email', String),
)

metadata.create_all(engine)

三、SQLAlchemy ORM 的使用

SQLAlchemy 的 ORM 允许我们将 Python 类映射到数据库表,使我们可以使用面向对象的方式来处理数据库。以下是一个简单的 ORM 示例:

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

Base = declarative_base()

class User(Base):
   __tablename__ = 'users'

   id = Column(Integer, primary_key=True)
   name = Column(String)
   email = Column(String)

Base.metadata.create_all(engine)

在上面的代码中,我们定义了一个 User 类,并将其映射到 users 表。然后,我们可以使用类和对象的方式来操作数据库:

from sqlalchemy.orm import Session

# 创建一个新的 Session
session = Session(engine)

# 创建一个新的 User 对象
new_user = User(name='John', email='john@example.com')

# 添加到 session
session.add(new_user)

# 提交(保存)更改
session.commit()

四、总结

在本篇文章中,我们介绍了 SQLAlchemy 的基础知识,包括如何安装和设置 SQLAlchemy,如何使用 SQLAlchemy 的 SQL 工具包和 ORM。SQLAlchemy 是一个强大的工具,它使处理数据库变得更简单、更 Pythonic。在接下来的学习中,你可以深入探索 SQLAlchemy 的高级特性,如事务管理、表关联等。