Added an output class, which will be used for output control and

formatting

src/pylorax/output.py

@@ -0,0 +1,101 @@
+#
+# output.py
+# output control
+#
+# Copyright (C) 2009  Red Hat, Inc.
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+# Red Hat Author(s):  Martin Gracik <mgracik@redhat.com>
+#
+
+import sys
+
+
+# color codes
+C_DEFAULT       = "\x1b[39m"
+C_RESET         = "\x1b[0m"
+
+C_BOLD          = "\x1b[1m"
+C_UNDERLINE     = "\x1b[4m"
+
+C_BLACK         = "\x1b[0;30m"
+C_WHITE         = "\x1b[1;37m"
+C_RED           = "\x1b[0;31m"
+C_GREEN         = "\x1b[0;32m"
+C_BLUE          = "\x1b[0;34m"
+C_LIGHTRED      = "\x1b[1;31m"
+C_LIGHTGREEN    = "\x1b[1;32m"
+C_LIGHTBLUE     = "\x1b[1;34m"
+
+C_LIGHTGRAY     = "\x1b[0;37m"
+C_DARKGRAY      = "\x1b[1;30m"
+
+
+class Output(object):
+
+    def __init__(self, output=sys.stdout, colors=True, encoding=None, verbose=False):
+        self.output = output
+        self.is_flushable = hasattr(self.output, "flush")
+
+        self.colors = colors
+        if encoding is None:
+            encoding = "utf-8"
+        self.encoding = encoding
+
+        self.verbose = verbose
+
+    def write(self, s, color=C_RESET, bold=False, underline=False):
+        self.output.write(self.compose(s, color=color, bold=bold, underline=underline))
+
+        if self.is_flushable:
+            self.output.flush()
+
+    def compose(self, s, color=C_RESET, bold=False, underline=False):
+        s = s.encode(self.encoding)
+        
+        if self.colors:
+            if bold:
+                s = "%s%s" % (C_BOLD, s)
+            if underline:
+                s = "%s%s" % (C_UNDERLINE, s)
+            s = "%s%s%s" % (color, s, C_RESET)
+        
+        return s
+
+    def writeline(self, s, color=C_RESET, bold=False, underline=False):
+        self.write("%s\n" % (s,), color=color, bold=bold, underline=underline)
+
+    def banner(self, s):
+        self.writeline(s, color=C_BLUE, bold=True)
+
+    def header(self, s):
+        self.writeline(s, bold=True)
+
+    def info(self, s):
+        self.writeline(s)
+
+    def error(self, s):
+        self.writeline(s, color=C_RED, bold=True)
+
+    def debug(self, s):
+        if self.verbose:
+            self.writeline(s)
+
+
+def initialize(verbose=False):
+    stdout = Output(output=sys.stdout, verbose=verbose)
+    stderr = Output(output=sys.stderr, verbose=verbose)
+    
+    return stdout, stderr