80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
|
" seen.py: written by sklnd in about two beers July 2009"
|
||
|
|
|
||
|
|
import sqlite3
|
||
|
|
import datetime, time
|
||
|
|
import hook
|
||
|
|
import os
|
||
|
|
|
||
|
|
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:]
|
||
|
|
|
||
|
|
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):
|
||
|
|
date = time.gmtime(results[0])
|
||
|
|
quote = results[1]
|
||
|
|
return time.strftime(query+" was last seen %a %Y-%m-%d %I:%M:%S %p GMT saying: \"<"+query+"> "+quote+"\"", date)
|
||
|
|
else:
|
||
|
|
return "I've never seen "+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), date datetime, quote varchar(250), chan varchar(50),"+ \
|
||
|
|
"primary key(name, chan));")
|
||
|
|
conn.commit()
|
||
|
|
|
||
|
|
return conn
|
||
|
|
|