SQLAlchemy学习笔记(二)

创建会话

我们现在准备去和数据库进行交互。ORM操作数据库的工具就是session,当我们第一次设置应用程序,和create_engin()在同一个级别的,我们定义一个Session类,这个是一个工厂为我们产生Seesion对象。

>>> from sqlalchemy.orm import sessionmaker
>>> Session = sessionmaker(bind=engine)

我们可以发现我们的sessionmaker()方法有一个bind的参数。当你的应用在没有Engine时,在你定义你的模型级对象是只需要像下面这样定义:

>>>Session = sessionmaker()

在这之后,当你创建了你的引擎create_engine(),想让引擎连接到会话上,可以使用configure():

>>> Session.configure(bind=engine)

这个定制的Session类将创建绑定到我们数据库的新的Session对象。当调用sessionmaker时候也可以定义其他的事物特征,之后的内容会讲到。然后,每当需要与数据库进行对话时,都会实例化一个会话Session:

>>> session = Session()

上面我们的Session和我们的数据库引擎想关联,但是它没有打开任何的数据库连接。
当它第一次使用时,它从引擎维护的连接池检索连接,并保持这个款连接,直到我们提交所有更改或关闭会话对象。

添加、更新一个对象

为了持久化我们的User对象,我们将使用add()将它添加到会话中:

>>>yu_user = User(name="yu", fullname="yuziyong", password="123")
>>> session.add(yu_user)

此刻,我们可以说这个实例是持久化的,
没有任何SQL语句执行,并且对象尚未由数据库中的行表示。这个Session将会在需要时发出SQL持久化yuziyong,使用一个称之为刷新的过程。如果我们查找数据库中的"yuziyong",所有持久化的信息将首先被刷新,并立即发出查询。
下面举个例子,我们创建一个新的查询,加载User实例,我们使用name进行过滤出名字为ricky的结果,
并指出我们只想获取完整列表中的第一个结果。返回一个等效于我们添加的用户实例:

>>> >>> our_user = session.query(User).filter_by(name="yu").first()
2017-03-08 22:31:13,130 INFO sqlalchemy.engine.base.Engine INSERT INTO users (name, fullname, password) VALUES (%(name)s, %(fullname)s, %(password)s) RETURNING users.id
2017-03-08 22:31:13,130 INFO sqlalchemy.engine.base.Engine {'fullname': 'yuziyong', 'password': '123', 'name': 'yu'}
2017-03-08 22:31:13,130 INFO sqlalchemy.engine.base.Engine SELECT users.id AS users_id, users.name AS users_name, users.fullname AS users_fullname, users.password AS users_password
FROM users
WHERE users.name = %(name_1)s
 LIMIT %(param_1)s
2017-03-08 22:31:13,130 INFO sqlalchemy.engine.base.Engine {'name_1': 'yu', 'param_1': 1}

事实上,Session已经识别出返回的行与在其内部对象映射中已经表示的行是相同的行,所以我们实际上得到了与刚才添加的相同的实例。

>>> our_user is yu_user
True

我们还可以使用add_all()来一次往session中添加多个:

>>> session.add_all([
...     User(name='wendy', fullname='Wendy Williams', password='foobar'),
...     User(name='mary', fullname='Mary Contrary', password='xxg527'),
...     User(name='fred', fullname='Fred Flinstone', password='blah')])

现在我们决定改变yu的密码:

>>> yu_user.password='000'

Session会注意到变化,例如:它知道yu的密码被修改:

>>> session.dirty
IdentitySet([<User(name=yu, fullname=yuziyong, password=000)>])
>>>

现在三个新的对象正在等待:

>>> session.new  
IdentitySet([<User(name='wendy', fullname='Wendy Williams', password='foobar')>,
<User(name='mary', fullname='Mary Contrary', password='xxg527')>,
<User(name='fred', fullname='Fred Flinstone', password='blah')>])

我们告诉Session,我们想要对数据库发出所有剩余的更改并提交事务,使用session.commit()

>>> session.commit()

commit()清除数据库中剩余的任何更改,并提交事务,会话引用的连接资源现在返回到连接池。

事物回滚

由于会话在一个事务中工作,我们可以回滚更改。让我们做两个更改,我们将还原yu_user的用户名设置为ziyong.

>>> yu_user.name = 'ziyong'

然后我们添加一个新的非法的用户:

>>> fake_user = User(name='fakeuser', fullname='Invalid', password='12345')
>>> session.add(fake_user)

查询会话,我们可以看到它们被刷入当前事务:

>>> session.query(User).filter(User.name.in_(['Edwardo', 'fakeuser'])).all()
2017-03-08 22:53:58,535 INFO sqlalchemy.engine.base.Engine BEGIN (implicit)
2017-03-08 22:53:58,535 INFO sqlalchemy.engine.base.Engine SELECT users.id AS users_id, users.fullname AS users_fullname, users.password AS users_password
FROM users
WHERE users.id = %(param_1)s
2017-03-08 22:53:58,535 INFO sqlalchemy.engine.base.Engine {'param_1': 6}
2017-03-08 22:53:58,535 INFO sqlalchemy.engine.base.Engine UPDATE users SET name=%(name)s WHERE users.id = %(users_id)s
2017-03-08 22:53:58,535 INFO sqlalchemy.engine.base.Engine {'users_id': 6, 'name': 'ziyong'}
2017-03-08 22:53:58,535 INFO sqlalchemy.engine.base.Engine INSERT INTO users (name, fullname, password) VALUES (%(name)s, %(fullname)s, %(password)s) RETURNING users.id
2017-03-08 22:53:58,535 INFO sqlalchemy.engine.base.Engine {'fullname': 'Invalid', 'password': '12345', 'name': 'fakauser'}
2017-03-08 22:53:58,535 INFO sqlalchemy.engine.base.Engine SELECT users.id AS users_id, users.name AS users_name, users.fullname AS users_fullname, users.password AS users_password
FROM users
WHERE users.name IN (%(name_1)s, %(name_2)s)
2017-03-08 22:53:58,535 INFO sqlalchemy.engine.base.Engine {'name_2': 'fakeuser', 'name_1': 'Edwardo'}
[]

回滚,我们能看到yu_user的名字回滚为yu,并且刚添加的非法用户也冲会话中删除。

>>> session.rollback()
2017-03-08 22:57:44,322 INFO sqlalchemy.engine.base.Engine ROLLBACK
>>> yu_user.name
2017-03-08 22:57:51,135 INFO sqlalchemy.engine.base.Engine BEGIN (implicit)
2017-03-08 22:57:51,135 INFO sqlalchemy.engine.base.Engine SELECT users.id AS users_id, users.name AS users_name, users.fullname AS users_fullname, users.password AS users_password
FROM users
WHERE users.id = %(param_1)s
2017-03-08 22:57:51,135 INFO sqlalchemy.engine.base.Engine {'param_1': 6}
u'yu'
>>> fake_user in session
False
>>>

再次查询:

>>> session.query(User).filter(User.name.in_(['yu', 'fakeuser'])).all()
2017-03-08 23:01:47,980 INFO sqlalchemy.engine.base.Engine SELECT users.id AS users_id, users.name AS users_name, users.fullname AS users_fullname, users.password AS users_password
FROM users
WHERE users.name IN (%(name_1)s, %(name_2)s)
2017-03-08 23:01:47,980 INFO sqlalchemy.engine.base.Engine {'name_2': 'fakeuser', 'name_1': 'yu'}
[<User(name=yu, fullname=yuziyong, password=000)>]
待续。。。

推荐阅读更多精彩内容