use 2 streams (input and output) instead of 1 in parse_from_memory()

The initial code was not well tested and according to Sergey it is not
working properly.

Signed-off-by: Sergey Satskiy <sergey.satskiy@gmail.com>
Signed-off-by: Olivier Matz <zer0@droids-corp.org>

diff --git a/diff2html.py b/diff2html.py
index 337667f..b361491 100644 (file)
--- a/diff2html.py
+++ b/diff2html.py
@@ -424,9 +424,11 @@ def main():
 
 def parse_from_memory(txt, exclude_headers, show_hunk_infos):
     " Parses diff from memory and returns a string with html "
-    stream = StringIO.StringIO(txt)
-    parse_input(stream, stream, exclude_headers, show_hunk_infos)
-    return stream.buf
+    input_stream = StringIO.StringIO(txt)
+    output_stream = StringIO.StringIO()
+    parse_input(input_stream, output_stream, exclude_headers, show_hunk_infos)
+    return output_stream.getvalue()
+
 
 if __name__ == "__main__":
     main()