remember: add s/from/to syntax for easier modifications

This commit is contained in:
Ryan Hitchman 2014-02-07 11:12:11 -08:00
parent f6fc544227
commit 791df9409d
1 changed files with 23 additions and 3 deletions

View File

@ -2,6 +2,9 @@
remember.py: written by Scaevolus 2010 remember.py: written by Scaevolus 2010
""" """
import string
import re
from util import hook from util import hook
@ -23,10 +26,12 @@ def get_memory(db, chan, word):
@hook.command @hook.command
@hook.command("r") @hook.command("r")
def remember(inp, nick='', chan='', db=None): def remember(inp, nick='', chan='', db=None):
".remember <word> [+]<data> -- maps word to data in the memory" ".remember <word> [+]<data> s/<before>/<after> -- maps word to data in the memory, or "
" does string replacement (not regex)"
db_init(db) db_init(db)
append = False append = False
replacement = False
try: try:
head, tail = inp.split(None, 1) head, tail = inp.split(None, 1)
@ -34,19 +39,32 @@ def remember(inp, nick='', chan='', db=None):
return remember.__doc__ return remember.__doc__
data = get_memory(db, chan, head) data = get_memory(db, chan, head)
_head, _tail = data.split(None, 1)
if tail[0] == '+': if tail[0] == '+':
append = True append = True
# ignore + symbol # ignore + symbol
new = tail[1:] new = tail[1:]
_head, _tail = data.split(None, 1)
# data is stored with the input so ignore it when re-adding it # 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 + ' '): if len(tail) > 1 and tail[1] in (string.punctuation + ' '):
tail = _tail + new tail = _tail + new
else: else:
tail = _tail + ' ' + new tail = _tail + ' ' + new
if len(tail) > 2 and tail[0] == 's' and tail[1] in string.punctuation:
args = tail.split(tail[1])
if len(args) == 4 and args[3] == '':
args = args[:-1]
if len(args) == 3:
replacement = True
_, src, dst = args
new_data = _tail.replace(src, dst)
if new_data == _tail:
return 'replacement left data unchanged'
tail = new_data
else:
return 'invalid replacement syntax -- try s$foo$bar instead?'
db.execute("replace into memory(chan, word, data, nick) values" db.execute("replace into memory(chan, word, data, nick) values"
" (?,lower(?),?,?)", (chan, head, head + ' ' + tail, nick)) " (?,lower(?),?,?)", (chan, head, head + ' ' + tail, nick))
db.commit() db.commit()
@ -54,6 +72,8 @@ def remember(inp, nick='', chan='', db=None):
if data: if data:
if append: if append:
return "appending %s to %s" % (new, data.replace('"', "''")) return "appending %s to %s" % (new, data.replace('"', "''"))
elif replacement:
return "replacing '%s' with '%s' in %s" % (src, dst, _tail)
else: else:
return 'forgetting "%s", remembering this instead.' % \ return 'forgetting "%s", remembering this instead.' % \
data.replace('"', "''") data.replace('"', "''")