26

我有一个使用 sqlalchemy 的具有许多小型数据库访问功能的 python 应用程序。我试图避免围绕这些函数使用大量样板会话处理代码。

我有很多看起来像这样的功能：

def get_ticket_history(Session, ticket_id):
    s = Session()
    try:
        rows = s.query(TicketHistory)\
                .filter(TicketHistory.ticket_fk==ticket_id)\
                .order_by(TicketHistory.id.desc()).all()
        s.commit()
        return rows
    except:
        s.rollback()
        raise
    finally:
        s.close()

我正在尝试重构这些功能，但不确定我是否有最好的方法。我目前拥有的最好的是以下内容：

def execute(Session, fn, *args, **kwargs):
    s = Session()
    try:
        ret = fn(s, *args, **kwargs)
        s.commit()
        return ret
    except:
        s.rollback()
        raise
    finally:
        s.close()

def get_ticket_history(self, ticket_id):
    def sql_fn(s):
        return s.query(TicketHistory)\
                .filter(TicketHistory.ticket_fk==ticket_id)\
                .order_by(TicketHistory.id.desc()).all()
    return execute(self.sentinel_session, sql_fn)

有没有更好或更惯用的方法来做到这一点？也许使用装饰器？

谢谢，乔恩

4

4 回答 4

46

SQLAlchemy 文档提供了一种使用上下文管理器执行此操作的可能方法。

http://docs.sqlalchemy.org/en/latest/orm/session_basics.html#when-do-i-construct-a-session-when-do-i-commit-it-and-when-do-i-关闭它

为了完整起见，在此处复制代码片段：

from contextlib import contextmanager

@contextmanager
def session_scope():
    """Provide a transactional scope around a series of operations."""
    session = Session()
    try:
        yield session
        session.commit()
    except:
        session.rollback()
        raise
    finally:
        session.close()

这session_scope可以干净地使用，而无需现在重复样板。

class ThingOne(object):
    def go(self, session):
        session.query(FooBar).update({"x": 5})

class ThingTwo(object):
    def go(self, session):
        session.query(Widget).update({"q": 18})

def run_my_program():
    with session_scope() as session:
        ThingOne().go(session)
        ThingTwo().go(session)
于 2015-04-22T17:49:33.050 回答
8

从 Sql alchemy 1.4 版开始： Session 可以用作上下文管理器，而无需使用外部辅助函数。

示例来自文档

Session = sessionmaker(engine)

with Session() as session:
    session.add(some_object)
    session.add(some_other_object)
    session.commit()

首先，提交事务并关闭会话，可以应用以下方法。会话 = sessionmaker（引擎）

with Session.begin() as session:
    session.add(some_object)
    session.add(some_other_object)
# commits transaction, closes session

文档： https ://docs.sqlalchemy.org/en/14/orm/session_api.html#sqlalchemy.orm.sessionmaker

于 2021-06-01T16:31:01.990 回答
0

morphyn 使用上下文管理器的建议很好。您可以通过将contextlib.contextmanager装饰器应用于与您的 first 非常相似的函数来制作这样的上下文管理器，将get_ticket_historytry 和 except 之间的代码替换为yield语句并将其重命名，例如transaction. PEP 343有一个与该名称几乎相同的示例。

然后，将该上下文管理器与 with 语句一起使用以重新实现get_ticket_history. 不过，看起来 SQLAlchemy 已经提供了该功能，作为方法begin：

http://docs.sqlalchemy.org/en/rel_0_8/orm/session.html#autocommit-mode

于 2013-02-10T19:17:20.300 回答
0

事务处理（开始、提交/回滚）使用with子句

with engine.begin() as connection:
    r1 = connection.execute(table1.select())
    connection.execute(table1.insert(), {"col1": 7, "col2": "this is some data"})

老问题，但我仍然偶然发现它，所以这里是文档中的相关链接： https ://docs.sqlalchemy.org/en/13/core/connections.html

于 2020-03-28T18:50:39.673 回答