64 lines
1.7 KiB
Python
64 lines
1.7 KiB
Python
|
from parliament import Context
|
||
|
|
from flask import Request
|
||
|
|
import json
|
||
|
|
|
||
|
|
|
||
|
|
# parse request body, json data or URL query parameters
|
||
|
|
def payload_print(req: Request) -> str:
|
||
|
|
if req.method == "POST":
|
||
|
|
if req.is_json:
|
||
|
|
return json.dumps(req.json) + "\n"
|
||
|
|
else:
|
||
|
|
# MultiDict needs some iteration
|
||
|
|
ret = "{"
|
||
|
|
|
||
|
|
for key in req.form.keys():
|
||
|
|
ret += '"' + key + '": "'+ req.form[key] + '", '
|
||
|
|
|
||
|
|
return ret[:-2] + "}\n" if len(ret) > 2 else "{}"
|
||
|
|
|
||
|
|
elif req.method == "GET":
|
||
|
|
# MultiDict needs some iteration
|
||
|
|
ret = "{"
|
||
|
|
|
||
|
|
for key in req.args.keys():
|
||
|
|
ret += '"' + key + '": "' + req.args[key] + '", '
|
||
|
|
|
||
|
|
return ret[:-2] + "}\n" if len(ret) > 2 else "{}"
|
||
|
|
|
||
|
|
|
||
|
|
# pretty print the request to stdout instantaneously
|
||
|
|
def pretty_print(req: Request) -> str:
|
||
|
|
ret = str(req.method) + ' ' + str(req.url) + ' ' + str(req.host) + '\n'
|
||
|
|
for (header, values) in req.headers:
|
||
|
|
ret += " " + str(header) + ": " + values + '\n'
|
||
|
|
|
||
|
|
if req.method == "POST":
|
||
|
|
ret += "Request body:\n"
|
||
|
|
ret += " " + payload_print(req) + '\n'
|
||
|
|
|
||
|
|
elif req.method == "GET":
|
||
|
|
ret += "URL Query String:\n"
|
||
|
|
ret += " " + payload_print(req) + '\n'
|
||
|
|
|
||
|
|
return ret
|
||
|
|
|
||
|
|
|
||
|
|
def main(context: Context):
|
||
|
|
"""
|
||
|
|
Function template
|
||
|
|
The context parameter contains the Flask request object and any
|
||
|
|
CloudEvent received with the request.
|
||
|
|
"""
|
||
|
|
|
||
|
|
# Add your business logic here
|
||
|
|
print("Received request")
|
||
|
|
|
||
|
|
if 'request' in context.keys():
|
||
|
|
ret = pretty_print(context.request)
|
||
|
|
print(ret, flush=True)
|
||
|
|
return payload_print(context.request), 200
|
||
|
|
else:
|
||
|
|
print("Empty request", flush=True)
|
||
|
|
return "{}", 200
|