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 回滚