| Index: swarm_client/third_party/colorama/winterm.py
|
| ===================================================================
|
| --- swarm_client/third_party/colorama/winterm.py (revision 235167)
|
| +++ swarm_client/third_party/colorama/winterm.py (working copy)
|
| @@ -1,120 +0,0 @@
|
| -
|
| -from . import win32
|
| -
|
| -
|
| -# from wincon.h
|
| -class WinColor(object):
|
| - BLACK = 0
|
| - BLUE = 1
|
| - GREEN = 2
|
| - CYAN = 3
|
| - RED = 4
|
| - MAGENTA = 5
|
| - YELLOW = 6
|
| - GREY = 7
|
| -
|
| -# from wincon.h
|
| -class WinStyle(object):
|
| - NORMAL = 0x00 # dim text, dim background
|
| - BRIGHT = 0x08 # bright text, dim background
|
| -
|
| -
|
| -class WinTerm(object):
|
| -
|
| - def __init__(self):
|
| - self._default = win32.GetConsoleScreenBufferInfo(win32.STDOUT).wAttributes
|
| - self.set_attrs(self._default)
|
| - self._default_fore = self._fore
|
| - self._default_back = self._back
|
| - self._default_style = self._style
|
| -
|
| - def get_attrs(self):
|
| - return self._fore + self._back * 16 + self._style
|
| -
|
| - def set_attrs(self, value):
|
| - self._fore = value & 7
|
| - self._back = (value >> 4) & 7
|
| - self._style = value & WinStyle.BRIGHT
|
| -
|
| - def reset_all(self, on_stderr=None):
|
| - self.set_attrs(self._default)
|
| - self.set_console(attrs=self._default)
|
| -
|
| - def fore(self, fore=None, on_stderr=False):
|
| - if fore is None:
|
| - fore = self._default_fore
|
| - self._fore = fore
|
| - self.set_console(on_stderr=on_stderr)
|
| -
|
| - def back(self, back=None, on_stderr=False):
|
| - if back is None:
|
| - back = self._default_back
|
| - self._back = back
|
| - self.set_console(on_stderr=on_stderr)
|
| -
|
| - def style(self, style=None, on_stderr=False):
|
| - if style is None:
|
| - style = self._default_style
|
| - self._style = style
|
| - self.set_console(on_stderr=on_stderr)
|
| -
|
| - def set_console(self, attrs=None, on_stderr=False):
|
| - if attrs is None:
|
| - attrs = self.get_attrs()
|
| - handle = win32.STDOUT
|
| - if on_stderr:
|
| - handle = win32.STDERR
|
| - win32.SetConsoleTextAttribute(handle, attrs)
|
| -
|
| - def get_position(self, handle):
|
| - position = win32.GetConsoleScreenBufferInfo(handle).dwCursorPosition
|
| - # Because Windows coordinates are 0-based,
|
| - # and win32.SetConsoleCursorPosition expects 1-based.
|
| - position.X += 1
|
| - position.Y += 1
|
| - return position
|
| -
|
| - def set_cursor_position(self, position=None, on_stderr=False):
|
| - if position is None:
|
| - #I'm not currently tracking the position, so there is no default.
|
| - #position = self.get_position()
|
| - return
|
| - handle = win32.STDOUT
|
| - if on_stderr:
|
| - handle = win32.STDERR
|
| - win32.SetConsoleCursorPosition(handle, position)
|
| -
|
| - def cursor_up(self, num_rows=0, on_stderr=False):
|
| - if num_rows == 0:
|
| - return
|
| - handle = win32.STDOUT
|
| - if on_stderr:
|
| - handle = win32.STDERR
|
| - position = self.get_position(handle)
|
| - adjusted_position = (position.Y - num_rows, position.X)
|
| - self.set_cursor_position(adjusted_position, on_stderr)
|
| -
|
| - def erase_data(self, mode=0, on_stderr=False):
|
| - # 0 (or None) should clear from the cursor to the end of the screen.
|
| - # 1 should clear from the cursor to the beginning of the screen.
|
| - # 2 should clear the entire screen. (And maybe move cursor to (1,1)?)
|
| - #
|
| - # At the moment, I only support mode 2. From looking at the API, it
|
| - # should be possible to calculate a different number of bytes to clear,
|
| - # and to do so relative to the cursor position.
|
| - if mode[0] not in (2,):
|
| - return
|
| - handle = win32.STDOUT
|
| - if on_stderr:
|
| - handle = win32.STDERR
|
| - # here's where we'll home the cursor
|
| - coord_screen = win32.COORD(0,0)
|
| - csbi = win32.GetConsoleScreenBufferInfo(handle)
|
| - # get the number of character cells in the current buffer
|
| - dw_con_size = csbi.dwSize.X * csbi.dwSize.Y
|
| - # fill the entire screen with blanks
|
| - win32.FillConsoleOutputCharacter(handle, ord(' '), dw_con_size, coord_screen)
|
| - # now set the buffer's attributes accordingly
|
| - win32.FillConsoleOutputAttribute(handle, self.get_attrs(), dw_con_size, coord_screen );
|
| - # put the cursor at (0, 0)
|
| - win32.SetConsoleCursorPosition(handle, (coord_screen.X, coord_screen.Y))
|
|
|