Merge pull request CloudBotIRC#110 from edillingham/brew-plugin

Brew plugin

Author: edwardslabs

config.default.json

@@ -84,7 +84,8 @@
         "wordnik": "",
         "yandex_translate": "",
         "lyricsnmusic": "",
-	"cleverbot":""
+        "cleverbot":"",
+        "brewerydb": ""
     },
     "database": "sqlite:///cloudbot.db",
     "plugin_loading": {

plugins/brew.py

@@ -0,0 +1,63 @@
+import requests
+
+from cloudbot import hook
+
+api_url = "http://api.brewerydb.com/v2/search?format=json"
+
+
+@hook.on_start()
+def load_key(bot):
+    global api_key
+    api_key = bot.config.get("api_keys", {}).get("brewerydb", None)
+
+
+@hook.command('brew')
+def brew(text, bot):
+    """<query> - returns the first brewerydb search result for <query>"""
+
+    if not api_key:
+        return "No brewerydb API key set."
+
+    params = {'key': api_key, 'type': 'beer', 'withBreweries': 'Y', 'q': text}
+    request = requests.get(api_url, params=params)
+
+    if request.status_code != requests.codes.ok:
+        return "Failed to fetch info ({})".format(request.status_code)
+
+    response = request.json()
+
+    output = "No results found."
+
+    try:
+        if 'totalResults' in response:
+            beer = response['data'][0]
+            brewery = beer['breweries'][0]
+
+            style = 'unknown style'
+            if 'style' in beer:
+                style = beer['style']['shortName']
+
+            abv = '?.?'
+            if 'abv' in beer:
+                abv = beer['abv']
+
+            url = '[no website found]'
+            if 'website' in brewery:
+                url = brewery['website']
+
+            content = {
+                'name': beer['nameDisplay'],
+                'style': style,
+                'abv': abv,
+                'brewer': brewery['name'],
+                'url': url
+            }
+
+            output = "{name} by {brewer} ({style}, {abv}% ABV) - {url}" \
+                .format(**content)
+
+    except Exception as e:
+        print(e)
+        output = "Error parsing results."
+
+    return output