Revision dfe11bad qa/qa_utils.py
| b/qa/qa_utils.py | ||
|---|---|---|
| 42 | 42 |
""" |
| 43 | 43 |
global _INFO_SEQ, _WARNING_SEQ, _ERROR_SEQ, _RESET_SEQ |
| 44 | 44 |
|
| 45 |
# Don't use colours if stdout isn't a terminal |
|
| 46 |
if not sys.stdout.isatty(): |
|
| 47 |
return |
|
| 48 |
|
|
| 45 | 49 |
try: |
| 46 | 50 |
import curses |
| 47 | 51 |
except ImportError: |
| ... | ... | |
| 201 | 205 |
return instances |
| 202 | 206 |
|
| 203 | 207 |
|
| 204 |
def _PrintWithColor(text, seq): |
|
| 205 |
f = sys.stdout |
|
| 206 |
|
|
| 207 |
if not f.isatty(): |
|
| 208 |
seq = None |
|
| 209 |
|
|
| 210 |
if seq: |
|
| 211 |
f.write(seq) |
|
| 212 |
|
|
| 213 |
f.write(text) |
|
| 214 |
f.write("\n")
|
|
| 215 |
|
|
| 216 |
if seq: |
|
| 217 |
f.write(_RESET_SEQ) |
|
| 218 |
|
|
| 219 |
|
|
| 220 |
def PrintWarning(text): |
|
| 221 |
return _PrintWithColor(text, _WARNING_SEQ) |
|
| 222 |
|
|
| 223 |
|
|
| 224 |
def PrintError(text): |
|
| 225 |
return _PrintWithColor(text, _ERROR_SEQ) |
|
| 208 |
def _FormatWithColor(text, seq): |
|
| 209 |
if not seq: |
|
| 210 |
return text |
|
| 211 |
return "%s%s%s" % (seq, text, _RESET_SEQ) |
|
| 226 | 212 |
|
| 227 | 213 |
|
| 228 |
def PrintInfo(text): |
|
| 229 |
return _PrintWithColor(text, _INFO_SEQ) |
|
| 214 |
FormatWarning = lambda text: _FormatWithColor(text, _WARNING_SEQ) |
|
| 215 |
FormatError = lambda text: _FormatWithColor(text, _ERROR_SEQ) |
|
| 216 |
FormatInfo = lambda text: _FormatWithColor(text, _INFO_SEQ) |
|
Also available in: Unified diff