with的一种用法(利用with清理资源与回退)

最新推荐文章于 2022-06-22 14:42:19 发布

原创最新推荐文章于 2022-06-22 14:42:19 发布 · 367 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python

Python 同时被 3 个专栏收录

49 篇文章

订阅专栏

骚操作

11 篇文章

订阅专栏

奇怪的知识又增加了

10 篇文章

订阅专栏

博客围绕Python展开，介绍了一种需求场景，即在对资源操作后执行其他动作，若该动作失败，需对之前资源操作回退。还给出示例，说明使用特定语句进行回退操作更优雅，无需每次调用都添加回退清理代码。

需求场景

在执行完对资源的操作后再执行一些动作，若此动作执行失败，需要对之前对资源的操作进行回退。

示例

在执行完do_something_to_db 函数后，需要执行after_do_db 函数，但是当after_do_db 函数执行错误，需要对do_something_to_db 函数之前所做的操作进行清理、回退
这时使用with语句可以比较方便的进行回退操作，相对于不加with的语句显得优雅许多不用每次调用after_do_db时都加try...except...回退清理代码

代码

class BaseException(Exception):
    message = "This is the base Exception"

    def __init__(self, **kw):
        self.msg = self.message % kw

    def __str__(self):
        return self.msg


class MyExcep(BaseException):
    message = ("error is: %(name)s, reason is %(reason)s")
    

def do_db(param1, param2):
    print("do db")
    print(param1, param2)
    return 123

def un_do_db(param1, param2):
    print("undo db")

def after_do_db(r):
    print(r)
    # return True
    raise Exception("Error")
    

@contextlib.contextmanager
def do_something_to_db(param1, param2):
    res = do_db(param1, param2)
    try:
        yield res
    except Exception:
        un_do_db(param1, param2)
        raise MyExcep(name="Error", reason="the db action is Error")

使用

# 使用with版本
with do_something_to_db('123', '345') as r:
    after_do_db(r)

# 不使用with版本
param1 = '123'
param2 = '345'
res = do_something_to_db(param1 , param2)
try:
    after_do_db(res)
except Exception as e:
    un_do_db(param1, param2)