pg.py: set color.USE_COLOR depending on the --color argument.
authorW. Trevor King <wking@tremily.us>
Wed, 19 Sep 2012 17:57:57 +0000 (13:57 -0400)
committerW. Trevor King <wking@tremily.us>
Wed, 19 Sep 2012 18:04:16 +0000 (14:04 -0400)
Otherwise the course loading and other things that avoid use_color
keyword args will end up getting printed in color by default.

bin/pg.py

index 17964c0ef8f8e9f9f0526cff3eb545832dcb1edf..86db4a20efc55ed9cfa791633f677834a3d0f5d1 100755 (executable)
--- a/bin/pg.py
+++ b/bin/pg.py
@@ -31,6 +31,7 @@ import pgp_mime as _pgp_mime
 
 from pygrader import __version__
 from pygrader import LOG as _LOG
+from pygrader import color as _color
 from pygrader.email import test_smtp as _test_smtp
 from pygrader.email import Responder as _Responder
 from pygrader.mailpipe import mailpipe as _mailpipe
@@ -164,6 +165,7 @@ if __name__ == '__main__':
     if args.verbose:
         _LOG.setLevel(max(_logging.DEBUG, _LOG.level - 10*args.verbose))
         _pgp_mime.LOG.setLevel(_LOG.level)
+    _color.USE_COLOR = args.color
 
     config = _configparser.ConfigParser()
     config.read([