Resolves ISSUE #5

Added new method to index.py to fix an issue with chunked encoding and missing request bodies Signed-off-by: Stephen Sullivan <[email protected]>
openfaas · Nov 3, 2018 · a64bdac · a64bdac
1 parent 366e37d
commit a64bdac
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 0 deletions.
diff --git a/template/python27-flask/index.py b/template/python27-flask/index.py
@@ -6,6 +6,18 @@
 
 app = Flask(__name__)
 
+@app.before_request
+def fix_transfer_encoding():
+    """
+    Sets the "wsgi.input_terminated" environment flag, thus enabling
+    Werkzeug to pass chunked requests as streams.  The gunicorn server
+    should set this, but it's not yet been implemented.
+    """
+
+    transfer_encoding = request.headers.get("Transfer-Encoding", None)
+    if transfer_encoding == u"chunked":
+        request.environ["wsgi.input_terminated"] = True
+
 @app.route("/", defaults={"path": ""}, methods=["POST", "GET"])
 @app.route("/<path:path>", methods=["POST", "GET"])
 def main_route(path):

diff --git a/template/python3-flask-armhf/index.py b/template/python3-flask-armhf/index.py
@@ -8,6 +8,18 @@
 
 app = Flask(__name__)
 
+@app.before_request
+def fix_transfer_encoding():
+    """
+    Sets the "wsgi.input_terminated" environment flag, thus enabling
+    Werkzeug to pass chunked requests as streams.  The gunicorn server
+    should set this, but it's not yet been implemented.
+    """
+
+    transfer_encoding = request.headers.get("Transfer-Encoding", None)
+    if transfer_encoding == u"chunked":
+        request.environ["wsgi.input_terminated"] = True
+
 @app.route("/", defaults={"path": ""}, methods=["POST", "GET"])
 @app.route("/<path:path>", methods=["POST", "GET"])
 def main_route(path):

diff --git a/template/python3-flask/index.py b/template/python3-flask/index.py
@@ -8,6 +8,18 @@
 
 app = Flask(__name__)
 
+@app.before_request
+def fix_transfer_encoding():
+    """
+    Sets the "wsgi.input_terminated" environment flag, thus enabling
+    Werkzeug to pass chunked requests as streams.  The gunicorn server
+    should set this, but it's not yet been implemented.
+    """
+
+    transfer_encoding = request.headers.get("Transfer-Encoding", None)
+    if transfer_encoding == u"chunked":
+        request.environ["wsgi.input_terminated"] = True
+
 @app.route("/", defaults={"path": ""}, methods=["POST", "GET"])
 @app.route("/<path:path>", methods=["POST", "GET"])
 def main_route(path):