python 中使用 SQLAlchemy

发表于 标签 ，

数据库表是一个二维表，包含多行多列。通过 python 获取数据库中的内容时，可以用一个 list 表示获取的多行记录，每一个元素的类型是 tuple，表示一行记录，比如，包含 id 和 name 的 user 表：

1
2
3
4
5
[
    ('1', 'Michael'),
    ('2', 'Bob'),
    ('3', 'Adam')
]

通过 ORM 技术：Object-Relational Mapping，可以把关系数据库的表结构映射到对象上。如：

1
2
3
4
5
6
7
8
9
10
class User(object):
    def __init__(self, id, name):
        self.id = id
        self.name = name

[
    User('1', 'Michael'),
    User('2', 'Bob'),
    User('3', 'Adam')
]

ORM 框架就是用来完成这种装换的，在 Java 中最常用的是 Hibernate，而在 Python 中，最有名的 ORM 框架是 SQLAlchemy。SQLAlchemy 的简单使用如下

创建表

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# 导入:
from sqlalchemy import Column, String, create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base

# 创建对象的基类:
Base = declarative_base()

# 定义User对象:
class User(Base):
    # 表的名字:
    __tablename__ = 'user'

    # 表的结构:
    id = Column(String(20), primary_key=True)
    name = Column(String(20))

# 初始化数据库连接:
engine = create_engine('mysql+mysqlconnector://root:password@localhost:3306/test')
# 创建DBSession类型:
DBSession = sessionmaker(bind=engine)

create_engine() 用来初始化数据库连接。SQLAlchemy 用一个字符串表示连接信息：

数据库类型+数据库驱动名称://用户名:口令@机器地址:端口号/数据库名

插入记录

由于有了 ORM，我们向数据库表中添加一行记录，可以视为添加一个 User 对象：

1
2
3
4
5
6
7
8
9
10
# 创建session对象:
session = DBSession()
# 创建新User对象:
new_user = User(id='5', name='Bob')
# 添加到session:
session.add(new_user)
# 提交即保存到数据库:
session.commit()
# 关闭session:
session.close()

session 对象可视为当前数据库连接。关键是获取 session，然后把对象添加到 session，最后提交并关闭。

查询记录

通过 ORM 查询出来的可以不再是 tuple，而是 User 对象。SQLAlchemy 提供的查询接口如下：

1
2
3
4
5
6
7
8
9
# 创建Session:
session = DBSession()
# 创建Query查询，filter是where条件，最后调用one()返回第一行，如果调用all()则返回所有行:
user = session.query(User).filter(User.id=='5').one()
# 打印类型和对象的name属性:
print 'type:', type(user)
print 'name:', user.name
# 关闭Session:
session.close()

运行结果如下：

1
2
type: <class '__main__.User'>
name: Bob

关 SQLAlchemy 更详细的用法可参考官方文档

参考：http://www.liaoxuefeng.com/