diff options
-rw-r--r-- | src/multiuserchat.py | 10 | ||||
-rw-r--r-- | src/tabs.py | 31 |
2 files changed, 34 insertions, 7 deletions
diff --git a/src/multiuserchat.py b/src/multiuserchat.py index 832b97a7..67fa32e6 100644 --- a/src/multiuserchat.py +++ b/src/multiuserchat.py @@ -67,13 +67,14 @@ def leave_groupchat(xmpp, jid, own_nick, msg): """ xmpp.plugin['xep_0045'].leaveMUC(jid, own_nick, msg) -def eject_user(xmpp, jid, nick, reason): +def set_user_role(xmpp, jid, nick, reason, role): """ - (try to) Eject an user from the room + (try to) Set the role of a MUC user + (role =) 'none': eject user) """ iq = xmpp.makeIqSet() query = ET.Element('{%s}query' % NS_MUC_ADMIN) - item = ET.Element('{%s}item' % NS_MUC_ADMIN, {'nick':nick, 'role':'none'}) + item = ET.Element('{%s}item' % NS_MUC_ADMIN, {'nick':nick, 'role':role}) if reason: reason_el = ET.Element('{%s}reason' % NS_MUC_ADMIN) reason_el.text = reason @@ -82,7 +83,6 @@ def eject_user(xmpp, jid, nick, reason): iq.append(query) iq['to'] = jid try: - iq.send() + return iq.send() except Exception as e: return e.iq - diff --git a/src/tabs.py b/src/tabs.py index 60d91777..3517c0c3 100644 --- a/src/tabs.py +++ b/src/tabs.py @@ -404,6 +404,9 @@ class MucTab(ChatTab): self.commands['ignore'] = (self.command_ignore, _("Usage: /ignore <nickname> \nIgnore: Ignore a specified nickname."), None) self.commands['unignore'] = (self.command_unignore, _("Usage: /unignore <nickname>\nUnignore: Remove the specified nickname from the ignore list."), self.completion_unignore) self.commands['kick'] = (self.command_kick, _("Usage: /kick <nick> [reason]\nKick: Kick the user with the specified nickname. You also can give an optional reason."), None) + self.commands['moderator'] = (self.command_moderator, _("Usage: /moderator <nick> [reason]\nModerator: Op the user with the specified nickname. You also can give an optional reason."), None) + self.commands['visitor'] = (self.command_visitor, _("Usage: /visitor <nick> [reason]\nVisitor: Mute the user with the specified nickname. You also can give an optional reason."), None) + self.commands['participant'] = (self.command_participant, _("Usage: /participant <nick> [reason]\nParticipant: Voice the user with the specified nickname. You also can give an optional reason."), None) self.commands['topic'] = (self.command_topic, _("Usage: /topic <subject>\nTopic: Change the subject of the room"), self.completion_topic) self.commands['query'] = (self.command_query, _('Usage: /query <nick> [message]\nQuery: Open a private conversation with <nick>. This nick has to be present in the room you\'re currently in. If you specified a message after the nickname, it will immediately be sent to this user'), None) self.commands['part'] = (self.command_part, _("Usage: /part [message]\nPart: disconnect from a room. You can specify an optional message."), None) @@ -623,9 +626,33 @@ class MucTab(ChatTab): """ /kick <nick> [reason] """ + self._command_change_role('none', 'kick', arg) + + def command_visitor(self, arg): + """ + /visitor <nick> [reason] + """ + self._command_change_role('visitor', 'visitor', arg) + + def command_participant(self, arg): + """ + /participant <nick> [reason] + """ + self._command_change_role('participant', 'participant', arg) + + def command_moderator(self, arg): + """ + /moderator <nick> [reason] + """ + self._command_change_role('moderator', 'moderator', arg) + + def _command_change_role(self, role, command, arg): + """ + Changes the role of the nick in args[0] + """ args = common.shell_split(arg) if len(args) < 1: - self.core.command_help('kick') + self.core.command_help('command') return nick = args[0] if len(args) >= 2: @@ -634,7 +661,7 @@ class MucTab(ChatTab): reason = '' if not self.get_room().joined: return - res = muc.eject_user(self.core.xmpp, self.get_name(), nick, reason) + res = muc.set_user_role(self.core.xmpp, self.get_name(), nick, reason, role) if res['type'] == 'error': self.core.room_error(res, self.get_name()) |