The Tor Browser / file comparison
comparison: js/src/tests/lib/terminal_win.py
js/src/tests/lib/terminal_win.py
- branch
- TOR_BUG_3246
- changeset 7
- 129ffea94266
equal
deleted
inserted
replaced
| -1:000000000000 | 0:f0f15330d136 |
|---|---|
| 1 """ | |
| 2 From Andre Burgaud's Blog, from the CTypes Wiki: | |
| 3 http://www.burgaud.com/bring-colors-to-the-windows-console-with-python/ | |
| 4 | |
| 5 Colors text in console mode application (win32). | |
| 6 Uses ctypes and Win32 methods SetConsoleTextAttribute and | |
| 7 GetConsoleScreenBufferInfo. | |
| 8 | |
| 9 $Id: color_console.py 534 2009-05-10 04:00:59Z andre $ | |
| 10 """ | |
| 11 | |
| 12 from ctypes import windll, Structure, c_short, c_ushort, byref | |
| 13 | |
| 14 SHORT = c_short | |
| 15 WORD = c_ushort | |
| 16 | |
| 17 class COORD(Structure): | |
| 18 """struct in wincon.h.""" | |
| 19 _fields_ = [ | |
| 20 ("X", SHORT), | |
| 21 ("Y", SHORT)] | |
| 22 | |
| 23 class SMALL_RECT(Structure): | |
| 24 """struct in wincon.h.""" | |
| 25 _fields_ = [ | |
| 26 ("Left", SHORT), | |
| 27 ("Top", SHORT), | |
| 28 ("Right", SHORT), | |
| 29 ("Bottom", SHORT)] | |
| 30 | |
| 31 class CONSOLE_SCREEN_BUFFER_INFO(Structure): | |
| 32 """struct in wincon.h.""" | |
| 33 _fields_ = [ | |
| 34 ("dwSize", COORD), | |
| 35 ("dwCursorPosition", COORD), | |
| 36 ("wAttributes", WORD), | |
| 37 ("srWindow", SMALL_RECT), | |
| 38 ("dwMaximumWindowSize", COORD)] | |
| 39 | |
| 40 # winbase.h | |
| 41 STD_INPUT_HANDLE = -10 | |
| 42 STD_OUTPUT_HANDLE = -11 | |
| 43 STD_ERROR_HANDLE = -12 | |
| 44 | |
| 45 # wincon.h | |
| 46 FOREGROUND_BLACK = 0x0000 | |
| 47 FOREGROUND_BLUE = 0x0001 | |
| 48 FOREGROUND_GREEN = 0x0002 | |
| 49 FOREGROUND_CYAN = 0x0003 | |
| 50 FOREGROUND_RED = 0x0004 | |
| 51 FOREGROUND_MAGENTA = 0x0005 | |
| 52 FOREGROUND_YELLOW = 0x0006 | |
| 53 FOREGROUND_GREY = 0x0007 | |
| 54 FOREGROUND_INTENSITY = 0x0008 # foreground color is intensified. | |
| 55 | |
| 56 BACKGROUND_BLACK = 0x0000 | |
| 57 BACKGROUND_BLUE = 0x0010 | |
| 58 BACKGROUND_GREEN = 0x0020 | |
| 59 BACKGROUND_CYAN = 0x0030 | |
| 60 BACKGROUND_RED = 0x0040 | |
| 61 BACKGROUND_MAGENTA = 0x0050 | |
| 62 BACKGROUND_YELLOW = 0x0060 | |
| 63 BACKGROUND_GREY = 0x0070 | |
| 64 BACKGROUND_INTENSITY = 0x0080 # background color is intensified. | |
| 65 | |
| 66 stdout_handle = windll.kernel32.GetStdHandle(STD_OUTPUT_HANDLE) | |
| 67 SetConsoleTextAttribute = windll.kernel32.SetConsoleTextAttribute | |
| 68 GetConsoleScreenBufferInfo = windll.kernel32.GetConsoleScreenBufferInfo | |
| 69 | |
| 70 def get_text_attr(): | |
| 71 csbi = CONSOLE_SCREEN_BUFFER_INFO() | |
| 72 GetConsoleScreenBufferInfo(stdout_handle, byref(csbi)) | |
| 73 return csbi.wAttributes | |
| 74 | |
| 75 DEFAULT_COLORS = get_text_attr() | |
| 76 | |
| 77 class Terminal(object): | |
| 78 COLOR = { | |
| 79 'black': 0x0000, | |
| 80 'blue': 0x0001, | |
| 81 'green': 0x0002, | |
| 82 'cyan': 0x0003, | |
| 83 'red': 0x0004, | |
| 84 'magenta': 0x0005, | |
| 85 'yellow': 0x0006, | |
| 86 'gray': 0x0007 | |
| 87 } | |
| 88 BRIGHT_INTENSITY = 0x0008 | |
| 89 BACKGROUND_SHIFT = 4 | |
| 90 | |
| 91 @classmethod | |
| 92 def set_color(cls, color): | |
| 93 """ | |
| 94 color: str - color definition string | |
| 95 """ | |
| 96 color_code = 0 | |
| 97 if color.startswith('bright'): | |
| 98 color_code |= cls.BRIGHT_INTENSITY | |
| 99 color = color[len('bright'):] | |
| 100 color_code |= Terminal.COLOR[color] | |
| 101 SetConsoleTextAttribute(stdout_handle, color_code) | |
| 102 | |
| 103 @classmethod | |
| 104 def reset_color(cls): | |
| 105 SetConsoleTextAttribute(stdout_handle, DEFAULT_COLORS) | |
| 106 | |
| 107 @classmethod | |
| 108 def clear_right(cls): | |
| 109 pass |
