From c2b673308f5d8ef7a295d03d1a1fa72b8b947c12 Mon Sep 17 00:00:00 2001 From: Mathieu Pasquet Date: Thu, 28 Nov 2013 21:22:17 +0100 Subject: =?UTF-8?q?Add=20an=20=E2=80=9Cautocorrect=E2=80=9D=20plugin=C2=A0?= =?UTF-8?q?with=20a=20sed-like=20syntax?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit TODO (eventually): - correct words when using “word*” or “*word” using levenshtein distance - allow regular expressions for replacement --- plugins/autocorrect.py | 67 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 67 insertions(+) create mode 100644 plugins/autocorrect.py (limited to 'plugins') diff --git a/plugins/autocorrect.py b/plugins/autocorrect.py new file mode 100644 index 00000000..61795b79 --- /dev/null +++ b/plugins/autocorrect.py @@ -0,0 +1,67 @@ +""" +This plugin lets you perform simple replacements on the last message. + +Installation +------------ + +Load the plugin:: + + /load autocorrect + +Usage +----- + +.. note:: This plugin only performs *simple* replacements, not with + regular expressions, despite the syntax. Although it would be + possible, that would be even less useful. + +Once the plugin is loaded, any message matching the following regex:: + + ^s/(.+?)/(.*?)(/|/g)?$ + +will be interpreted as a replacement, and the substitution will be +applied to the last sent message. + +For example, if you sent the message:: + + This tab lists all public rooms on a MUC service. It is currently very limited but will be improved in the future. There currently is no way to search a room. + +And you now want to replace “MUC” with “multi-user chat”, you input:: + + s/MUC/multi-user chat + +And poezio will correct the message for you. +""" + +from plugin import BasePlugin +import re + +sed_re = re.compile('^s/(.+?)/(.*?)(/|/g)?$') + +class Plugin(BasePlugin): + def init(self): + self.api.add_event_handler('muc_say', self.sed_fix) + self.api.add_event_handler('conversation_say', self.sed_fix) + self.api.add_event_handler('private_say', self.sed_fix) + + def sed_fix(self, msg, tab): + if not tab.last_sent_message: + return + body = tab.last_sent_message['body'] + match = sed_re.match(msg['body']) + if not match: + return + remove, put, matchall = match.groups() + + replace_all = False + if matchall == '/g': + replace_all = True + + if replace_all: + new_body = body.replace(remove, put) + else: + new_body = body.replace(remove, put, 1) + + if body != new_body: + msg['body'] = new_body + msg['replace']['id'] = tab.last_sent_message['id'] -- cgit v1.2.3