當(dāng)你想打電話給朋友時(shí)句惯,你是否得用手機(jī)撥通他的號(hào)碼才能建立起一個(gè)會(huì)話?同樣的,你想和數(shù)據(jù)庫(kù)交談也得先通過 SQLAlchemy 建立一個(gè)會(huì)話诊霹。
你可以把 sessionmaker 想象成一個(gè)手機(jī),engine 當(dāng)做數(shù)據(jù)庫(kù)的號(hào)碼渣淳,撥通這個(gè)“號(hào)碼”我們就創(chuàng)建了一個(gè) Session 類脾还。
我們可以這樣建立一個(gè) 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
# 連接數(shù)據(jù)庫(kù)
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
# 創(chuàng)建表
Base.metadata.create_all(engine)
session 建立后就可以通過它來操作數(shù)據(jù)庫(kù)。
添加數(shù)據(jù)
添加一條數(shù)據(jù):
ed_user = User(name='ed', fullname='Ed Jones', password='edspassword')
session.add(ed_user)
session.commit()
添加多條數(shù)據(jù):
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
來檢查被修改的數(shù)據(jù):
>>> 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')>
])
修改數(shù)據(jù)
修改某條數(shù)據(jù):
our_user = session.query(User).filter_by(name='ed').first()
our_user.password = 'f8s7ccs'
session.commit()
使用 session.dirty
來檢查被修改的數(shù)據(jù):
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()
進(jìn)行回滾操作:
our_user = session.query(User).filter_by(id=1).first()
our_user.name = 'Edwardo'
session.rollback()
session.commit()
這時(shí)數(shù)據(jù)庫(kù)的數(shù)據(jù)沒有被修改入愧。