#!/usr/bin/env python import sys from flask import Flask, request from functools import wraps app = Flask(__name__) is_teapot = sys.argv[-1] == '--teapot' available_additions = ['milk', 'chocolate'] app.brewing = None def method(fx): @wraps(fx) def wr(): if is_teapot: return '', 418 res = fx() extra_headers = { 'Additions-List': ';'.join(available_additions) } if len(res) > 2: res[-1].update(extra_headers) else: res = res + (extra_headers,) return res return wr @app.route('/', methods=['BREW'])