h/plugins/remember.py

71 lines
1.7 KiB
Python
Raw Normal View History

2009-04-03 03:50:38 +00:00
"""
remember.py: written by Scaevolus 2010
2009-04-03 03:50:38 +00:00
"""
from util import hook
2010-03-01 02:32:41 +00:00
def db_init(db):
db.execute("create table if not exists memory(chan, word, data, nick,"
" primary key(chan, word))")
db.commit()
2009-04-03 03:50:38 +00:00
2010-03-01 02:32:41 +00:00
def get_memory(db, chan, word):
2010-08-24 22:14:55 +00:00
row = db.execute("select data from memory where chan=? and word=lower(?)",
(chan, word)).fetchone()
if row:
2010-02-23 02:01:27 +00:00
return row[0]
else:
return None
2010-03-01 02:32:41 +00:00
2009-04-03 03:50:38 +00:00
@hook.command
def remember(inp, nick='', chan='', db=None):
2009-04-03 03:50:38 +00:00
".remember <word> <data> -- maps word to data in the memory"
db_init(db)
try:
head, tail = inp.split(None, 1)
except ValueError:
return remember.__doc__
data = get_memory(db, chan, head)
db.execute("replace into memory(chan, word, data, nick) values"
" (?,lower(?),?,?)", (chan, head, head + ' ' + tail, nick))
db.commit()
if data:
2010-02-23 02:01:27 +00:00
return 'forgetting "%s", remembering this instead.' % \
data.replace('"', "''")
else:
return 'done.'
2010-03-01 02:32:41 +00:00
2009-04-03 03:50:38 +00:00
@hook.command
def forget(inp, chan='', db=None):
2009-04-03 03:50:38 +00:00
".forget <word> -- forgets the mapping that word had"
db_init(db)
data = get_memory(db, chan, inp)
if not chan.startswith('#'):
return "I won't forget anything in private."
2009-04-03 03:50:38 +00:00
if data:
db.execute("delete from memory where chan=? and word=lower(?)",
(chan, inp))
db.commit()
2010-08-10 19:38:09 +00:00
return 'forgot `%s`' % data.replace('`', "'")
else:
return "I don't know about that."
2010-03-01 02:32:41 +00:00
@hook.regex(r'^\? ?(.+)')
def question(inp, chan='', say=None, db=None):
2009-04-03 03:50:38 +00:00
"?<word> -- shows what data is associated with word"
db_init(db)
2009-04-03 03:50:38 +00:00
data = get_memory(db, chan, inp.group(1).strip())
if data:
say(data)