h/plugins/seen.py

78 lines
1.9 KiB
Python
Raw Normal View History

" seen.py: written by sklnd in about two beers July 2009"
import os
2009-07-08 17:28:15 +00:00
import time
2009-07-08 17:57:02 +00:00
from datetime import datetime
2009-07-08 17:28:15 +00:00
import sqlite3
from util import hook, timesince
2009-07-08 17:28:15 +00:00
2009-07-08 17:49:45 +00:00
dbname = "skybot.db"
2009-07-08 17:28:15 +00:00
def adapt_datetime(ts):
return time.mktime(ts.timetuple())
2009-07-08 17:48:22 +00:00
sqlite3.register_adapter(datetime, adapt_datetime)
2009-07-08 17:28:15 +00:00
2009-11-21 00:57:00 +00:00
@hook.tee
2009-07-08 17:28:15 +00:00
def seeninput(bot, input):
2009-11-21 00:57:00 +00:00
if input.command != 'PRIVMSG':
return
dbpath = os.path.join(bot.persist_dir, dbname)
conn = dbconnect(dbpath)
cursor = conn.cursor()
2009-11-21 00:57:00 +00:00
cursor.execute("INSERT OR REPLACE INTO seen(name, date, quote, chan)"
"values(?,?,?,?)", (input.nick.lower(), datetime.now(),
2009-07-08 17:28:15 +00:00
input.msg, input.chan))
conn.commit()
conn.close()
2009-07-08 17:28:15 +00:00
@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.inp
2009-11-21 00:57:00 +00:00
if query.lower() == input.nick.lower():
return "Have you looked in a mirror lately?"
dbpath = os.path.join(bot.persist_dir, dbname)
conn = dbconnect(dbpath)
cursor = conn.cursor()
2009-11-21 00:57:00 +00:00
command = "SELECT date, quote FROM seen WHERE name LIKE ? AND chan = ?" \
"ORDER BY date DESC"
cursor.execute(command, (query, input.chan))
results = cursor.fetchone()
conn.close()
if(results != None):
2009-07-08 17:48:22 +00:00
reltime = timesince.timesince(datetime.fromtimestamp(results[0]))
return '%s was last seen %s ago saying: %s' % \
2009-07-08 17:28:15 +00:00
(query, reltime, results[1])
else:
2009-07-08 17:28:15 +00:00
return "I've never seen %s" % query
def dbconnect(db):
2009-07-08 17:28:15 +00:00
"check to see that our db has the the seen table and return a connection."
conn = sqlite3.connect(db)
2009-11-21 00:57:00 +00:00
conn.execute("CREATE TABLE IF NOT EXISTS "
"seen(name varchar(30) not null, date datetime not null, "
"quote varchar(250) not null, chan varchar(32) not null, "
"primary key(name, chan));")
conn.commit()
return conn