Added the reminders module.

1 files changed, 136 insertions, 0 deletions

diff --git a/modules/remind.py b/modules/remind.py
new file mode 100755
index 0000000..ec1a4d1
--- /dev/null
+++ b/modules/remind.py
@@ -0,0 +1,136 @@
+#!/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, 'rb')
+      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, 'wb')
+   for unixtime, reminders in data.iteritems(): 
+      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.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(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']
+
+if __name__ == '__main__': 
+   print __doc__.strip()