phenny-1/modules/remind.py

#!/usr/bin/env python
"""
remind.py - Phenny Reminder Module
Copyright 2011, Sean B. Palmer, inamidst.com
Licensed under the Eiffel Forum License 2.

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

import os, re, time, threading

def filename(self): 
    name = self.nick + '-' + self.config.host + '.reminders.db'
    return os.path.join(os.path.expanduser('~/.phenny'), name)

def load_database(name): 
    data = {}
    if os.path.isfile(name): 
        f = open(name, 'r')
        for line in f: 
            unixtime, channel, nick, message = line.split('\t')
            message = message.rstrip('\n')
            t = int(unixtime)
            reminder = (channel, nick, message)
            try: data[t].append(reminder)
            except KeyError: data[t] = [reminder]
        f.close()
    return data

def dump_database(name, data): 
    f = open(name, 'w')
    for unixtime, reminders in data.items(): 
        for channel, nick, message in reminders: 
            f.write('%s\t%s\t%s\t%s\n' % (unixtime, channel, nick, message))
    f.close()

def setup(phenny): 
    phenny.rfn = filename(phenny)
    phenny.rdb = load_database(phenny.rfn)

    def monitor(phenny): 
        time.sleep(5)
        while True: 
            now = int(time.time())
            unixtimes = [int(key) for key in phenny.rdb]
            oldtimes = [t for t in unixtimes if t <= now]
            if oldtimes: 
                for oldtime in oldtimes: 
                    for (channel, nick, message) in phenny.rdb[oldtime]: 
                        if message: 
                            phenny.msg(channel, nick + ': ' + message)
                        else: phenny.msg(channel, nick + '!')
                    del phenny.rdb[oldtime]
                dump_database(phenny.rfn, phenny.rdb)
            time.sleep(2.5)

    targs = (phenny,)
    t = threading.Thread(target=monitor, args=targs)
    t.daemon = True
    t.start()

scaling = {
    'years': 365.25 * 24 * 3600, 
    'year': 365.25 * 24 * 3600, 
    'yrs': 365.25 * 24 * 3600, 
    'y': 365.25 * 24 * 3600, 

    'months': 29.53059 * 24 * 3600, 
    'month': 29.53059 * 24 * 3600, 
    'mo': 29.53059 * 24 * 3600, 

    'weeks': 7 * 24 * 3600, 
    'week': 7 * 24 * 3600, 
    'wks': 7 * 24 * 3600, 
    'wk': 7 * 24 * 3600, 
    'w': 7 * 24 * 3600, 

    'days': 24 * 3600, 
    'day': 24 * 3600, 
    'd': 24 * 3600, 

    'hours': 3600, 
    'hour': 3600, 
    'hrs': 3600, 
    'hr': 3600, 
    'h': 3600, 

    'minutes': 60, 
    'minute': 60, 
    'mins': 60, 
    'min': 60, 
    'm': 60, 

    'seconds': 1, 
    'second': 1, 
    'secs': 1, 
    'sec': 1, 
    's': 1
}

periods = '|'.join(list(scaling.keys()))
p_command = r'\.in ([0-9]+(?:\.[0-9]+)?)\s?((?:%s)\b)?:?\s?(.*)' % periods
r_command = re.compile(p_command)

def remind(phenny, input): 
    m = r_command.match(input.bytes)
    if not m: 
        return phenny.reply("Sorry, didn't understand the input.")
    length, scale, message = m.groups()

    length = float(length)
    factor = scaling.get(scale, 60)
    duration = length * factor

    if duration % 1: 
        duration = int(duration) + 1
    else: duration = int(duration)

    t = int(time.time()) + duration
    reminder = (input.sender, input.nick, message)

    try: phenny.rdb[t].append(reminder)
    except KeyError: phenny.rdb[t] = [reminder]

    dump_database(phenny.rfn, phenny.rdb)

    if duration >= 60: 
        w = ''
        if duration >= 3600 * 12: 
            w += time.strftime(' on %d %b %Y', time.gmtime(t))
        w += time.strftime(' at %H:%MZ', time.gmtime(t))
        phenny.reply('Okay, will remind%s' % w)
    else: phenny.reply('Okay, will remind in %s secs' % duration)
remind.commands = ['in']

r_time = re.compile(r'^([0-9]{2}[:.][0-9]{2})')
r_zone = re.compile(r'( ?([A-Za-z]+|[+-]\d\d?))')

import calendar
from modules.clock import TimeZones

def at(phenny, input):
    message = input[4:]

    m = r_time.match(message)
    if not m: 
        return phenny.reply("Sorry, didn't understand the time spec.")
    t = m.group(1).replace('.', ':')
    message = message[len(t):]

    m = r_zone.match(message)
    if not m: 
        return phenny.reply("Sorry, didn't understand the zone spec.")
    z = m.group(2)
    message = message[len(m.group(1)):].strip()

    if z.startswith('+') or z.startswith('-'):
        tz = int(z)

    if z in TimeZones:
        tz = TimeZones[z]
    else: return phenny.reply("Sorry, didn't understand the time zone.")

    d = time.strftime("%Y-%m-%d", time.gmtime())
    d = time.strptime(("%s %s" % (d, t)), "%Y-%m-%d %H:%M")

    d = int(calendar.timegm(d) - (3600 * tz))
    duration = int((d - time.time()) / 60)

    if duration < 1:
        return phenny.reply("Sorry, that date is this minute or in the past. And only times in the same day are supported!")

    # phenny.say("%s %s %s" % (t, tz, d))

    reminder = (input.sender, input.nick, message)
    # phenny.say(str((d, reminder)))
    try: phenny.rdb[d].append(reminder)
    except KeyError: phenny.rdb[d] = [reminder]

    phenny.sending.acquire()
    dump_database(phenny.rfn, phenny.rdb)
    phenny.sending.release()

    phenny.reply("Reminding at %s %s - in %s minute(s)" % (t, z, duration))
at.commands = ['at']

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