SQLAlchemy —— session

当你想打电话给朋友时，你是否得用手机拨通他的号码才能建立起一个会话？同样的，你想和数据库交谈也得先通过 SQLAlchemy 建立一个会话。

你可以把 sessionmaker 想象成一个手机，engine 当做数据库的号码，拨通这个“号码”我们就创建了一个 Session 类。

我们可以这样建立一个 session：

from sqlalchemy.orm import sessionmaker

Session = sessionmaker(bind=engine)
session = Session()

完整例子：

from sqlalchemy import create_engine, MetaData
from sqlalchemy import Table, Column, Date, Integer, String, ForeignKey
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

# 连接数据库
engine = create_engine('sqlite:///school.db', echo=False)

Base = declarative_base()

Session = sessionmaker(bind=engine)
session = Session()

class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String(50))
    fullname = Column(String(50))
    password = Column(String(12))

    def __str__(self):
        return self.name

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

---

session 建立后就可以通过它来操作数据库。

添加数据

添加一条数据：

ed_user = User(name='ed', fullname='Ed Jones', password='edspassword')

session.add(ed_user)
session.commit()

添加多条数据：

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')
    ])

session.commit()

使用 session.dirty 来检查被修改的数据：

>>> session.new
IdentitySet([
    , 
    , 
    
    ])

修改数据

修改某条数据：

our_user = session.query(User).filter_by(name='ed').first()
our_user.password = 'f8s7ccs'
session.commit()

使用 session.dirty 来检查被修改的数据：

our_user = session.query(User).filter_by(name='ed').first()
our_user.password = 'f8s7ccs'

>>> session.dirty
IdentitySet([])

回滚

使用 session.rollback() 进行回滚操作：

our_user = session.query(User).filter_by(id=1).first()

our_user.name = 'Edwardo'
session.rollback()
session.commit()

这时数据库的数据没有被修改。