Python retrying库retry_on_exception参数如何传递多个自定义参数
解决方案
retrying库的retry_on_exception参数要求传入的可调用对象默认仅接收异常对象作为唯一入参,要额外传入类/实例引用,可以根据你的参数是否随实例变化选择以下方案:
方案1:传入固定类引用
如果需要传入的类引用是固定值,不需要随实例变化,直接用functools.partial预绑定参数即可:
from functools import partial # 装饰器写法 @retry( stop_max_attempt_number=3, retry_on_exception=partial(retry_if_db_error_or_passwd_change, class_reference=你要传入的类对象) ) def add_request_to_db(self, req_id): with make_session(self.session_maker) as session: session.merge(RequestHolder(request_id=req_id)) session.commit()
partial会自动把预传入的类对象作为第二个参数传给判断函数,调用时仅接收异常作为第一个参数,完全符合retrying库的参数要求。
方案2:传入当前实例引用
如果需要传入的是当前类的实例(也就是方法的self,每个实例属性可能不同),因为装饰器在类定义阶段就会执行,此时还没有实例对象,不能直接绑定参数,推荐把重试逻辑封装到方法内部的嵌套函数中:
def add_request_to_db(self, req_id): # 嵌套函数在实例方法调用时才定义,可正常捕获当前self @retry( stop_max_attempt_number=3, retry_on_exception=lambda e: retry_if_db_error_or_passwd_change(e, self) ) def inner_op(): with make_session(self.session_maker) as session: session.merge(RequestHolder(request_id=req_id)) session.commit() return inner_op()
注意事项
你现有判断逻辑中的'Access denied ' in exception存在类型错误,异常对象不能直接判断字符串是否包含,需要改为'Access denied ' in str(exception)避免运行时报错。
内容的提问来源于stack exchange,提问作者darthsithius
相关产品推荐
相关产品推荐

