First :D

plugins/factoids.py

@@ -0,0 +1,104 @@
+"""
+remember.py: written by Scaevolus 20101
+"""
+
+from util import hook
+import re
+
+
+def db_init(db):
+    db.execute("create table if not exists mem(word, data, nick,"
+               " primary key(word))")
+    db.commit()
+
+
+def get_memory(db, word):
+
+    row = db.execute("select data from mem where word=lower(?)", [word]).fetchone()
+    if row:
+        return row[0]
+    else:
+        return None
+
+
+@hook.regex(r'^\+ ?(.*)')
+@hook.command("r")
+def remember(inp, nick='', db=None, say=None, input=None, notice=None):
+    "+<word> [+]<data> -- maps word to data in the memory"
+    if input.nick not in input.bot.config["admins"]:
+        return
+    binp = inp.group(0)
+    bind = binp.replace('+', '', 1)
+    db_init(db)
+
+
+    append = False
+
+    try:
+        head, tail = bind.split(None, 1)
+    except ValueError:
+        return remember.__doc__
+
+    data = get_memory(db, head)
+
+    if tail[0] == '+':
+        append = True
+        # ignore + symbol
+        new = tail[1:]
+        _head, _tail = data.split(None, 1)
+        # data is stored with the input so ignore it when re-adding it
+        import string
+        if len(tail) > 1 and tail[1] in (string.punctuation + ' '):
+            tail = _tail + new
+        else:
+            tail = _tail + ' ' + new
+
+    db.execute("replace into mem(word, data, nick) values"
+               " (lower(?),?,?)", (head, tail, nick))
+    db.commit()
+
+    if data:
+        if append:
+            notice("Appending %s to %s" % (new, data.replace('"', "''")))
+        else:
+            notice('Forgetting existing data (%s), remembering this instead!' % \
+                    data.replace('"', "''"))
+            return
+    else:
+        notice('Remembered!')
+        return
+
+
+@hook.command
+def forget(inp, db=None):
+    "-<word> -- forgets the mapping that word had"
+
+    binp = inp.group(0)
+    bind = binp.replace('-', '', 1)
+    print bind
+
+    try:
+        head, tail = bind.split(None, 1)
+    except ValueError:
+        return remember.__doc__
+
+    db_init(db)
+    data = get_memory(db, binp)
+
+    if data:
+        db.execute("delete from mem where word=lower(?)",
+                   (inp))
+        db.commit()
+        return 'forgot `%s`' % data.replace('`', "'")
+    else:
+        return "I don't know about that."
+
+@hook.command("info")
+@hook.regex(r'^\? ?(.+)')
+def question(inp, say=None, db=None):
+    "?<word> -- shows what data is associated with word"
+    db_init(db)
+
+    data = get_memory(db, inp.group(1).strip())
+    if data:
+        say(data)