如何捕获在装饰程序中引发的错误？

Question

我正在使用flask auth，它提供了一些辅助装饰程序。我已经添加了下面所有的各种方法，但是我想问的问题是如何捕获授权的\u处理程序装饰程序抛出的任何问题。这是一个关于装饰师的一般性问题，但我认为一个真实的例子可能会有所帮助。

如果装修工爆炸了，我怎么能抓住它？

import os
import flask
import flask_oauth

CONSUMER_KEY = os.environ['CONSUMER_KEY']
CONSUMER_SECRET = os.environ['CONSUMER_SECRET']

oauth = flask_oauth.OAuth()
twitter = oauth.remote_app(
    'twitter',
    base_url='https://api.twitter.com/1/',
    request_token_url='https://api.twitter.com/oauth/request_token',
    access_token_url='https://api.twitter.com/oauth/access_token',
    authorize_url='https://api.twitter.com/oauth/authenticate',
    consumer_key=CONSUMER_KEY,
    consumer_secret=CONSUMER_SECRET
)

app = flask.Flask(__name__)

@app.route('/login')
def login():
    return twitter.authorize(
        callback=url_for(
            'oauth_authorized',
            next=request.args.get('next') or request.referrer or None)
    )

@app.route('/oauth-authorized')
# what happens if this raises an error?
@twitter.authorized_handler
def oauth_authorized(resp):
    print 'foo-bar'

Answer 1

执行函数定义。因此，假设引发的异常特定于该修饰符，您可以将函数定义（包括修饰符）包装在try/except中：

try:
    @app.route('/oauth-authorized')
    @twitter.authorized_handler
    def oauth_authorized(resp):
        print 'foo-bar'
except WhateverError as e:
    print "twitter.authorized_handler raised an error", e

当然，如果引发异常，这将使oauth\u authorized未定义。在您的情况下，这可能是可以的，因为您可能不希望它被路由。但是，如果这不是您想要的，您可以向块中添加一个虚拟定义，但块除外。

或者，因为装饰器只是函数（好吧，任何可调用的），而@符号只是函数调用的语法糖，所以您可以在try/除中包装authorized_handler装饰：

def oauth_authorized(resp):
    print 'foo-bar'
try:    # apply decorator
    oauth_authorized = twitter.authorized_handler(oauth_authorized)
except Exception as e:
    print "twitter.authorized_handler raised an error", e
else:   # no error decorating with authorized_handler, apply app.route
    oauth_authorized = app.route('/oauth-authorized')(oauth_authorized)

如果authorized_handler修饰失败，这将为您留下未修饰版本的函数，但它不会被路由。你甚至可以把上面的功能放在它自己的功能中，并把它作为装饰器使用！