dr.botzo/modules/Weather.py
Brian S. Stephan 0bd681c324 convert most modules to use ! commands.
okay, it's time. we got around for a while with all sorts of silly
config options and exceptions and common strings triggering bot
commands. but now it's time to man up and expect modules to be
more strict and less loosey-goosey.

convert the popular modules (i.e. the ones that still work) to
trigger on !pi rather than pi, etc. usually, this is achieved via
regex searches, although there are some weird bugs (ones i'm hoping
are caused by other recursion/alias bugs and not this commit).

more code around this will be gutted soon, but this, at least,
means you can't say 'tweet that shit, yo' and accidentally trigger
the bot.
2011-01-07 20:37:24 -06:00

73 lines
3.2 KiB
Python

# coding: utf-8
"""
Weather - query various weather services for info
Copyright (C) 2010 Brian S. Stephan
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
import re
from extlib import irclib
from extlib import pywapi
from urllib import quote
from urllib2 import URLError
from Module import Module
class Weather(Module):
"""Provide weather lookup services to the bot."""
def do(self, connection, event, nick, userhost, replypath, what, admin_unlocked):
"""Query Google Weather for a location's weather."""
match = re.search('^!weather\s+(.*)$', what)
if match:
query = match.group(1)
try:
google_weather = pywapi.get_weather_from_google(quote(query))
city = google_weather['forecast_information']['city'].encode('utf-8')
condition = google_weather['current_conditions']['condition'].encode('utf-8')
temp_f = google_weather['current_conditions']['temp_f'].encode('utf-8')
temp_c = google_weather['current_conditions']['temp_c'].encode('utf-8')
wind = google_weather['current_conditions']['wind_condition'].encode('utf-8')
humidity = google_weather['current_conditions']['humidity'].encode('utf-8')
wind_speed_regex = re.compile("Wind: [NSEW]+ at ([0-9]+) mph")
matches = wind_speed_regex.search(wind)
if matches is not None and float(temp_f) < 50.0:
wind_speed = matches.group(1)
windchill = 35.74 + (0.6215 * float(temp_f)) - (35.75 * float(wind_speed)**0.16) + (0.4275 * float(temp_f) * float(wind_speed)**0.16)
else:
windchill = ""
weatherstr = "Current weather for " + city + ": " + condition + ". " + temp_f + "°F (" + temp_c + "°C), " + wind + ", " + humidity
if windchill is not "":
weatherstr += ", Wind chill: " + str(round(windchill, 2)) + "°F"
weatherstr += "."
for city in google_weather['forecasts']:
weatherstr += " " + city['day_of_week'].encode('utf-8') + ": " + city['condition'].encode('utf-8') + ". High " + city['high'].encode('utf-8') + "°F, Low " + city['low'].encode('utf-8') + "°F."
return self.reply(connection, replypath, weatherstr)
except URLError as e:
return self.reply(connection, replypath, "error connecting to google weather:" + str(e))
except IndexError as e:
return self.reply(connection, replypath, "error in pywapi: " + str(e))
# vi:tabstop=4:expandtab:autoindent
# kate: indent-mode python;indent-width 4;replace-tabs on;