大量的 try 应该怎么复用

[plko345]

一个 flask web 应用，有很多数据库操作 method，大部分 commit 是如下，想问下大家会怎么去服用

# ...

try:

    session.add(sth)

    session.commit()

except Exception as e:

    session.rollback()

    raise e

我的一些想法

1. 函数调用，感觉不是很好

2. 装...装饰器

3. google 搜索到有人用 with 关键字实现，似乎还不错

4. 在一个基类里面做一些强制措施

[MintZX]

2 4 都行啊

[ericls]

为什么 function 不好？

[learningman]

with 最好了，自己重载一个 ContextManager

[christopheredwar]

是 sqlalchemy 麽，可以封装成一个函数，传 kwargs，往里传参数就好。

[xiaoxinxiaobai]

放 exception handler 中统一处理

[hsfzxjy]

写成 contextmanager

[plko345]

@ericls 额，我也没想的特别清楚，就觉得虽然也能用，但似乎有更合适的办法

[plko345]

@christopheredwar 是，你说的是第一种吗？

[plko345]

@xiaoxinxiaobai 要怎么做，能否举个例子

[BBCCBB]

with 或者 装饰器 ==, 可以搞成 java spring 事务这种方式.

[expkzb]

放到 function 里不是挺好的，也就 sth 一个参数，再抛出一个 e 就行了

[ipwx]

1 2 3 4 都行
----

抛砖引玉，1：

def auto_commit(session, f):
....try:
........ret = f()
....except Exception:
........session.rollback()
........raise
....else:
........session.commit()
........return ret

auto_commit(session, lambda: session.add(sth))


2:

from functools import wraps

def auto_commit(f):
....@wraps(f)
....def wrapper(session, *args, **kwargs):
........try:
............ret = f(session)
........except Exception:
............session.rollback()
............raise
........else:
............session.commit()
............return ret
....return wrapper


@auto_commit
def my_fn(session):
........session.add(sth)

my_fn(session)

[ipwx]

3:

from contextlib import contextmanager


@contextmanager
def auto_commit(session):
....try:
........yield session
....except Exception:
........session.rollback()
........raise
....else:
........session.commit()


with auto_commit(session):
....session.add(sth)

[Trim21]

你的 session 如果不是自己封装的话很可能已经支持 with 了

[plko345]

@ipwx 大佬你的 commit 在 else 中的话, 如果 commit 失败也能 rollback 吗?

[plko345]

@Trim21 没有封装, 就是原来的 session, 如果支持的话要怎么用, 我去看看文档

[Trim21]

@plko345 #16 不知道你用得什么 ORM 框架，sqlalchemy 的话是这样的


with session.begin():
....session.add(some_object)
....session.add(some_other_object)

with 后面的代码块抛出了 exception 的话会自动 rollback

[ipwx]

@plko345 啊对，你如果有这个需求。。。那就不要放 else 呗。是我想当然了

[skys215]

没写过 flask

不是只有事务才需要用 commit （ mysql 级的）吗？

[vZexc0m]

继承 SQLAlchemy 基类，重新一个 save 方法，保存时调用 db.save()即可。