You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

110 lines
2.2 KiB

#!/usr/bin/env python
"""
seen.py - Phenny Seen Module
Copyright 2017 sfan5
Licensed under GNU General Public License v2.0
"""
import os.path
import time
from threading import Thread, Lock
import sqlite3
DBPATH = "data/seen.sqlite"
updates = list()
updates_l = Lock()
def updatethread():
global updates, updates_l
db = sqlite3.connect(DBPATH)
while True:
if len(updates) == 0:
time.sleep(15)
continue
updates_l.acquire()
up = updates
updates = list()
updates_l.release()
db.executemany("REPLACE INTO seen(channel, time, nick) VALUES (?, ?, ?)", up)
db.commit()
db.close()
def pushupdate(sender, nick):
ts = int(time.mktime(time.gmtime()))
nick = nick.lower()
updates_l.acquire()
updates.append((sender, ts, nick))
updates_l.release()
class SeenApi(object):
@staticmethod
def seen(nick):
db = sqlite3.connect(DBPATH)
c = db.execute("SELECT channel, time FROM seen WHERE nick = ?", (nick, ))
r = c.fetchone()
db.close()
return r
_export = {
'seen': SeenApi,
}
def seen(phenny, input):
"""seen <nick> - Reports when <nick> was last seen."""
nick = input.group(2)
if not nick:
return phenny.reply("Need a nickname to search for...")
nick = nick.lower()
log.log("event", "%s queried Seen database for '%s'" % (log.fmt_user(input), nick), phenny)
r = SeenApi.seen(nick)
if r is None:
return phenny.reply("Sorry, I haven't seen %s around." % nick)
channel, t = r[0], r[1]
t = time.strftime('%Y-%m-%d %H:%M:%S UTC', time.gmtime(t))
phenny.reply("%s was last seen at %s on %s" % (nick, t, channel))
seen.rule = (['seen'], r'(\S+)')
def create_note(event=None):
def f(phenny, input):
if input.sender.startswith("#"):
pushupdate(input.sender, input.nick)
f.rule = r'.*'
f.priority = "low"
f.thread = False
f.nohook = True
if event is not None:
f.event = event
return f
note = create_note()
note_join = create_note("JOIN")
note_part = create_note("PART")
if not os.path.exists(DBPATH):
db = sqlite3.connect(DBPATH)
db.execute('''
CREATE TABLE `seen` (
`nick` tinytext NOT NULL,
`channel` tinytext NOT NULL,
`time` int NOT NULL,
PRIMARY KEY (`nick`)
)
''')
db.close()
t = Thread(target=updatethread, name="seen.py database thread")
t.start()
if __name__ == '__main__':
print(__doc__.strip())