118 lines
3.4 KiB
Python
Executable file
118 lines
3.4 KiB
Python
Executable file
""" tell.py: written by sklnd in July 2009
|
|
2010.01.25 - modified by Scaevolus"""
|
|
|
|
import time
|
|
import re
|
|
|
|
from util import hook, timesince
|
|
|
|
|
|
def db_init(db):
|
|
"""check to see that our db has the tell table and return a dbection."""
|
|
db.execute("create table if not exists tell"
|
|
"(user_to, user_from, message, chan, time,"
|
|
"primary key(user_to, message))")
|
|
db.commit()
|
|
|
|
return db
|
|
|
|
|
|
def get_tells(db, user_to):
|
|
return db.execute("select user_from, message, time, chan from tell where"
|
|
" user_to=lower(?) order by time",
|
|
(user_to.lower(),)).fetchall()
|
|
|
|
|
|
@hook.singlethread
|
|
@hook.event('PRIVMSG')
|
|
def tellinput(paraml, input=None, notice=None, db=None, bot=None, nick=None, conn=None):
|
|
if 'showtells' in input.msg.lower():
|
|
return
|
|
|
|
db_init(db)
|
|
|
|
tells = get_tells(db, nick)
|
|
|
|
if tells:
|
|
user_from, message, time, chan = tells[0]
|
|
reltime = timesince.timesince(time)
|
|
|
|
reply = "{} sent you a message {} ago from {}: {}".format(user_from, reltime, chan,
|
|
message)
|
|
if len(tells) > 1:
|
|
reply += " (+{} more, {}showtells to view)".format(len(tells) - 1, conn.conf["command_prefix"])
|
|
|
|
db.execute("delete from tell where user_to=lower(?) and message=?",
|
|
(nick, message))
|
|
db.commit()
|
|
notice(reply)
|
|
|
|
|
|
@hook.command(autohelp=False)
|
|
def showtells(inp, nick='', chan='', notice=None, db=None):
|
|
"showtells -- View all pending tell messages (sent in a notice)."
|
|
|
|
db_init(db)
|
|
|
|
tells = get_tells(db, nick)
|
|
|
|
if not tells:
|
|
notice("You have no pending tells.")
|
|
return
|
|
|
|
for tell in tells:
|
|
user_from, message, time, chan = tell
|
|
past = timesince.timesince(time)
|
|
notice("{} sent you a message {} ago from {}: {}".format(user_from, past, chan, message))
|
|
|
|
db.execute("delete from tell where user_to=lower(?)",
|
|
(nick,))
|
|
db.commit()
|
|
|
|
|
|
@hook.command
|
|
def tell(inp, nick='', chan='', db=None, input=None, notice=None):
|
|
"""tell <nick> <message> -- Relay <message> to <nick> when <nick> is around."""
|
|
query = inp.split(' ', 1)
|
|
|
|
if len(query) != 2:
|
|
notice(tell.__doc__)
|
|
return
|
|
|
|
user_to = query[0].lower()
|
|
message = query[1].strip()
|
|
user_from = nick
|
|
|
|
if chan.lower() == user_from.lower():
|
|
chan = 'a pm'
|
|
|
|
if user_to == user_from.lower():
|
|
notice("Have you looked in a mirror lately?")
|
|
return
|
|
|
|
if user_to.lower() == input.conn.nick.lower():
|
|
# user is looking for us, being a smartass
|
|
notice("Thanks for the message, {}!".format(user_from))
|
|
return
|
|
|
|
if not re.match("^[A-Za-z0-9_|.\-\]\[]*$", user_to.lower()):
|
|
notice("I cant send a message to that user!")
|
|
return
|
|
|
|
db_init(db)
|
|
|
|
if db.execute("select count() from tell where user_to=?",
|
|
(user_to,)).fetchone()[0] >= 10:
|
|
notice("That person has too many messages queued.")
|
|
return
|
|
|
|
try:
|
|
db.execute("insert into tell(user_to, user_from, message, chan,"
|
|
"time) values(?,?,?,?,?)", (user_to, user_from, message,
|
|
chan, time.time()))
|
|
db.commit()
|
|
except db.IntegrityError:
|
|
notice("Message has already been queued.")
|
|
return
|
|
|
|
notice("Your message has been sent!")
|