当你想打电话给朋友时,你是否得用手机拨通他的号码才能建立起一个会话?同样的,你想和数据库交谈也得先通过 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([
<User(name='fred', fullname='Fred Flinstone', password='blah')>,
<User(name='wendy', fullname='Wendy Williams', password='foobar')>,
<User(name='mary', fullname='Mary Contrary', password='xxg527')>
])
修改数据
修改某条数据:
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([<User(name='ed', fullname='Ed Jones', password='f8s7ccs')>])
回滚
使用 session.rollback()
进行回滚操作:
our_user = session.query(User).filter_by(id=1).first()
our_user.name = 'Edwardo'
session.rollback()
session.commit()
这时数据库的数据没有被修改。