h/plugins/quote.py

88 lines
2.6 KiB
Python
Raw Normal View History

2010-01-22 09:56:20 +00:00
import random
import re
import time
from util import hook
2010-02-02 04:42:34 +00:00
def add_quote(db, chan, nick, add_nick, msg):
db.execute('''insert or fail into quote (chan, nick, add_nick,
2010-03-01 02:32:41 +00:00
msg, time) values(?,?,?,?,?)''',
(chan, nick, add_nick, msg, time.time()))
2010-02-02 04:42:34 +00:00
db.commit()
2010-01-22 09:56:20 +00:00
2010-03-01 02:32:41 +00:00
2010-02-02 04:42:34 +00:00
def get_quotes_by_nick(db, chan, nick):
return db.execute("select time, nick, msg from quote where deleted!=1 "
"and chan=? and lower(nick)=lower(?) order by time",
(chan, nick)).fetchall()
2010-03-01 02:32:41 +00:00
2010-02-02 04:42:34 +00:00
def get_quotes_by_chan(db, chan):
return db.execute("select time, nick, msg from quote where deleted!=1 "
"and chan=? order by time", (chan,)).fetchall()
def format_quote(q, num, n_quotes):
ctime, nick, msg = q
2010-03-01 02:32:41 +00:00
return "[%d/%d] %s <%s> %s" % (num, n_quotes,
time.strftime("%Y-%m-%d", time.gmtime(ctime)), nick, msg)
2010-03-01 02:32:41 +00:00
2010-01-22 09:56:20 +00:00
@hook.command('q')
@hook.command
2010-02-02 04:42:34 +00:00
def quote(inp, nick='', chan='', db=None):
".q/.quote <nick|#chan> [#n]/.quote add <nick> <msg> -- gets " \
"random or [#n]th quote by <nick> or from <#chan>/adds quote"
2010-01-22 09:56:20 +00:00
2010-02-02 04:42:34 +00:00
db.execute("create table if not exists quote"
"(chan, nick, add_nick, msg, time real, deleted default 0, "
"primary key (chan, nick, msg))")
2010-02-02 04:42:34 +00:00
db.commit()
2010-01-22 09:56:20 +00:00
2010-04-25 21:40:59 +00:00
add = re.match(r"add[^\w@]+(\S+?)>?\s+(.*)", inp, re.I)
retrieve = re.match(r"(\S+)(?:\s+#?(-?\d+))?$", inp)
if add:
quoted_nick, msg = add.groups()
try:
add_quote(db, chan, quoted_nick, nick, msg)
db.commit()
except db.IntegrityError:
return "message already stored, doing nothing."
return "quote added."
elif retrieve:
select, num = retrieve.groups()
by_chan = False
if select.startswith('#'):
by_chan = True
quotes = get_quotes_by_chan(db, select)
else:
quotes = get_quotes_by_nick(db, chan, select)
2010-01-22 09:56:20 +00:00
n_quotes = len(quotes)
2010-01-22 09:56:20 +00:00
if not n_quotes:
return "no quotes found"
2010-01-22 09:56:20 +00:00
if num:
num = int(num)
2010-01-22 09:56:20 +00:00
if num:
if num > n_quotes or (num < 0 and num < -n_quotes):
return "I only have %d quote%s for %s" % (n_quotes,
('s', '')[n_quotes == 1], select)
elif num < 0:
selected_quote = quotes[num]
num = n_quotes + num + 1
2010-01-22 09:56:20 +00:00
else:
selected_quote = quotes[num - 1]
else:
num = random.randint(1, n_quotes)
selected_quote = quotes[num - 1]
return format_quote(selected_quote, num, n_quotes)
else:
return quote.__doc__