add foodforus module

modules/foodforus.py

@@ -0,0 +1,95 @@
+#!/usr/bin/python3
+"""
+foodforus.py - foodforus module
+author: mutantmonkey <mutantmonkey@mutantmonkey.in>
+"""
+
+from urllib.error import HTTPError
+from tools import GrumbleError
+import hashlib
+import json
+import web
+
+API_URL = 'http://foodfor.vtluug.org'
+
+
+def _sign_vote(api_key, args):
+    data = "ffu0" + api_key
+    for k, v in sorted(args.items()):
+        if k == 'sig':
+            continue
+        data += '{0}{1}'.format(k, v)
+    h = hashlib.sha256()
+    h.update(data.encode('utf-8'))
+    return h.hexdigest()
+
+
+def food(phenny, input):
+    """.food"""
+    key = input.group(2) or input.sender
+    try:
+        req = web.get(API_URL + '/food/' + web.quote(key.strip()))
+        data = json.loads(req)
+    except (HTTPError, IOError):
+        raise GrumbleError("Uh oh, I couldn't contact foodforus. HOW WILL WE "\
+                "EAT NOW‽")
+
+    restaurants = data['restaurants'][:4]
+    times = data['times'][:4]
+
+    restr = ", ".join(["{0} ({1})".format(r[0], r[1]) for r in
+        restaurants])
+    tistr = ", ".join(["{0} ({1})".format(t[0], t[1]) for t in times])
+    
+    if len(restr) > 0 and len(tistr) > 0:
+        return phenny.say("{0} at {1}".format(restr, tistr))
+    else:
+        return phenny.say("Sorry, people need to vote before we can food!")
+food.rule = (['food'], r'(.*)')
+
+
+def foodvote(phenny, input):
+    """.foodvote"""
+    if not input.group(2) or not input.group(3):
+        return phenny.reply("You need to specify a place and time, as in "\
+                ".foodvote hokie haus 18:45")
+
+    key = input.group(4) or input.sender
+    postdata = {
+        'user': input.nick,
+        'restaurant': input.group(2),
+        'start': input.group(3),
+        'key': key.strip(),
+    }
+    postdata['sig'] = _sign_vote(phenny.config.foodforus_api_key, postdata)
+
+    try:
+        req = web.post(API_URL + '/vote', postdata)
+        data = json.loads(req)
+    except (HTTPError, IOError):
+        raise GrumbleError("Uh oh, I couldn't contact foodforus. HOW WILL WE "\
+                "EAT NOW‽")
+
+    phenny.reply("Your vote has been recorded.")
+foodvote.rule = (['foodvote'], r'(.*) (\d{2}:\d{2})( .*)?')
+
+
+def pickfood(phenny, input):
+    key = input.group(2) or input.sender
+    try:
+        req = web.get(API_URL + '/food/' + web.quote(key.strip()))
+        data = json.loads(req)
+    except (HTTPError, IOError):
+        raise GrumbleError("Uh oh, I couldn't contact foodforus. HOW WILL WE "\
+                "EAT NOW‽")
+
+    if len(data['restaurants']) > 0 and len(data['times']) > 0:
+        restaurant = data['restaurants'][0]
+        time = data['times'][0]
+
+        phenny.say("Food is {place} ({place_votes} votes) at {time} "\
+                "({time_votes} votes). Happy fooding!".format(place=restaurant[0],
+                place_votes=restaurant[1], time=time[0], time_votes=time[1]))
+    else:
+        phenny.say("Sorry, people need to vote before we can food!")
+pickfood.rule = (['pickfood'], r'(.*)')