Merge lp:~keegan-csmith/ibid/uct into lp:ibid

[Keegan Carruthers-Smith]

[lp:~keegan-csmith/ibid/uct]

1050. By Keegan Carruthers-Smith on 2012-03-01

Don't load uct plugins by default.

1051. By Keegan Carruthers-Smith on 2012-03-01

Add copyright header and fix feature/help definition.

1052. By Keegan Carruthers-Smith on 2012-03-01

Add a UCT help category.

Unmerged revisions

1052. By Keegan Carruthers-Smith on 2012-03-01

Add a UCT help category.

1051. By Keegan Carruthers-Smith on 2012-03-01

Add copyright header and fix feature/help definition.

1050. By Keegan Carruthers-Smith on 2012-03-01

Don't load uct plugins by default.

1049. By Keegan Carruthers-Smith on 2012-03-01

UCT Club burger menu parser \o/

1048. By Keegan Carruthers-Smith on 2012-03-01

Rename jammie.py to uct.py

1047. By Keegan Carruthers-Smith on 2012-03-01

Be more relaxed for place name matching.

1046. By Keegan Carruthers-Smith on 2012-03-01

Use a more relaxed regex for jammie schedule matching.

1045. By Keegan Carruthers-Smith on 2012-02-28

Plugin for parsing the jammie shuttle schedule.

Jammie Shuttle is the UCT bus service. This plugin uses a beta
WhereIsMyTransport site. So is destined to break in the future.

=== modified file 'ibid/plugins/__init__.py'
--- ibid/plugins/__init__.py	2011-02-27 12:33:07 +0000
+++ ibid/plugins/__init__.py	2012-03-01 15:54:17 +0000
@@ -45,6 +45,7 @@
             ('message', u'delivering messages', -5),
             ('south africa', u'South African stuff', 10),
             ('sysadmin', u'System Administration', 5),
+            ('uct', u'University of Cape Town stuff', 10),
         ):
     if cat not in ibid.categories:
         ibid.categories[cat] = {

=== added file 'ibid/plugins/uct.py'
--- ibid/plugins/uct.py	1970-01-01 00:00:00 +0000
+++ ibid/plugins/uct.py	2012-03-01 15:54:17 +0000
@@ -0,0 +1,107 @@
+# Copyright (c) 2012, Keegan Carruthers-Smith
+# Released under terms of the MIT/X/Expat Licence. See COPYING for details.
+
+import re
+import urllib
+import urllib2
+
+from BeautifulSoup import BeautifulSoup
+from ibid.plugins import Processor, match
+
+features = {}
+
+features['jammie'] = {
+    'description': u'Find out where your Jammie Shuttle is',
+    'categories': ('lookup', 'south africa', 'uct'),
+}
+class Jammie(Processor):
+    usage = u'''jammie from <depart> to <destination>'
+                jammie from <depart>
+                jammie to <destination>'''
+    features = ('jammie',)
+    autoload = False
+
+    places = ('bremner', 'claremont', 'clarinus', 'dean str', 'forest hill',
+              'glendower', 'groote schuur', 'health science', 'hiddingh hall',
+              'kopano', 'liesbeeck', 'mowbray', 'pinelands grove',
+              'rochester', 'sandown', 'tugwell', 'uct')
+
+    @match(r'jammie( from (\S+(:? \S+)?))?( to {any})?')
+    def jammie(self, event, ignore1, depart, ignore2, ignore3, destination):
+        dep = Jammie._search_for_place(depart)
+        dest = Jammie._search_for_place(destination)
+
+        if dep is None or dest is None:
+            if dep is None:
+                place = depart
+            else:
+                place = destination
+            event.addresponse(u"Sorry, I don't know the place '%s', however "
+                              u"I do know the places %s",
+                              (place, ', '.join(Jammie.places)))
+            return
+
+        times = Jammie._get_times(dep, dest)
+        if times['live']:
+            event.addresponse(u'; '.join(times['live']))
+        elif times['schedule']:
+            event.addresponse(u'Schedule for %(dep)s to %(dest)s: '
+                              u'%(schedule)s',
+                              dict(dep=dep, dest=dest,
+                                   schedule='; '.join(times['schedule'])))
+        else:
+            event.addresponse(u'No schedule or live data found for %(dep)s '
+                              u'to %(dest)s', dict(dep=dep, dest=dest))
+
+    @classmethod
+    def _search_for_place(cls, place):
+        '''Search for place in places.
+        If not found try to match the first word for each place.'''
+        if place is None:
+            return 'uct'
+        place = place.lower()
+        if place in cls.places:
+            return place
+        if len(place) < 3:
+            return None
+        for p in cls.places:
+            if p.startswith(place):
+                return p
+
+    @classmethod
+    def _get_times(cls, depart, destination):
+        query = urllib.urlencode((('dep', depart.lower()),
+                                  ('des', destination.lower())))
+        resp = urllib2.urlopen(
+            'http://41.134.54.204:8014/MOBILEp/JAM3.pgm?' + query).read()
+
+        live = re.findall(r'(?:<h4>|<BR>)(.*?\(Live\).*?)<BR>', resp)
+        schedule = re.findall(r'\(Schedule\)(.*)</h4>', resp)
+        if schedule:
+            schedule = schedule[0].split('<BR><BR>')[1:]
+        return dict(live=live, schedule=schedule)
+
+
+features['uctburger'] = {
+    'description': u'Find out what todays gourmet burger is at UCT Club',
+    'categories': ('lookup', 'south africa', 'uct'),
+}
+class UCTBurger(Processor):
+    usage = u'what is the burger [at uct club]'
+    features = ('uctburger',)
+    autoload = False
+
+    @match(r'what is the (?:gourmet )?burger(?: at uct club(?: today)?)?\??')
+    def burger(self, event):
+        menu = urllib2.urlopen('http://www.uctclub.uct.ac.za/index.php?pid=43').read()
+        for prev, next in zip(menu.splitlines(), menu.splitlines()[1:]):
+            if 'gourmet burger' in prev.lower():
+                break
+        else:
+            event.addresponse(u'Could not find the gourmet burger on the menu')
+            return
+        desc = re.findall(r'>([^<]*?)<br', next)[0]
+        desc = BeautifulSoup(desc, convertEntities=BeautifulSoup.HTML_ENTITIES)
+        event.addresponse(desc)
+
+# vi: set et sta sw=4 ts=4: