From f2f812412ca15462627968301f216fd1efb966f7 Mon Sep 17 00:00:00 2001 From: Chris Skalenda Date: Sat, 4 Jul 2009 22:27:05 -0500 Subject: [PATCH] Adding .tell. --- plugins/tell.py | 85 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 85 insertions(+) create mode 100644 plugins/tell.py diff --git a/plugins/tell.py b/plugins/tell.py new file mode 100644 index 0000000..1744678 --- /dev/null +++ b/plugins/tell.py @@ -0,0 +1,85 @@ +" tell.py: written by sklnd in July 2009" + +import sqlite3 +import hook +import os + +dbname = "skydb" + +@hook.command(hook=r'(.*)', prefix=False, ignorebots=True) +def tellinput(bot,input): + dbpath = os.path.join(bot.persist_dir, dbname) + conn = dbconnect(dbpath) + + cursor = conn.cursor() + command = "select count(name) from tell where name = ? and chan = ?" + results = cursor.execute(command, (input.nick,input.chan)).fetchone() + + + if results[0] > 0: + command = "select id, user_from, quote from tell where name = ? and chan = ?" + tells = cursor.execute(command, (input.nick, input.chan)).fetchall() + + for tell in tells: + bot.reply("{0} says '{1}'".format(tell[1], tell[2])) + command = "delete from tell where id = ?" + cursor.execute(command, (tell[0],)) + + conn.commit() + conn.close() + + + +@hook.command +def tell(bot, input): + + ".tell - Tell somebody something later. Unless you suck, in which case tell yourself to go take a hike" + + if len(input.msg) < 6: + return tell.__doc__ + + query = input.msg[6:].partition(" ") + + if query[0] == input.nick: + return "No." + + + if query[2] != "": + dbpath = os.path.join(bot.persist_dir, dbname) + conn = dbconnect(dbpath) + filterUser = conn.execute("select count(*) from tell_probation where name=?", (input.nick,)).fetchone() + + if filterUser[0] > 0: + return "No." + + cursor = conn.cursor() + command = "insert into tell(name, user_from, quote, chan) values(?,?,?,?)" + cursor.execute(command, (query[0], input.nick, query[2], input.chan)) + + conn.commit() + conn.close() + return "I'll let (him|her|it) know." + + else: + return tell.__doc__ + +# 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=?", ("tell" ,)).fetchone() + + if results[0] == 0: + conn.execute("create table if not exists "+ \ + "tell(id integer primary key autoincrement, name varchar(50), user_from varchar(50), quote varchar(250), chan varchar(50));") + + conn.commit() + + results = conn.execute("select count(*) from sqlite_master where name=?", ("tell_probation" ,)).fetchone() + if results[0] == 0: + conn.execute("create table if not exists "+ \ + "tell_probation(name varchar(50),"+ \ + "primary key(name));") + conn.commit() + + return conn +