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

[diff2html.git] / diff2html.py

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 "
 
 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()
 
 if __name__ == "__main__":
     main()