Allow to pass writer to render_html (for saving to string etc.)

Author: abitrolly

Diff for: lib/panela/panela_colors.py

@@ -636,7 +636,7 @@ def parse(self):
             # end of line
             self.data.append([x, y, None, None, None])
 
-    def render_html(self, colorize=True):
+    def render_html(self, colorize=True, writer=sys.stdout):
 
         prev_style = (None, None)
 
@@ -645,30 +645,30 @@ def render_html(self, colorize=True):
             if char == None:  # end of line
                 if colorize:
                     if prev_style != (None, None):
-                        sys.stdout.write("</span>")
+                        writer.write("</span>")
                     prev_style = (None, None)
-                sys.stdout.write("\n")
+                writer.write("\n")
             else:
                 if colorize:
                     if style != prev_style:
                         if prev_style != (None, None):
-                            sys.stdout.write("</span>")
+                            writer.write("</span>")
                         if style != (None, None):
                             fore = style[0] if style[0] else 'unset'
                             back = style[1] if style[1] else 'unset'
-                            sys.stdout.write("<span style=\"" +
+                            writer.write("<span style=\"" +
                                     "color: %s; " % fore +
                                     "background-color: %s\">" % back)
                         prev_style = style
 
                 if char == '<':
-                    sys.stdout.write("&lt;")
+                    writer.write("&lt;")
                 elif char == '>':
-                    sys.stdout.write("&gt;")
+                    writer.write("&gt;")
                 elif char == '&':
-                    sys.stdout.write("&amp;")
+                    writer.write("&amp;")
                 else:
-                    sys.stdout.write(char)
+                    writer.write(char)
 
 
     def apply_mask(self):