minetestbot-modules/seen.py

#!/usr/bin/env python
"""
seen.py - Phenny Seen Module
Copyright 2008, Sean B. Palmer, inamidst.com
Modified by sfan5 2013
Licensed under the Eiffel Forum License 2.

http://inamidst.com/phenny/
"""

import time
from threading import Thread, Lock
import sqlite3

updates = list()
update_l = Lock()
dblock = Lock()

def opendb():
    db = sqlite3.connect("seen.sqlite")
    return db

def updatethread():
    global updates, update_l
    db = opendb()
    c = db.cursor()
    while True:
        if len(updates) > 0:
            update_l.acquire()
            up = updates
            updates = list()
            update_l.release()
            dblock.acquire()
            for u in up:
                c.execute("SELECT * FROM seen WHERE nick = ?", (u[2],))
                if c.fetchone() != None:
                    d = (u[0], u[1], u[2])
                    c.execute('UPDATE seen SET channel = ?, time = ? WHERE nick = ?', d)
                else:
                    d = (u[2], u[0], u[1])
                    c.execute('INSERT INTO seen VALUES (?,?,?)', d)
            db.commit()
            dblock.release()
        else:
            time.sleep(5)

def pushupdate(sender, nick):
    ts = int(time.mktime(time.gmtime()))
    nick = nick.lower()
    update_l.acquire()
    updates.append((sender, ts, nick))
    update_l.release()

def api_seen(nick):
  dblock.acquire()
  db = opendb()
  c = db.cursor()
  c.execute("SELECT channel, time FROM seen WHERE nick = ?", (nick,))
  r = c.fetchone()
  c.close()
  db.close()
  dblock.release()
  return r

class SomeObject(object):
  pass

seen_api = SomeObject()
seen_api.seen = api_seen

_export = {
  'seen': seen_api,
}

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 = api_seen(nick)

    if r:
        channel, t = r[0], r[1]
        t = time.strftime('%Y-%m-%d %H:%M:%S UTC', time.gmtime(t))

        msg = "%s was last seen at %s on %s" % (nick, t, channel)
        phenny.reply(msg)
    else:
        phenny.reply("Sorry, I haven't seen %s around." % nick)

seen.rule = (['seen'], r'(\S+)')

def note(phenny, input):
    if input.sender.startswith('#'):
        pushupdate(input.sender, input.nick)

note.rule = r'.*'
note.priority = 'low'
note.thread = False
note.nohook = True

def note_join(phenny, input):
    if input.sender.startswith('#'):
        pushupdate(input.sender, input.nick)

note_join.rule = r'.*'
note_join.event = 'JOIN'
note_join.priority = 'low'
note_join.thread = False
note_join.nohook = True

def note_part(phenny, input):
    if input.sender.startswith('#'):
        pushupdate(input.sender, input.nick)

note_part.rule = r'.*'
note_part.event = 'PART'
note_part.priority = 'low'
note_part.thread = False
note_part.nohook = True

db = sqlite3.connect("seen.sqlite")
c = db.cursor()
c.execute('''CREATE TABLE IF NOT EXISTS seen (nick text, channel text, time int)''')
c.close()
db.close()

t = Thread(target=updatethread, name="seen.py database thread")
t.start()

if __name__ == '__main__':
   print(__doc__.strip())