remember plugin

bot.py

@@ -69,6 +69,7 @@ print 'Connecting to IRC'
 bot.irc = irc.irc(network, nick)
 bot.irc.join(channel)
 bot.commandprefix = '^(?:\.|'+nick+'[:,]*\s*)'
+bot.persist_dir = os.path.abspath('persist')
 
 print 'Running main loop'
 
@@ -84,10 +85,13 @@ class Input(object):
         self.host = host
         self.paraml = paraml
         self.msg = msg
+        if command == "PRIVMSG":
+            self.chan = paraml[0]
 
 class FakeBot(object):
     def __init__(self, bot, input, func):
         self.bot = bot
+        self.persist_dir = bot.persist_dir
         self.input = input
         self.msg = bot.irc.msg
         self.cmd = bot.irc.cmd

plugins/remember.py

@@ -0,0 +1,89 @@
+#!/usr/bin/env python
+"""
+remember.py: written by Scaevolus 2009
+"""
+
+from __future__ import with_statement
+
+import os
+import thread
+import collections
+import codecs
+
+import hook
+
+lock = thread.allocate_lock()
+memory = {}
+
+def load_memory(filename, mtimes={}):
+    if not os.path.exists(filename):
+        return {}
+    mtime = os.stat(filename).st_mtime
+    if mtimes.get(filename, 0) != mtime:
+        mtimes[filename] = mtime
+        return dict((x.split(None, 1)[0].lower(), x) for x in 
+                codecs.open(filename, 'r', 'utf-8'))
+
+def save_memory(filename, memory):
+    out = codecs.open(filename, 'w', 'utf-8')
+    out.write('\n'.join(sorted(memory.itervalues())))
+    out.flush()
+    out.close()
+
+def make_filename(dir, chan):
+    return os.path.join(dir, 'memory')
+
+@hook.command
+def remember(bot, input):
+    ".remember <word> <data> -- maps word to data in the memory"
+    with lock: 
+        filename = make_filename(bot.persist_dir, input.chan)
+        memory.setdefault(filename, load_memory(filename))
+        
+        try:
+            head, tail = input.inp.split(None, 1)
+        except ValueError:
+            return remember.__doc__
+        
+        tail = tail.strip()
+        low = head.lower()
+        if low not in memory[filename]:
+            bot.reply("done.")
+        else:
+            bot.reply('forgetting that "%s", remembering this instead.' % 
+                    memory[filename][low])
+        memory[filename][low] = input.inp.strip()
+        save_memory(filename, memory[filename])
+
+@hook.command
+def forget(bot, input): 
+    ".forget <word> -- forgets the mapping that word had"
+    with lock:
+        filename = make_filename(bot.persist_dir, input.chan)
+        memory.setdefault(filename, load_memory(filename))
+
+        if not input.inp.strip():
+            return forget.__doc__
+        
+        print input.inp
+        low = input.inp.strip().lower()
+        print repr(low)
+        if low not in memory[filename]:
+            return "I don't know about that."
+        if not hasattr(input, 'chan'):
+            return "I won't forget anything in private."
+        bot.say("Forgot that %s" % memory[filename][low])
+        del memory[filename][low]
+        save_memory(filename, memory[filename])
+
+@hook.command(hook='\?(.+)', prefix=False)
+def question(bot, input):
+    "?<word> -- shows what data is associated with word"
+    with lock:
+        filename = make_filename(bot.persist_dir, input.chan)
+        memory.setdefault(filename, load_memory(filename))
+
+        word = input.inp.split()[0].lower()
+        print memory[filename]
+        if word in memory[filename]:
+            bot.say("%s" % memory[filename][word])