diff --git a/rbd-target-api.py b/rbd-target-api.py index ae59d04..f76b2c2 100644 --- a/rbd-target-api.py +++ b/rbd-target-api.py @@ -42,7 +42,13 @@ from gwcli.utils import (APIRequest, valid_gateway, valid_client, valid_credentials, get_remote_gateways, valid_snapshot_name, GatewayAPIError) -app = Flask(__name__) +class RBDAPIFlask(Flask): + def process_response(self, response): + response.headers['server'] = 'rbd-api' + return(response) + +app = RBDAPIFlask(__name__) + # workaround for https://github.com/pallets/flask/issues/2549 app.config['JSONIFY_PRETTYPRINT_REGULAR'] = False