diff options
| -rw-r--r-- | poezio/tabs/muctab.py | 77 |
1 files changed, 37 insertions, 40 deletions
diff --git a/poezio/tabs/muctab.py b/poezio/tabs/muctab.py index 584754e0..3515d58d 100644 --- a/poezio/tabs/muctab.py +++ b/poezio/tabs/muctab.py @@ -282,6 +282,40 @@ class MucTab(ChatTab): "\x19%s}The subject of the room is: \x19%s}%s %s" % (info_text, norm_text, self.topic, user_string)) + @refresh_wrapper.always + def recolor(self, random_colors=False): + """Recolor the current MUC users""" + deterministic = config.get_by_tabname('deterministic_nick_colors', self.name) + if deterministic: + for user in self.users: + if user is self.own_user: + continue + color = self.search_for_color(user.nick) + if color != '': + continue + user.set_deterministic_color() + return + # Sort the user list by last talked, to avoid color conflicts + # on active participants + compare_users = lambda x: x.last_talked + sorted_users = sorted(self.users, key=compare_users, reverse=True) + full_sorted_users = sorted_users[:] + # search our own user, to remove it from the list + # Also remove users whose color is fixed + for user in full_sorted_users: + color = self.search_for_color(user.nick) + if user is self.own_user: + sorted_users.remove(user) + elif color != '': + sorted_users.remove(user) + user.change_color(color, deterministic) + colors = list(get_theme().LIST_COLOR_NICKNAMES) + if random_colors: + random.shuffle(colors) + for i, user in enumerate(sorted_users): + user.color = colors[i % len(colors)] + self.text_win.rebuild_everything(self._text_buffer) + def on_input(self, key, raw): if not raw and key in self.key_func: @@ -1162,47 +1196,10 @@ class MucTab(ChatTab): def command_recolor(self, args): """ /recolor [random] - Re-assign color to the participants of the room + Re-assigns color to the participants of the room """ - deterministic = config.get_by_tabname('deterministic_nick_colors', self.name) - if deterministic: - for user in self.users: - if user.nick == self.own_nick: - continue - color = self.search_for_color(user.nick) - if color != '': - continue - user.set_deterministic_color() - if args[0] == 'random': - self.core.information('"random" was provided, but poezio is ' - 'configured to use deterministic colors', - 'Warning') - self.user_win.refresh(self.users) - self.input.refresh() - return - compare_users = lambda x: x.last_talked - users = list(self.users) - sorted_users = sorted(users, key=compare_users, reverse=True) - full_sorted_users = sorted_users[:] - # search our own user, to remove it from the list - # Also remove users whose color is fixed - for user in full_sorted_users: - color = self.search_for_color(user.nick) - if user.nick == self.own_nick: - sorted_users.remove(user) - user.color = get_theme().COLOR_OWN_NICK - elif color != '': - sorted_users.remove(user) - user.change_color(color, deterministic) - colors = list(get_theme().LIST_COLOR_NICKNAMES) - if args[0] == 'random': - random.shuffle(colors) - for i, user in enumerate(sorted_users): - user.color = colors[i % len(colors)] - self.text_win.rebuild_everything(self._text_buffer) - self.user_win.refresh(self.users) - self.text_win.refresh() - self.input.refresh() + random_colors = args[0] == 'random' + self.recolor(random_colors) @command_args_parser.quoted(2, 2, ['']) def command_color(self, args): |