summaryrefslogtreecommitdiff
path: root/plugins/replace.py
diff options
context:
space:
mode:
Diffstat (limited to 'plugins/replace.py')
-rw-r--r--plugins/replace.py66
1 files changed, 66 insertions, 0 deletions
diff --git a/plugins/replace.py b/plugins/replace.py
new file mode 100644
index 00000000..d1fbac57
--- /dev/null
+++ b/plugins/replace.py
@@ -0,0 +1,66 @@
+"""
+Replace a pattern from a message you're about to send, by the result of a
+function. For example you can insert the current time in your sentence by
+writing %time% in it.
+"""
+
+from plugin import BasePlugin
+import tabs
+import datetime
+import random
+from sleekxmpp.xmlstream.stanzabase import JID
+
+class Plugin(BasePlugin):
+ def init(self):
+ self.patterns = {}
+ self.add_event_handler('conversation_say', self.replace_pattern)
+ self.add_event_handler('muc_say', self.replace_pattern)
+ self.add_event_handler('private_say', self.replace_pattern)
+ self.patterns['time'] = replace_time
+ self.patterns['date'] = replace_date
+ self.patterns['datetime'] = replace_datetime
+ self.patterns['random_nick'] = replace_random_user
+ self.patterns['dice'] = replace_dice
+
+ def replace_pattern(self, message, tab):
+ """
+ Look for a %*% pattern in the message and replace it by the result
+ of the corresponding function.
+ """
+ body = message['body']
+ for pattern in self.patterns:
+ new = body
+ while True:
+ # we don't use a replace on all occurence, otherwise the
+ # result would be the same for all occurence of the pattern
+ # and that's not desirable in some of them (for example the
+ # ones that provide random results)
+ new = body.replace('%%%s%%' % pattern,
+ self.patterns[pattern](message, tab), 1)
+ if new == body:
+ break
+ body = new
+ message['body'] = body
+
+
+def replace_time(message, tab):
+ return datetime.datetime.now().strftime("%X")
+
+def replace_date(message, tab):
+ return datetime.datetime.now().strftime("%x")
+
+def replace_datetime(message, tab):
+ return datetime.datetime.now().strftime("%c")
+
+def replace_random_user(message, tab):
+ if isinstance(tab, tabs.MucTab):
+ return random.choice(tab.users).nick
+ elif isinstance(tab, tabs.PrivateTab):
+ return random.choice([JID(tab.name).resource, tab.own_nick])
+ else isinstance(tab, tabs.ConversationTab):
+ # that doesn’t make any sense. By why use this pattern in a
+ # ConversationTab anyway?
+ return str(tab.name)
+
+def replace_dice(message, tab):
+ return str(random.randrange(1, 7))