109 lines
2.7 KiB
Python
Executable File
109 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""
|
|
seen.py - Phenny Seen Module
|
|
Copyright 2013, sfan5
|
|
"""
|
|
|
|
import random
|
|
from thread import start_new_thread, allocate_lock
|
|
import sqlite3
|
|
import time
|
|
import hashlib
|
|
|
|
tell_list = []
|
|
tell_pending = []
|
|
|
|
def tell_diskwr():
|
|
global tell_pending, tell_list
|
|
db = sqlite3.connect("tell.sqlite")
|
|
c = db.cursor()
|
|
for tr in tell_pending:
|
|
if tr[0] == "del":
|
|
c.execute("DELETE FROM tell WHERE id = ?", (tr[1], ))
|
|
elif tr[0] == "add":
|
|
c.execute("INSERT INTO tell (nick, tellee, msg, time) VALUES (?,?,?,?)", tr[1])
|
|
tell_list.append((c.lastrowid, ) + tr[1]) # We actually insert the entry into the list here
|
|
else:
|
|
print("[Tell] Internal error: Unknown action type '%s'" % tr[0])
|
|
c.close()
|
|
db.commit()
|
|
db.close()
|
|
tell_pending = []
|
|
|
|
def tell(phenny, input):
|
|
arg = input.group(2)
|
|
if not arg:
|
|
return phenny.reply("Need a nickname...")
|
|
if not ' ' in arg:
|
|
return phenny.reply("...and text")
|
|
teller = input.nick
|
|
target = arg.split(" ")[0]
|
|
text = " ".join(arg.split(" ")[1:])
|
|
if target.lower() == teller.lower():
|
|
return phenny.say("You can tell that to yourself")
|
|
elif target.lower() == phenny.nick.lower():
|
|
return phenny.say("I'm not dumb, you know?")
|
|
elif target[-1] == ":":
|
|
return phenny.reply("Do not put an : at the end of nickname")
|
|
|
|
d = (teller, target, text, int(time.time()))
|
|
tell_pending.append(("add", d))
|
|
# We do not insert the entry into tell_list yet because we don't know the id it will have
|
|
tell_diskwr() # Write the change to disk
|
|
|
|
response = "I'll pass that on when %s is around" % target
|
|
rand = random.random()
|
|
if rand > 0.85: response = "yeah, sure, whatever"
|
|
elif rand > 0.75: response = "yeah, yeah"
|
|
|
|
phenny.reply(response)
|
|
|
|
tell.commands = ["tell"]
|
|
|
|
def checktell(phenny, input):
|
|
for e in tell_list:
|
|
if e[2].lower() == input.nick.lower():
|
|
phenny.say("%s: %s <%s> %s" % (
|
|
input.nick,
|
|
time.strftime('%m-%d %H:%M UTC',
|
|
time.gmtime(e[4])),
|
|
e[1],
|
|
e[3]))
|
|
tell_list.remove(e)
|
|
tell_pending.append(("del", e[0]))
|
|
tell_diskwr() # Write the change to disk
|
|
break
|
|
|
|
def note(phenny, input):
|
|
if input.sender.startswith('#'):
|
|
checktell(phenny, input)
|
|
|
|
note.rule = r'.*'
|
|
note.priority = 'low'
|
|
note.nohook = True
|
|
|
|
def note_join(phenny, input):
|
|
if input.sender.startswith('#'):
|
|
checktell(phenny, input)
|
|
|
|
note_join.rule = r'.*'
|
|
note_join.event = 'JOIN'
|
|
note_join.priority = 'low'
|
|
note_join.nohook = True
|
|
|
|
db = sqlite3.connect("tell.sqlite")
|
|
c = db.cursor()
|
|
c.execute("CREATE TABLE IF NOT EXISTS tell (id INTEGER PRIMARY KEY, nick TEXT, tellee TEXT, msg TEXT, time INTEGER)")
|
|
c.execute("SELECT * FROM tell")
|
|
while True:
|
|
e = c.fetchone()
|
|
if not e:
|
|
break
|
|
tell_list.append(e)
|
|
c.close()
|
|
db.commit()
|
|
db.close()
|
|
|
|
if __name__ == '__main__':
|
|
print __doc__.strip()
|