CloudBot/plugins/weather.py

"weather, thanks to google"
from util import hook, http


def fahrenheit_to_celcius(f):
    return int(round((int(f) - 32) / 1.8, 0))


@hook.command(autohelp=False)
def forecast(inp, nick='', server='',
    reply=None, db=None, notice=None, say=None):
    ".forecast <location> [dontsave] -- Gets a weather forecast" \
    " for <location> from Google."
    loc = inp

    dontsave = loc.endswith(" dontsave")
    if dontsave:
        loc = loc[:-9].strip().lower()

    db.execute("create table if not exists weather(nick primary key, loc)")

    if not loc:
        loc = db.execute("select loc from weather where nick=lower(?)",
                            (nick,)).fetchone()
        if not loc:
            notice(forecast.__doc__)
            return
        loc = loc[0]

    w = http.get_xml('http://www.google.com/ig/api', weather=loc)
    w = w.find('weather')

    if w.find('problem_cause') is not None:
        notice("Couldn't fetch weather data for '%s', try using a zip or " \
                "postal code." % inp)
        return
    city = w.find('forecast_information/city').get('data')

    out = "%s: " % city

    for elem in w.findall('forecast_conditions'):
        info = dict((e.tag, e.get('data')) for e in elem)
        info['high'] = elem.find('high').get('data')
        info['low'] = elem.find('low').get('data')
        info['high_c'] = fahrenheit_to_celcius(elem.find('high').get('data'))
        info['low_c'] = fahrenheit_to_celcius(elem.find('low').get('data'))
        out += '\x02%(day_of_week)s\x02: %(condition)s (High: %(high)sF' \
          '/%(high_c)sC) (Low: %(low)sF/%(low_c)sC) ' % info

    return out


@hook.command(autohelp=False)
def weather(inp, nick='', server='', reply=None, db=None, notice=None):
    ".weather <location> [dontsave] -- Gets weather data"\
    " for <location> from Google."
    loc = inp

    dontsave = loc.endswith(" dontsave")
    if dontsave:
        loc = loc[:-9].strip().lower()

    db.execute("create table if not exists weather(nick primary key, loc)")

    if not loc:
        loc = db.execute("select loc from weather where nick=lower(?)",
                            (nick,)).fetchone()
        if not loc:
            notice(weather.__doc__)
            return
        loc = loc[0]

    w = http.get_xml('http://www.google.com/ig/api', weather=loc)
    w = w.find('weather')

    if w.find('problem_cause') is not None:
        notice("Couldn't fetch weather data for '%s', try using a zip or " \
                "postal code." % inp)
        return

    info = dict((e.tag, e.get('data')) for e in w.find('current_conditions'))
    info['city'] = w.find('forecast_information/city').get('data')
    info['high'] = w.find('forecast_conditions/high').get('data')
    info['low'] = w.find('forecast_conditions/low').get('data')
    info['high_c'] = fahrenheit_to_celcius(info['high'])
    info['low_c'] = fahrenheit_to_celcius(info['low'])

    reply('%(city)s: %(condition)s, %(temp_f)sF/%(temp_c)sC (High: %(high)sF' \
          '/%(high_c)sC) (Low: %(low)sF/%(low_c)sC), %(humidity)s, ' \
          '%(wind_condition)s.' % info)

    if inp and not dontsave:
        db.execute("insert or replace into weather(nick, loc) values (?,?)",
                     (nick.lower(), loc))
        db.commit()