大量的 try 应该怎么复用

推荐学习书目

› Learn Python the Hard Way

Python Sites

› PyPI - Python Package Index

› http://diveintopython.org/toc/index.html

› Pocoo

值得关注的项目

› PyPy

› Celery

› Jinja2

› Read the Docs

› gevent

› pyenv

› virtualenv

› Stackless Python

› Beautiful Soup

› 结巴中文分词

› Green Unicorn

› Sentry

› Shovel

› Pyflakes

› pytest

Python 编程

› pep8 Checker

Styles

› PEP 8

› Google Python Style Guide

› Code Style from The Hitchhiker's Guide

This topic created in 1787 days ago, the information mentioned may be changed or developed.

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


# ...

try:

    session.add(sth)

    session.commit()

except Exception as e:

    session.rollback()

    raise e

我的一些想法

函数调用，感觉不是很好
装...装饰器
google 搜索到有人用 with 关键字实现，似乎还不错
在一个基类里面做一些强制措施

try

基类

llba

MIT

20 replies • 2021-07-19 10:13:07 +08:00

MintZX

Jul 18, 2021

2 4 都行啊

ericls

Jul 18, 2021

为什么 function 不好？

learningman

Jul 18, 2021

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

christopheredwar

Jul 18, 2021

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

xiaoxinxiaobai

Jul 18, 2021 via Android

放 exception handler 中统一处理

hsfzxjy

Jul 18, 2021 via Android

写成 contextmanager

plko345

Jul 18, 2021 via Android

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

plko345

Jul 18, 2021 via Android

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

plko345

Jul 18, 2021 via Android

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

BBCCBB

Jul 18, 2021

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

expkzb

Jul 18, 2021 via Android

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

ipwx

Jul 18, 2021

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

Jul 18, 2021

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

Jul 18, 2021 via Android

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

plko345

Jul 18, 2021

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

plko345

Jul 18, 2021

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

Trim21

Jul 18, 2021

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

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

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

ipwx

Jul 18, 2021

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

skys215

Jul 18, 2021

没写过 flask

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

vZexc0m

Jul 19, 2021

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