Question

如何在Python中为多个函数重用异常处理代码？

我正在开发一个使用Stripe Python库的项目。 https://stripe.com/docs/api/python#errors

这是他们的文档中的一些示例代码。

try:
  # Use Stripe's bindings...
  pass
except stripe.error.CardError, e:
  # Since it's a decline, stripe.error.CardError will be caught
  body = e.json_body
  err  = body['error']

  print "Status is: %s" % e.http_status
  print "Type is: %s" % err['type']
  print "Code is: %s" % err['code']
  # param is '' in this case
  print "Param is: %s" % err['param']
  print "Message is: %s" % err['message']
except stripe.error.InvalidRequestError, e:
  # Invalid parameters were supplied to Stripe's API
  pass
except stripe.error.AuthenticationError, e:
  # Authentication with Stripe's API failed
  # (maybe you changed API keys recently)
  pass
except stripe.error.APIConnectionError, e:
  # Network communication with Stripe failed
  pass
except stripe.error.StripeError, e:
  # Display a very generic error to the user, and maybe send
  # yourself an email
  pass
except Exception, e:
  # Something else happened, completely unrelated to Stripe
  pass

我需要编写几个函数来对Stripe系统执行各种调用来处理我的事务。例如;检索令牌，创建客户，为卡充电等等。我是否必须在每个函数中重复try / except代码，或者有没有办法使try块的内容动态化？

我想在我的Flask视图代码中使用这些不同的函数作为条件，所以如果我能从每个函数中找回错误/成功消息，那也会有所帮助。

Answer 1

编写一个装饰器，在try块中调用装饰视图，并处理任何与Stripe相关的异常。

from functools import wraps

def handle_stripe(f):
    @wraps(f)
    def decorated(*args, **kwargs):
        try:
            return f(*args, **kwargs)
        except MyStripeException as e:
            return my_exception_response
        except OtherStripeException as e:
            return other_response

    return decorated

@app.route('/my_stripe_route')
@handle_stripe
def my_stripe_route():
    do_stripe_stuff()
    return my_response

如何在Python中重用多个函数的异常处理代码？

1 个答案: