Change Utils.color to take a string to wrap in color

1 files changed, 3 insertions, 3 deletions

diff --git a/Utils.py b/Utils.py
index e2da0c43..40445606 100644
--- a/Utils.py
+++ b/Utils.py
@@ -156,12 +156,12 @@ FONT_BOLD, FONT_ITALIC, FONT_UNDERLINE, FONT_INVERT = ("\x02", "\x1D",
     "\x1F", "\x16")
 FONT_COLOR, FONT_RESET = "\x03", "\x0F"
 
-def color(foreground, background=None):
+def color(s, foreground, background=None):
     foreground = str(foreground).zfill(2)
     if background:
         background = str(background).zfill(2)
-    return "%s%s%s" % (FONT_COLOR, foreground,
-        "" if not background else ",%s" % background)
+    return "%s%s%s%s%s" % (FONT_COLOR, foreground,
+        "" if not background else ",%s" % background, s, FONT_COLOR)
 
 def bold(s):
     return "%s%s%s" % (FONT_BOLD, s, FONT_BOLD)