Python SQLAlchemy 连接MySQL的CURD操作 使用上下文管理 session

使用 contextmanager 来管理

from sqlalchemy import create_engine
from sqlalchemy.orm import scoped_session,sessionmaker
 
 
db_connect = "mysql+pymysql://root:password@localhost:3306/db_name?charset=utf8"
 
create=create_engine(db_connect)
SessionType=scoped_session(sessionmaker(bind=create,expire_on_commit=False))
 
def GetSession():
    return SessionType()
 
from contextlib import contextmanager
 
@contextmanager
def session_socpe():
    session=GetSession()
    try:
        yield session
        session.commit()
    except:
        session.rollback()
        raise
    finally:
        session.close()

　　

 用法
 查询

with session_socpe() as session:
    obj = session.query(Model类).filter(Model类.字段==参数).first()

　　

增加

with session_socpe() as session:
    obj = Model类(字段1=值1, 字段2=值2, 字段3=值3)
    session.add(obj)

　　

 删除

with session_socpe() as session:
　　obj = session.query(Model类).filter(Model类.字段==参数).delete()

　　

修改

with session_socpe() as session:
　　obj = session.query(Model类).filter(Model类.字段==参数).update({"修改字段":"修改值"})

　　

每次修改时，不用频繁地session.commit()

也方便了 万一出错后 的 rollback 回滚