如何在flask应用程序上根据状态码运行函数

from flask import Flask, abort
from werkzeug.exceptions import HTTPException

app = Flask(__name__)

def run_on_status_code(status_code):
    def decorator(func):
        def wrapper(*args, **kwargs):
            try:
                response = func(*args, **kwargs)
                return response
            except HTTPException as e:
                if e.code == status_code:
                    # 运行函数
                    return func(*args, **kwargs)
                else:
                    # 抛出异常
                    raise e
        return wrapper
    return decorator

@app.route('/')
@run_on_status_code(200)
def index():
    return 'Hello, World!'

@app.route('/error')
@run_on_status_code(404)
def error():
    abort(404)