# Copyright (C) 2009, 2010 Roman Zimbelmann # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see . import _curses from ranger.ext.iter_tools import flatten from ranger.shared import SettingsAware class CursesShortcuts(SettingsAware): """ This class defines shortcuts to faciliate operations with curses. color(*keys) -- sets the color associated with the keys from the current colorscheme. color_at(y, x, wid, *keys) -- sets the color at the given position color_reset() -- resets the color to the default addstr(*args) -- failsafe version of self.win.addstr(*args) """ def addstr(self, *args): try: self.win.addstr(*args) except (_curses.error, TypeError): pass def addnstr(self, *args): try: self.win.addnstr(*args) except (_curses.error, TypeError): pass def color(self, *keys): """Change the colors from now on.""" keys = flatten(keys) attr = self.settings.colorscheme.get_attr(*keys) try: self.win.attrset(attr) except _curses.error: pass def color_at(self, y, x, wid, *keys): """Change the colors at the specified position""" keys = flatten(keys) attr = self.settings.colorscheme.get_attr(*keys) try: self.win.chgat(y, x, wid, attr) except _curses.error: pass def color_reset(self): """Change the colors to the default colors""" CursesShortcuts.color(self, 'reset')