CloudBot/disabled_stuff/tell.py

""" tell.py: written by sklnd in July 2009
       2010.01.25 - modified by Scaevolus"""

import time
import re

from util import hook, timesince

db_ready = []


def db_init(db, conn):
    """Check that our db has the tell table, create it if not."""
    global db_ready
    if not conn.name in db_ready:
        db.execute("create table if not exists tell"
                   "(user_to, user_from, message, chan, time,"
                   "primary key(user_to, message))")
        db.commit()
        db_ready.append(conn.name)


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(inp, input=None, notice=None, db=None, nick=None, conn=None):
    if 'showtells' in input.msg.lower():
        return

    db_init(db, conn)

    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, conn=None):
    """showtells -- View all pending tell messages (sent in a notice)."""

    db_init(db, conn)

    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, conn=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 smart-ass
        notice("Thanks for the message, {}!".format(user_from))
        return

    if not re.match("^[A-Za-z0-9_|.\-\]\[]*$", user_to.lower()):
        notice("I can't send a message to that user!")
        return

    db_init(db, conn)

    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!")