From: José Fonseca <jfonseca@vmware.com>
Date: Fri, 1 May 2009 17:57:50 +0000 (+0100)
Subject: Allow to control color usage from command line.
X-Git-Url: https://git.cworth.org/git?a=commitdiff_plain;h=fabf515439712de141cd916d08d7b6bcbe1feaee;p=apitrace

Allow to control color usage from command line.
---

diff --git a/xml2txt.py b/xml2txt.py
index 8b40a55..3d29d27 100755
--- a/xml2txt.py
+++ b/xml2txt.py
@@ -20,6 +20,7 @@
 
 
 import sys
+import optparse
 import xml.parsers.expat
 
 
@@ -317,9 +318,20 @@ class TraceParser(XmlParser):
 
 
 def main():
-    formatter = AnsiFormatter()
+    parser = optparse.OptionParser(
+        usage="\n\t%prog [options] [file] ...")
+    parser.add_option(
+        '--color', '--colour',
+        type="choice", choices=('never', 'always', 'auto'), metavar='WHEN',
+        dest="color", default="auto",
+        help="coloring: never, always, or auto [default: %default]")
+    (options, args) = parser.parse_args(sys.argv[1:])
+
+    if options.color == 'always' or options.color == 'auto' and sys.stdout.isatty():
+        formatter = AnsiFormatter()
+    else:
+        formatter = Formatter()
     
-    args = sys.argv[1:]
     if args:
         for arg in args:
             if arg.endswith('.gz'):