From dd0649dbd5ba5463535a8a2cb4c2335a78e4fec6 Mon Sep 17 00:00:00 2001 From: Olivier Matz Date: Fri, 15 Jun 2012 15:36:25 +0200 Subject: [PATCH] allow to be called with strings instead of IO files Signed-off-by: Sergey Satskiy Signed-off-by: Olivier Matz --- diff2html.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/diff2html.py b/diff2html.py index d573d8f..337667f 100644 --- a/diff2html.py +++ b/diff2html.py @@ -33,7 +33,7 @@ # and display those directly. -import sys, re, htmlentitydefs, getopt +import sys, re, htmlentitydefs, getopt, StringIO # minimum line size, we add a zero-sized breakable space every # LINESIZE characters @@ -422,6 +422,11 @@ def main(): parse_input(input_file, output_file, exclude_headers, show_hunk_infos) +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 if __name__ == "__main__": main() -- 2.20.1