diff options
Diffstat (limited to 'examples/stream_modify.py')
-rw-r--r-- | examples/stream_modify.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/examples/stream_modify.py b/examples/stream_modify.py new file mode 100644 index 00000000..56d26e6d --- /dev/null +++ b/examples/stream_modify.py @@ -0,0 +1,22 @@ +""" +This inline script modifies a streamed response. +If you do not need streaming, see the modify_response_body example. +Be aware that content replacement isn't trivial: + - If the transfer encoding isn't chunked, you cannot simply change the content length. + - If you want to replace all occurences of "foobar", make sure to catch the cases + where one chunk ends with [...]foo" and the next starts with "bar[...]. +""" + + +def modify(chunks): + """ + chunks is a generator that can be used to iterate over all chunks. + Each chunk is a (prefix, content, suffix) tuple. + For example, in the case of chunked transfer encoding: ("3\r\n","foo","\r\n") + """ + for prefix, content, suffix in chunks: + yield prefix, content.replace("foo", "bar"), suffix + + +def responseheaders(context, flow): + flow.response.stream = modify
\ No newline at end of file |