Rewrote plugins/time.py to use Wolfram Alpha

plugins/time.py

@@ -1,243 +1,34 @@
-# Plugin by Lukeroge with some code from Phenny
+# Plugin by Lukeroge
 
 from util import hook
 from util import http
-import re
-import time
-import locale
-import datetime
-from BeautifulSoup import BeautifulSoup
 
-tags_re = re.compile(r'<[^<]*?>')
+api_url = 'http://api.wolframalpha.com/v2/query?format=plaintext'
-
-TimeZones = {'KST': 9, 'CADT': 10.5, 'EETDST': 3, 'MESZ': 2, 'WADT': 9,
-             'EET': 2, 'MST': -7, 'WAST': 8, 'IST': 5.5, 'B': 2,
-             'MSK': 3, 'X': -11, 'MSD': 4, 'CETDST': 2, 'AST': -4,
-             'HKT': 8, 'JST': 9, 'CAST': 9.5, 'CET': 1, 'CEST': 2,
-             'EEST': 3, 'EAST': 10, 'METDST': 2, 'MDT': -6, 'A': 1,
-             'UTC': 0, 'ADT': -3, 'EST': -5, 'E': 5, 'D': 4, 'G': 7,
-             'F': 6, 'I': 9, 'H': 8, 'K': 10, 'PDT': -7, 'M': 12,
-             'L': 11, 'O': -2, 'MEST': 2, 'Q': -4, 'P': -3, 'S': -6,
-             'R': -5, 'U': -8, 'T': -7, 'W': -10, 'WET': 0, 'Y': -12,
-             'CST': -6, 'EADT': 11, 'Z': 0, 'GMT': 0, 'WETDST': 1,
-             'C': 3, 'WEST': 1, 'CDT': -5, 'MET': 1, 'N': -1, 'V': -9,
-             'EDT': -4, 'UT': 0, 'PST': -8, 'MEZ': 1, 'BST': 1,
-             'ACS': 9.5, 'ATL': -4, 'ALA': -9, 'HAW': -10, 'AKDT': -8,
-             'AKST': -9,
-             'BDST': 2}
-
-TZ1 = {
- 'NDT': -2.5,
- 'BRST': -2,
- 'ADT': -3,
- 'EDT': -4,
- 'CDT': -5,
- 'MDT': -6,
- 'PDT': -7,
- 'YDT': -8,
- 'HDT': -9,
- 'BST': 1,
- 'MEST': 2,
- 'SST': 2,
- 'FST': 2,
- 'CEST': 2,
- 'EEST': 3,
- 'WADT': 8,
- 'KDT': 10,
- 'EADT': 13,
- 'NZD': 13,
- 'NZDT': 13,
- 'GMT': 0,
- 'UT': 0,
- 'UTC': 0,
- 'WET': 0,
- 'WAT': -1,
- 'AT': -2,
- 'FNT': -2,
- 'BRT': -3,
- 'MNT': -4,
- 'EWT': -4,
- 'AST': -4,
- 'EST': -5,
- 'ACT': -5,
- 'CST': -6,
- 'MST': -7,
- 'PST': -8,
- 'YST': -9,
- 'HST': -10,
- 'CAT': -10,
- 'AHST': -10,
- 'NT': -11,
- 'IDLW': -12,
- 'CET': 1,
- 'MEZ': 1,
- 'ECT': 1,
- 'MET': 1,
- 'MEWT': 1,
- 'SWT': 1,
- 'SET': 1,
- 'FWT': 1,
- 'EET': 2,
- 'UKR': 2,
- 'BT': 3,
- 'ZP4': 4,
- 'ZP5': 5,
- 'ZP6': 6,
- 'WST': 8,
- 'HKT': 8,
- 'CCT': 8,
- 'JST': 9,
- 'KST': 9,
- 'EAST': 10,
- 'GST': 10,
- 'NZT': 12,
- 'NZST': 12,
- 'IDLE': 12
-}
-
-TZ2 = {
- 'ACDT': 10.5,
- 'ACST': 9.5,
- 'ADT': 3,
- 'AEDT': 11,
- 'AEST': 10,
- 'AHDT': 9,
- 'AHST': 10,
- 'AST': 4,
- 'AT': 2,
- 'AWDT': -9,
- 'AWST': -8,
- 'BAT': -3,
- 'BDST': -2,
- 'BET': 11,
- 'BST': -1,
- 'BT': -3,
- 'BZT2': 3,
- 'CADT': -10.5,
- 'CAST': -9.5,
- 'CAT': 10,
- 'CCT': -8,
- # 'CDT': 5,
- 'CED': -2,
- 'CET': -1,
- 'CST': 6,
- 'EAST': -10,
- # 'EDT': 4,
- 'EED': -3,
- 'EET': -2,
- 'EEST': -3,
- 'EST': 5,
- 'FST': -2,
- 'FWT': -1,
- 'GMT': 0,
- 'GST': -10,
- 'HDT': 9,
- 'HST': 10,
- 'IDLE': -12,
- 'IDLW': 12,
- # 'IST': -5.5,
- 'IT': -3.5,
- 'JST': -9,
- 'JT': -7,
- 'KST': -9,
- 'MDT': 6,
- 'MED': -2,
- 'MET': -1,
- 'MEST': -2,
- 'MEWT': -1,
- 'MST': 7,
- 'MT': -8,
- 'NDT': 2.5,
- 'NFT': 3.5,
- 'NT': 11,
- 'NST': -6.5,
- 'NZ': -11,
- 'NZST': -12,
- 'NZDT': -13,
- 'NZT': -12,
- 'PDT': 7,
- 'PST': 8,
- 'ROK': -9,
- 'SAD': -10,
- 'SAST': -9,
- 'SAT': -9,
- 'SDT': -10,
- 'SST': -2,
- 'SWT': -1,
- 'USZ3': -4,
- 'USZ4': -5,
- 'USZ5': -6,
- 'USZ6': -7,
- 'UT': 0,
- 'UTC': 0,
- 'UZ10': -11,
- 'WAT': 1,
- 'WET': 0,
- 'WST': -8,
- 'YDT': 8,
- 'YST': 9,
- 'ZP4': -4,
- 'ZP5': -5,
- 'ZP6': -6
-}
-
-TZ3 = {
-   'AEST': 10,
-   'AEDT': 11
-}
-
-# TimeZones.update(TZ2) # do these have to be negated?
-TimeZones.update(TZ1)
-TimeZones.update(TZ3)
-
-r_local = re.compile(r'\([a-z]+_[A-Z]+\)')
-
-
-# gets the time from a timezone (code from phenny)
-def get_time(tz):
-    TZ = tz.upper()
-
-    if (TZ == 'UTC') or (TZ == 'Z'):
-        msg = time.strftime("\x02%I:%M%p\x02 %A - \x02Time\x02 in \x02 " \
-                            "GMT", time.gmtime())
-        return msg
-    elif r_local.match(tz):
-        locale.setlocale(locale.LC_TIME, (tz[1:-1], 'UTF-8'))
-        msg = time.strftime("\x02%I:%M%p\x02 %A - \x02Time\x02 in \x02" \
-                            + str(TZ), time.gmtime())
-        return msg
-    elif TZ in TimeZones:
-        offset = TimeZones[TZ] * 3600
-        timenow = time.gmtime(time.time() + offset)
-        msg = time.strftime("\x02%I:%M%p\x02 %A - \x02Time\x02 in \x02" \
-                            + str(TZ), timenow)
-        return msg
-    elif tz and tz[0] in ('+', '-') and 4 <= len(tz) <= 6:
-        timenow = time.gmtime(time.time() + (int(tz[:3]) * 3600))
-        msg = time.strftime("\x02%I:%M%p\x02 %A - \x02Time\x02 in\x02" \
-                            " GMT" + str(tz), timenow)
-        return msg
-    else:
-        timenow = time.gmtime(time.time() + (int(tz) * 3600))
-        msg = time.strftime("\x02%I:%M%p\x02 %A - \x02Time\x02 in\x02" \
-                            " GMT " + str(tz), timenow)
-        return msg
 
 
 @hook.command("time")
