h/plugins/seen.py

" seen.py: written by sklnd in about two beers July 2009"

import sqlite3
import datetime, time
import hook
import os
from timesince import timesince

dbname = "skydb"

def adapt_datetime(ts):
    return time.mktime(ts.timetuple())

sqlite3.register_adapter(datetime.datetime, adapt_datetime)

@hook.command(hook=r'(.*)', prefix=False, ignorebots=False)
def seeninput(bot,input):
    dbpath = os.path.join(bot.persist_dir, dbname)

    conn = dbconnect(dbpath)

    cursor = conn.cursor()
    command = "select count(name) from seen where name = ? and chan = ?"
    cursor.execute(command, (input.nick,input.chan))

    if(cursor.fetchone()[0] == 0):
        command = "insert into seen(name, date, quote, chan) values(?,?,?,?)"
        cursor.execute(command, (input.nick, datetime.datetime.now(), input.msg, input.chan))
    else:
        command = "update seen set date=?, quote=? where name = ? and chan = ?"
        cursor.execute(command, (datetime.datetime.now(), input.msg, input.nick, input.chan))

    conn.commit()
    conn.close()

@hook.command
def seen(bot, input):
    ".seen <nick> - Tell when a nickname was last in active in irc"

    if len(input.msg) < 6:
        return seen.__doc__

    query = input.msg[6:].strip()

    if query == input.nick:
        return "Have you looked in a mirror lately?"

    dbpath = os.path.join(bot.persist_dir, dbname)
    conn = dbconnect(dbpath)
    cursor = conn.cursor()

    command = "select date, quote from seen where name = ? and chan = ?"
    cursor.execute(command, (query, input.chan))
    results = cursor.fetchone()

    conn.close()

    if(results != None):
        reltime = timesince(datetime.datetime.fromtimestamp(results[0]))
        return '%(name)s was last seen %(timespan)s ago saying: <%(name)s> %(quote)s' % \
                    {'name': query, 'timespan': reltime, 'quote': results[1]}
    else:
        return "I've never seen %(name)s" % {'name':query}


# check to see that our db has the the seen table, and return a connection.
def dbconnect(db):
    conn = sqlite3.connect(db)
    results = conn.execute("select count(*) from sqlite_master where name=?", ("seen" ,)).fetchone()

    if(results[0] == 0):
        conn.execute("create table if not exists "+ \
                     "seen(name varchar(50) not null, date datetime not null, "+ \
                     "quote varchar(250) not null, chan varchar(50) not null, "+ \
                     "primary key(name, chan));")
        conn.commit()

    return conn