-def time_command(inp):
+def time_command(inp, bot=None):
     ".time <area> -- Gets the time in <area>"
 
-    request = http.get('http://www.google.com/search', q="time in " + inp)
+    query = "current time in %s" % inp
-    soup = BeautifulSoup(request)
+	
-    response = soup.find('td', {'style': 'font-size:medium'})
+    api_key = bot.config.get("api_keys", {}).get("wolframalpha", None)
+    if not api_key:
+        return "error: no wolfram alpha api key set"
 
-    if response:
+    request = http.get_xml(api_url, input=query, appid=api_key)
-        output = response.renderContents()
+    time = " ".join(request.xpath("//pod[@title='Result']/subpod/plain" \
-        output = tags_re.sub('\x02', output.strip())
+                    "text/text()"))
-        return output
+    time = time.replace("  |  ", ", ")
+    place = " ".join(request.xpath("//pod[@title='Input interpretation']/" \
+                     "subpod/plaintext/text()"))[16:].title()
+
+    if time:
+	    # if wolfram alpha had to guess a place, then show the place it chose
+        #if request.xpath("//assumptions"):
+        #    return "%s - \x02%s\x02" % (time, place)
+        #else:
+        #    return time
+		return "%s - \x02%s\x02" % (time, place)
     else:
-        try:
+        return "Could not get the time for '%s'" % inp
-            return get_time(inp)
-        except ValueError:
-            return "Could not get the time for " + inp + "!"